change testGather to MXNet

deepjavalibrary · May 6, 2022 · 98a1de4 · 98a1de4
1 parent b887bc1
commit 98a1de4
Showing 1 changed file with 1 addition and 1 deletion.
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
@@ -55,7 +55,7 @@ public void testPick() {
 
     @Test
     public void testGather() {
-        Engine engine = Engine.getEngine("PyTorch");
+        Engine engine = Engine.getEngine("MXNet");
         try (NDManager manager = engine.newBaseManager()) {
             NDArray arr = manager.arange(20f).reshape(-1, 4);
             long[] idx = {0, 0, 2, 1, 1, 2};