Import scalartype from NDTensors

ITensor · Jun 25, 2024 · 6d275f1 · 6d275f1
1 parent 754e46c
commit 6d275f1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/caches/beliefpropagationcache.jl b/src/caches/beliefpropagationcache.jl
@@ -12,6 +12,7 @@ using NamedGraphs.PartitionedGraphs:
   partitionedges,
   unpartitioned_graph
 using SimpleTraits: SimpleTraits, Not, @traitfn
+using NDTensors: NDTensors
 
 default_message(elt, inds_e) = ITensor[denseblocks(delta(elt, i)) for i in inds_e]
 default_messages(ptn::PartitionedGraph) = Dictionary()
@@ -97,7 +98,7 @@ for f in [
   end
 end
 
-ITensorNetworks.scalartype(bp_cache) = scalartype(tensornetwork(bp_cache))
+NDTensors.scalartype(bp_cache) = scalartype(tensornetwork(bp_cache))
 
 function default_message(bp_cache::BeliefPropagationCache, edge::PartitionEdge)
   return default_message(bp_cache)(scalartype(bp_cache), linkinds(bp_cache, edge))