Try fixing tests

ITensor · May 14, 2024 · ae96276 · ae96276
1 parent 2eeb1b8
commit ae96276
Showing 1 changed file with 3 additions and 13 deletions.
diff --git a/src/lib/ITensorMPS/ext/ITensorMPSChainRulesCoreExt/abstractmps.jl b/src/lib/ITensorMPS/ext/ITensorMPSChainRulesCoreExt/abstractmps.jl
@@ -1,18 +1,8 @@
 using Adapt: adapt
 using ChainRulesCore: ChainRulesCore, HasReverseMode, NoTangent, RuleConfig, rrule_via_ad
 using ITensors:
-  ITensors,
-  ITensor,
-  apply,
-  dag,
-  hassameinds,
-  inds,
-  inner,
-  itensor,
-  mapprime,
-  replaceprime,
-  swapprime
-using ITensors.ITensorMPS: MPO, MPS, siteinds
+  ITensors, ITensor, dag, hassameinds, inds, itensor, mapprime, replaceprime, swapprime
+using ITensors.ITensorMPS: ITensorMPS, MPO, MPS, apply, inner, siteinds
 using NDTensors: datatype
 
 function ChainRulesCore.rrule(
@@ -196,7 +186,7 @@ function ChainRulesCore.rrule(
   end
   y = typeof(x)(y_data)
   if !set_limits
-    y = ITensors.set_ortho_lims(y, ortho_lims(x))
+    y = ITensorMPS.set_ortho_lims(y, ortho_lims(x))
   end
   return y, map_pullback
 end