From bc6b672584bf43e5b657db63cfeb9d59a2f548d7 Mon Sep 17 00:00:00 2001 From: Johnnie Gray Date: Tue, 28 Jan 2025 16:21:25 -0800 Subject: [PATCH] fix some imports --- quimb/tensor/tensor_2d_tebd.py | 3 ++- quimb/tensor/tensor_arbgeom_tebd.py | 6 ++---- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/quimb/tensor/tensor_2d_tebd.py b/quimb/tensor/tensor_2d_tebd.py index a0a9772a..bc361c2a 100644 --- 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 index 97efd0a0..a783f55a 100644 --- 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