[numpy] Fix test failures under NumPy 2.0.

PiperOrigin-RevId: 663919481
google · Aug 17, 2024 · fc69a03 · fc69a03
1 parent 890052d
commit fc69a03
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/trax/tf_numpy/examples/mnist/train_test.py b/trax/tf_numpy/examples/mnist/train_test.py
@@ -46,8 +46,8 @@ def testRuns(self):
 def fake_mnist_data():
 
   def gen_examples(num_examples):
-    x = np.array(
-        np.random.randn(num_examples, 784), copy=False, dtype=np.float32)
+    x = np.asarray(
+        np.random.randn(num_examples, 784), dtype=np.float32)
     y = np.zeros((num_examples, 10), dtype=np.float32)
     y[:][0] = 1.
     return (x, y)