NumPy 2.0 (commonly bundled with Python 3.11+) renamed the newshape parameter to shape in np.reshape(). Using positional arguments instead of keyword arguments works across all NumPy versions

Author: polyaxon-ci

traceml/tests/test_events_processing/test_event_values.py

@@ -185,8 +185,8 @@ def test_prepare_video(self):
         video_before = np.random.random((4, 10, 3, 20, 20))
         video_after = prepare_video(np.copy(video_before))
         video_before = np.swapaxes(video_before, 0, 1)
-        video_before = np.reshape(video_before, newshape=(10, -1))
-        video_after = np.reshape(video_after, newshape=(10, -1))
+        video_before = np.reshape(video_before, (10, -1))
+        video_after = np.reshape(video_after, (10, -1))
         np.testing.assert_array_almost_equal(
             np.sum(video_before, axis=1), np.sum(video_after, axis=1)
         )

traceml/traceml/processors/events_processors/events_video_processors.py

@@ -123,6 +123,6 @@ def is_power2(num):
     n_rows = 2 ** ((b.bit_length() - 1) // 2)
     n_cols = data.shape[0] // n_rows
 
-    data = np.reshape(data, newshape=(n_rows, n_cols, t, c, h, w))
+    data = np.reshape(data, (n_rows, n_cols, t, c, h, w))
     data = np.transpose(data, axes=(2, 0, 4, 1, 5, 3))
-    return np.reshape(data, newshape=(t, n_rows * h, n_cols * w, c))
+    return np.reshape(data, (t, n_rows * h, n_cols * w, c))