test fix

deepjavalibrary · Jun 29, 2022 · f55102f · f55102f
1 parent a9f3f2d
commit f55102f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/integration/src/main/java/ai/djl/integration/tests/ndarray/NDIndexTest.java b/integration/src/main/java/ai/djl/integration/tests/ndarray/NDIndexTest.java
@@ -159,6 +159,7 @@ public void testSetArray() {
             Assert.assertEquals(original, expected);
 
             // set by index array
+            original = manager.arange(1, 10).reshape(3, 3);
             NDArray index = manager.create(new long[] {0, 1}, new Shape(2));
             value = manager.create(new int[] {666, 777, 888, 999}, new Shape(2, 2));
             original.set(new NDIndex("{}, :{}", index, 2), value);