Merge pull request #56 from JoeyT1994/tcifix

Fix TCI Extension

ext/ITensorNumericalAnalysisTCIExt/tci_util.jl

@@ -12,9 +12,10 @@ using ITensorNumericalAnalysis:
   const_itn,
   ITensorNetworkFunction,
   itensornetwork,
-  base
+  base,
+  dimension
 
-random_initial_pivot(s::IndsNetworkMap) = [v => rand(1:dim(v)) for v in vertices(s)]
+random_initial_pivot(s::IndsNetworkMap) = [v => rand(1:dim(s[v])) for v in vertices(s)]
 
 #f should be an ndimensional function that maps a vector of scalars of length ndimensional to a scalar
 function ITensorTCI.interpolate(
@@ -37,11 +38,12 @@ function ITensorTCI.interpolate(
     initial_pivot = random_initial_pivot(s_renamed)
   else
     # manually rename
-    initial_pivot = [forward_dict[v] => initial_pivot[v] for v in vertices(s)]
+    # assuming from calculate_ind_values
+    initial_pivot = [forward_dict[v] => initial_pivot[only(s[v])] + 1 for v in vertices(s)]
   end
 
   tn = ITensorTCI.interpolate(
-    input -> f(input_to_scalars(input; base=float(base(s)))),
+    input -> f(input_to_scalars(input; ndims=dimension(s), base=float(base(s)))),
     ttn(itensornetwork(tn));
     initial_pivot,
     kwargs...,
@@ -53,8 +55,7 @@ function ITensorTCI.interpolate(
 end
 
 #Takes a vector of [(dimension, digit) => bit] and converts to vector of scalars
-function input_to_scalars(input; base=2.0)
-  ndims = maximum(first.(input))
+function input_to_scalars(input; ndims, base=2.0)
   x = zeros(ndims)
   for pair in input
     (i, j) = pair[1]