Fix various failures due to missing imports

Author: fonnesbeck

pymc_extras/model/transforms/autoreparam.py

@@ -24,7 +24,6 @@
 from pymc.pytensorf import toposort_replace
 from pytensor.graph.basic import Apply, Variable
 from pytensor.tensor.random.op import RandomVariable
-from pymc_extras.model.transforms.autoreparam import vip_reparametrize
 
 _log = logging.getLogger("pmx")
 

pymc_extras/statespace/core/representation.py

@@ -5,10 +5,8 @@
 import pytensor.tensor as pt
 
 from pymc_extras.statespace.utils.constants import (
-    JITTER_DEFAULT,
-    LONG_MATRIX_NAMES,
-    MISSING_FILL,
-    SHORT_NAME_TO_LONG,
+    NEVER_TIME_VARYING,
+    VECTOR_VALUED,
 )
 
 floatX = pytensor.config.floatX

pymc_extras/statespace/filters/kalman_filter.py

@@ -12,12 +12,8 @@
 from pytensor.tensor.slinalg import solve_triangular
 
 from pymc_extras.statespace.filters.utilities import (
-    compute_forecast_error_cov,
-    compute_kalman_gain,
-    compute_predicted_state,
-    compute_predicted_state_cov,
-    compute_updated_state,
-    compute_updated_state_cov,
+    quad_form_sym,
+    split_vars_into_seq_and_nonseq,
     stabilize,
 )
 from pymc_extras.statespace.utils.constants import JITTER_DEFAULT, MISSING_FILL

pymc_extras/statespace/filters/kalman_smoother.py

@@ -5,8 +5,8 @@
 from pytensor.tensor.nlinalg import matrix_dot
 
 from pymc_extras.statespace.filters.utilities import (
-    compute_predicted_state,
-    compute_predicted_state_cov,
+    quad_form_sym,
+    split_vars_into_seq_and_nonseq,
     stabilize,
 )
 from pymc_extras.statespace.utils.constants import JITTER_DEFAULT