Fixed distribution test import

pymc-devs · Dec 11, 2024 · 9ecc046 · 9ecc046
1 parent 62a8c0e
commit 9ecc046
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/statespace/test_distributions.py b/tests/statespace/test_distributions.py
@@ -11,12 +11,17 @@
 from pymc_extras.statespace.filters.distributions import (
     LinearGaussianStateSpace,
     LinearGaussianStateSpaceRV,
+    SequenceMvNormal,
+    _LinearGaussianStateSpace,
 )
 from pymc_extras.statespace.utils.constants import (
+    ALL_STATE_DIM,
     JITTER_DEFAULT,
     LONG_MATRIX_NAMES,
     MISSING_FILL,
+    OBS_STATE_DIM,
     SHORT_NAME_TO_LONG,
+    TIME_DIM,
 )
 from tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
     rng,