fix imports

strongio · Jan 10, 2025 · 7bf7156 · 7bf7156
1 parent 0e09515
commit 7bf7156
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/torchcast/kalman_filter/binomial_filter.py b/torchcast/kalman_filter/binomial_filter.py
@@ -10,11 +10,11 @@
 
 from torchcast.covariance import Covariance
 from torchcast.covariance.util import mini_cov_mask
-from torchcast.internals.utils import identity
+from torchcast.internals.utils import identity, class_or_instancemethod
 from torchcast.kalman_filter import KalmanFilter
 from torchcast.kalman_filter.ekf import EKFStep, EKFPredictions
 from torchcast.process import Process
-from torchcast.utils import conf2bounds, class_or_instancemethod
+from torchcast.state_space.predictions import conf2bounds
 
 if TYPE_CHECKING:
     from pandas import DataFrame

diff --git a/torchcast/kalman_filter/ekf.py b/torchcast/kalman_filter/ekf.py
@@ -6,8 +6,9 @@
 from torch import Tensor
 
 from .kalman_filter import KalmanStep
+from ..internals.utils import class_or_instancemethod
 from ..state_space import Predictions
-from ..utils import TimeSeriesDataset, class_or_instancemethod
+from ..utils import TimeSeriesDataset
 
 
 class EKFStep(KalmanStep):