From 98a1de437ccb8b6119014af41691f37ec3d79334 Mon Sep 17 00:00:00 2001
From: Kexin <fengx463@umn.edu>
Date: Thu, 5 May 2022 21:58:15 -0700
Subject: [PATCH] change testGather to MXNet

---
 .../main/java/ai/djl/integration/tests/ndarray/NDIndexTest.java | 2 +-
 1 file changed, 1 insertion(+), 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
index fdb7a6d833a..642bad98183 100644
--- 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};