Skip to content

Commit 84d3581

Browse files
authored
[ci] Fixes build on macOS aarch64 machine (#3191)
1 parent e62674c commit 84d3581

File tree

3 files changed

+7
-3
lines changed

3 files changed

+7
-3
lines changed

examples/src/test/java/ai/djl/examples/inference/TimeSeriesTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ public void testM5Forecasting() throws ModelException, TranslateException, IOExc
5555

5656
@Test
5757
public void testAirPassenger() throws ModelException, TranslateException, IOException {
58-
TestRequirements.engine("MXNet");
58+
TestRequirements.linux();
5959

6060
float[] result = AirPassengersDeepAR.predict();
6161
logger.info("{}", result);

examples/src/test/java/ai/djl/examples/training/TrainResNetTest.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ public void testTrainResNet() throws ModelException, IOException, TranslateExcep
4444
@Test
4545
public void testTrainResNetSymbolicNightly()
4646
throws ModelException, IOException, TranslateException {
47-
TestRequirements.engine("MXNet");
47+
TestRequirements.linux();
4848
TestRequirements.nightly();
4949
TestRequirements.gpu("MXNet");
5050

@@ -65,6 +65,7 @@ public void testTrainResNetSymbolicNightly()
6565
@Test
6666
public void testTrainResNetImperativeNightly()
6767
throws ModelException, IOException, TranslateException {
68+
TestRequirements.linux();
6869
TestRequirements.nightly();
6970
TestRequirements.gpu("MXNet");
7071

integration/build.gradle

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,10 @@ dependencies {
1010
implementation project(":model-zoo")
1111
implementation project(":testing")
1212

13-
runtimeOnly project(":engines:mxnet:mxnet-model-zoo")
13+
// Don't use MXNet for aarch64
14+
if (System.properties['os.arch'] != "aarch64") {
15+
runtimeOnly project(":engines:mxnet:mxnet-model-zoo")
16+
}
1417
runtimeOnly project(":engines:pytorch:pytorch-model-zoo")
1518
runtimeOnly project(":engines:pytorch:pytorch-jni")
1619
runtimeOnly project(":engines:tensorflow:tensorflow-model-zoo")

0 commit comments

Comments
 (0)