fix some imports

jcmgray · Jan 29, 2025 · bc6b672 · bc6b672
1 parent 9292f62
commit bc6b672
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/quimb/tensor/tensor_2d_tebd.py b/quimb/tensor/tensor_2d_tebd.py
@@ -7,7 +7,8 @@
 import scipy.sparse.linalg as spla
 from autoray import conj, dag, do, reshape
 
-from ..utils import default_to_neutral_style, pairwise
+from ..utils import pairwise
+from ..utils_plot import default_to_neutral_style
 from .contraction import contract_strategy
 from .drawing import get_colors
 from .optimize import TNOptimizer

diff --git a/quimb/tensor/tensor_arbgeom_tebd.py b/quimb/tensor/tensor_arbgeom_tebd.py
@@ -10,12 +10,10 @@
 from ..core import eye, kron, qarray
 from ..utils import (
     ExponentialGeometricRollingDiffMean,
-    default_to_neutral_style,
     ensure_dict,
 )
-from ..utils import (
-    progbar as Progbar,
-)
+from ..utils import progbar as Progbar
+from ..utils_plot import default_to_neutral_style
 from .drawing import get_colors, get_positions
 from .tensor_core import Tensor