diff --git a/extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/SpTextEmbeddingTest.java b/extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/SpTextEmbeddingTest.java index 8d816f203b8..360c323889f 100644 --- a/extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/SpTextEmbeddingTest.java +++ b/extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/SpTextEmbeddingTest.java @@ -12,7 +12,6 @@ */ package ai.djl.sentencepiece; -import ai.djl.testing.TestRequirements; import ai.djl.training.util.DownloadUtils; import org.testng.Assert; @@ -26,10 +25,9 @@ import java.util.Collections; public class SpTextEmbeddingTest { + @BeforeTest public void downloadModel() throws IOException { - TestRequirements.notArm(); - Path modelFile = Paths.get("build/test/models/sententpiece_test_model.model"); if (Files.notExists(modelFile)) { DownloadUtils.download( diff --git a/extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/SpTokenizerTest.java b/extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/SpTokenizerTest.java index 4821609edd3..92a1b0ab1ea 100644 --- a/extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/SpTokenizerTest.java +++ b/extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/SpTokenizerTest.java @@ -32,8 +32,6 @@ public class SpTokenizerTest { @BeforeTest public void downloadModel() throws IOException { - TestRequirements.notArm(); - Path modelFile = Paths.get("build/test/sp_model/sp_model.model"); if (Files.notExists(modelFile)) { DownloadUtils.download( diff --git a/extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/SpVocabularyTest.java b/extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/SpVocabularyTest.java index f342795391c..a0e565d5473 100644 --- a/extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/SpVocabularyTest.java +++ b/extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/SpVocabularyTest.java @@ -13,7 +13,6 @@ package ai.djl.sentencepiece; -import ai.djl.testing.TestRequirements; import ai.djl.training.util.DownloadUtils; import org.testng.Assert; @@ -29,8 +28,6 @@ public class SpVocabularyTest { @BeforeTest public void downloadModel() throws IOException { - TestRequirements.notArm(); - Path modelFile = Paths.get("build/test/models/sententpiece_test_model.model"); if (Files.notExists(modelFile)) { DownloadUtils.download(