diff --git a/examples/src/test/java/ai/djl/examples/inference/TimeSeriesTest.java b/examples/src/test/java/ai/djl/examples/inference/TimeSeriesTest.java index cd37ef957c0..614a43560bb 100644 --- a/examples/src/test/java/ai/djl/examples/inference/TimeSeriesTest.java +++ b/examples/src/test/java/ai/djl/examples/inference/TimeSeriesTest.java @@ -55,7 +55,7 @@ public void testM5Forecasting() throws ModelException, TranslateException, IOExc @Test public void testAirPassenger() throws ModelException, TranslateException, IOException { - TestRequirements.engine("MXNet"); + TestRequirements.linux(); float[] result = AirPassengersDeepAR.predict(); logger.info("{}", result); diff --git a/examples/src/test/java/ai/djl/examples/training/TrainResNetTest.java b/examples/src/test/java/ai/djl/examples/training/TrainResNetTest.java index b55bc78f9de..f33be26de90 100644 --- a/examples/src/test/java/ai/djl/examples/training/TrainResNetTest.java +++ b/examples/src/test/java/ai/djl/examples/training/TrainResNetTest.java @@ -44,7 +44,7 @@ public void testTrainResNet() throws ModelException, IOException, TranslateExcep @Test public void testTrainResNetSymbolicNightly() throws ModelException, IOException, TranslateException { - TestRequirements.engine("MXNet"); + TestRequirements.linux(); TestRequirements.nightly(); TestRequirements.gpu("MXNet"); @@ -65,6 +65,7 @@ public void testTrainResNetSymbolicNightly() @Test public void testTrainResNetImperativeNightly() throws ModelException, IOException, TranslateException { + TestRequirements.linux(); TestRequirements.nightly(); TestRequirements.gpu("MXNet"); diff --git a/integration/build.gradle b/integration/build.gradle index 5417aa36a5c..f7003fb9f03 100644 --- a/integration/build.gradle +++ b/integration/build.gradle @@ -10,7 +10,10 @@ dependencies { implementation project(":model-zoo") implementation project(":testing") - runtimeOnly project(":engines:mxnet:mxnet-model-zoo") + // Don't use MXNet for aarch64 + if (System.properties['os.arch'] != "aarch64") { + runtimeOnly project(":engines:mxnet:mxnet-model-zoo") + } runtimeOnly project(":engines:pytorch:pytorch-model-zoo") runtimeOnly project(":engines:pytorch:pytorch-jni") runtimeOnly project(":engines:tensorflow:tensorflow-model-zoo")