Fixing pre-commit, jax test failing

Author: larryshamalama

pymc/distributions/continuous.py

@@ -3689,13 +3689,8 @@ def dist(cls, x_points, pdf_points, *args, **kwargs):
     def get_moment(rv, size, x_points, pdf_points, cdf_points):
         # cdf_points argument is unused
         # moment = at.as_tensor(0.)
-        print(x_points)
-        print("")
-        for pdf in pdf_points.eval():
-            print(pdf)
-        print("")
         moment = at.sum(at.mul(x_points, pdf_points))
-        print(moment.eval())
+
         if not rv_size_is_none(size):
             moment = at.full(size, moment)
 

pymc/initial_point.py

@@ -272,7 +272,6 @@ def make_initial_point_expression(
         if isinstance(strategy, str):
             if strategy == "moment":
                 value = get_moment(variable)
-                print(value.eval())
             elif strategy == "prior":
                 value = variable
             else:

pymc/tests/test_distributions_moments.py

@@ -805,8 +805,18 @@ def test_categorical_moment(p, size, expected):
     "x_points, pdf_points, size, expected",
     [
         (np.array([-1, 1]), np.array([0.4, 0.6]), None, 0.2),
-        (np.array([-4, -1, 3, 9, 19]), np.array([0.1 , 0.15, 0.2 , 0.25, 0.3]), None, 1.5458937198067635),
-        (np.array([-22, -4, 0, 8, 13]), np.tile(1 / 5, 5), (5, 3), np.full((5, 3), -0.14285714285714296)),
+        (
+            np.array([-4, -1, 3, 9, 19]),
+            np.array([0.1, 0.15, 0.2, 0.25, 0.3]),
+            None,
+            1.5458937198067635,
+        ),
+        (
+            np.array([-22, -4, 0, 8, 13]),
+            np.tile(1 / 5, 5),
+            (5, 3),
+            np.full((5, 3), -0.14285714285714296),
+        ),
         (
             np.arange(-100, 10),
             np.arange(1, 111) / 6105,