Fix VI xfail test

Author: ricardoV94

pymc/tests/variational/test_inference.py

@@ -307,17 +307,19 @@ def test_remove_scan_op():
         buff.close()
 
 
-@pytest.mark.xfail(reason="Broke from static shape handling with Aesara 2.8.8")
 def test_var_replacement():
     X_mean = pm.floatX(np.linspace(0, 10, 10))
     y = pm.floatX(np.random.normal(X_mean * 4, 0.05))
+    inp_size = aesara.shared(np.array(10, dtype="int64"), name="inp_size")
     with pm.Model():
-        inp = pm.Normal("X", X_mean, size=X_mean.shape)
+        inp = pm.Normal("X", X_mean, size=(inp_size,))
         coef = pm.Normal("b", 4.0)
         mean = inp * coef
-        pm.Normal("y", mean, 0.1, observed=y)
+        pm.Normal("y", mean, 0.1, shape=inp.shape, observed=y)
         advi = pm.fit(100)
         assert advi.sample_node(mean).eval().shape == (10,)
+
+        inp_size.set_value(11)
         x_new = pm.floatX(np.linspace(0, 10, 11))
         assert advi.sample_node(mean, more_replacements={inp: x_new}).eval().shape == (11,)