Fixes and XFAILs related to static shape handling

Author: michaelosthege

pymc/tests/gp/test_gp.py

@@ -76,6 +76,7 @@ class TestMarginalApproxSigmaParams(TestSigmaParams):
 
     gp_implementation = pm.gp.MarginalApprox
 
+    @pytest.mark.xfail(reason="Possible shape problem, see #6366")
     def test_catch_warnings(self):
         """Warning from using the old noise parameter."""
         with self.model:

pymc/tests/ode/test_utils.py

@@ -26,7 +26,7 @@ def ode_func(y, t, p):
         return np.exp(-t) - p[0] * y[0]
 
     # Computation of graidients with PyTensor
-    augmented_ode_func = augment_system(ode_func, 1, 1 + 1)
+    augmented_ode_func = augment_system(ode_func, n_states=1, n_theta=1)
 
     # This is the new system, ODE + Sensitivities, which will be integrated
     def augmented_system(Y, t, p):