fix mypy in test_array.py

scalableminds · Feb 15, 2025 · 995a5d5 · 995a5d5
1 parent b4c53a9
commit 995a5d5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_array.py b/tests/test_array.py
@@ -1463,4 +1463,6 @@ async def test_sharding_coordinate_selection() -> None:
         shards=(2, 4, 4),
     )
     arr[:] = np.arange(2 * 3 * 4).reshape((2, 3, 4))
-    assert (arr[1, [0, 1]] == np.array([[12, 13, 14, 15], [16, 17, 18, 19]])).all()  # type: ignore[index]
+    result = arr[1, [0, 1]]
+    assert isinstance(result, NDArrayLike)
+    assert (result == np.array([[12, 13, 14, 15], [16, 17, 18, 19]])).all()  # type: ignore[index]