Debugging pytensor shape issue

Author: aandorra-mia

pymc_extras/statespace/core/statespace.py

@@ -741,7 +741,7 @@ def _insert_random_variables(self):
 
         replacement_dict = {var: pymc_model[name] for name, var in self._name_to_variable.items()}
         self.subbed_ssm = vectorize_graph(matrices, replace=replacement_dict)
-        for name, matrix in zip(LONG_MATRIX_NAMES, self.subbed_ssm):
+        for name, matrix in zip(MATRIX_NAMES, self.subbed_ssm):
             matrix.name = name
 
     def _insert_data_variables(self):

pymc_extras/statespace/filters/distributions.py

@@ -446,7 +446,6 @@ def step(mu, cov, rng):
 
 @_logprob.register(KalmanFilterRV)
 def sequence_mvnormal_logp(op, values, mus, covs, logp, rng, **kwargs):
-    print(values[0].type.shape, mus.type.shape, covs.type.shape)
     return check_parameters(
         logp,
         pt.eq(values[0].shape[-2], mus.shape[-2]),