dont wrap

Author: williambdean

tests/link/mlx/test_math.py

@@ -1,4 +1,3 @@
-import mlx.core as mx
 import numpy as np
 
 import pytensor
@@ -12,9 +11,12 @@ def test_mlx_dot():
     out = x.dot(y)
     fn = pytensor.function([x, y], out, mode="MLX")
 
-    test_x = mx.array(np.random.normal(size=(3, 2)))
-    test_y = mx.array(np.random.normal(size=(2, 4)))
-    np.testing.assert_allclose(
-        fn(test_x, test_y),
-        np.dot(test_x, test_y),
-    )
+    seed = sum(map(ord, "test_mlx_dot"))
+    rng = np.random.default_rng(seed)
+
+    test_x = rng.normal(size=(3, 2))
+    test_y = rng.normal(size=(2, 4))
+
+    actual = fn(test_x, test_y)
+    expected = np.dot(test_x, test_y)
+    np.testing.assert_allclose(actual, expected)