Skip to content

Commit

Permalink
Fix windows continuous build issus
Browse files Browse the repository at this point in the history
  • Loading branch information
KexinFeng committed May 9, 2022
1 parent a720822 commit 512120c
Showing 1 changed file with 1 addition and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
*/
package ai.djl.integration.tests.ndarray;

import ai.djl.engine.Engine;
import ai.djl.ndarray.NDArray;
import ai.djl.ndarray.NDManager;
import ai.djl.ndarray.index.NDIndex;
Expand Down Expand Up @@ -55,11 +56,6 @@ public void testPick() {

@Test
public void testGather() {
// Currently in windows gradle cannot find all the engines to fill in -classpath, except for
// MXNet.
// In the dependencies, changing runtimeOnly to api however will remedy the problem.
// TODO: remove this when gradle problem is fixed.
TestRequirements.notWindows();
try (NDManager manager = NDManager.newBaseManager()) {
NDArray arr = manager.arange(20f).reshape(-1, 4);
NDArray index = manager.create(new long[] {0, 0, 2, 1, 1, 2}, new Shape(3, 2));
Expand All @@ -71,8 +67,6 @@ public void testGather() {

@Test
public void testTake() {
// TODO: remove this when gradle problem in windows shown above is fixed.
TestRequirements.notWindows();
try (NDManager manager = NDManager.newBaseManager()) {
NDArray arr = manager.arange(1, 7f).reshape(-1, 3);
NDArray index = manager.create(new long[] {0, 4, 1, 2}, new Shape(2, 2));
Expand Down

0 comments on commit 512120c

Please sign in to comment.