Make model zoo test weekly (deepjavalibrary#1004)

Change-Id: I1c73df17cb077b9ce8905fcc2fc8bbb37b9688d8
AzizZayed · Jun 10, 2021 · 77809f4 · 77809f4
1 parent 0aec8ca
commit 77809f4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/integration/src/main/java/ai/djl/integration/tests/model_zoo/ModelZooTest.java b/integration/src/main/java/ai/djl/integration/tests/model_zoo/ModelZooTest.java
@@ -23,6 +23,7 @@
 import ai.djl.util.Utils;
 import java.io.IOException;
 import java.nio.file.Paths;
+import java.util.Calendar;
 import java.util.List;
 import java.util.ServiceLoader;
 import org.testng.SkipException;
@@ -49,7 +50,10 @@ public void tearDown() {
     @Test
     public void testDownloadModels() throws IOException, ModelException {
         if (!Boolean.getBoolean("nightly") || Boolean.getBoolean("offline")) {
-            throw new SkipException("Nightly only");
+            throw new SkipException("Weekly only");
+        }
+        if (Calendar.SATURDAY == Calendar.getInstance().get(Calendar.DAY_OF_WEEK)) {
+            throw new SkipException("Weekly only");
         }
 
         ServiceLoader<ZooProvider> providers = ServiceLoader.load(ZooProvider.class);