chore: add dependency for example (#1581)

star-whale · Dec 5, 2022 · 8983a8c · 8983a8c
1 parent 4308466
commit 8983a8c
Show file tree

Hide file tree

Showing 10 changed files with 15 additions and 1 deletion.
diff --git a/example/PennFudanPed/requirements.txt b/example/PennFudanPed/requirements.txt
@@ -1,3 +1,4 @@
 Pillow
 torch
+torchvision
 pycocotools
diff --git a/example/cifar10/requirements.txt b/example/cifar10/requirements.txt
@@ -1 +1,3 @@
 Pillow
+torch
+torchvision
diff --git a/example/mnist/Makefile b/example/mnist/Makefile
@@ -9,3 +9,4 @@ prepare-data:
 	[ -f data/t10k-labels-idx1-ubyte ] || ([ -z $(CN) ] && curl -o data/t10k-labels-idx1-ubyte https://starwhale-examples.s3.us-west-1.amazonaws.com/dataset/mnist/t10k-labels-idx1-ubyte) || curl -o data/t10k-labels-idx1-ubyte https://starwhale-examples.oss-cn-beijing.aliyuncs.com/dataset/mnist/t10k-labels-idx1-ubyte
 	[ -d models ] ||  mkdir -p models
 	[ -f models/mnist_cnn.pt ] || ([ -z $(CN) ] && curl -o models/mnist_cnn.pt https://starwhale-examples.s3.us-west-1.amazonaws.com/model/mnist.pth) || curl -o models/mnist_cnn.pt https://starwhale-examples.oss-cn-beijing.aliyuncs.com/model/mnist.pth
+	python3 -m pip install -r requirements.txt
diff --git a/example/mnist/requirements.txt b/example/mnist/requirements.txt
@@ -0,0 +1,2 @@
+torch
+torchvision
diff --git a/example/speech_command/Makefile b/example/speech_command/Makefile
@@ -10,5 +10,6 @@ prepare-data:
 	[ -f data/SpeechCommands/speech_commands_v0.02/validation_list.txt ] || tar -xzf data/speech_commands_v0.02.tar.gz -C data/SpeechCommands/speech_commands_v0.02
 	[ -d models ] ||  mkdir -p models
 	[ -f models/m5.pth ] || ([ -z $(CN) ] && curl -o models/m5.pth https://starwhale-examples.s3.us-west-1.amazonaws.com/model/speech_cmd_m5.pth) || curl -o models/m5.pth https://starwhale-examples.oss-cn-beijing.aliyuncs.com/model/speech_cmd_m5.pth
+	python3 -m pip install -r requirements.txt
 
 
diff --git a/example/speech_command/requirements.txt b/example/speech_command/requirements.txt
@@ -0,0 +1,2 @@
+torch
+torchaudio
diff --git a/example/text_cls_AG_NEWS/Makefile b/example/text_cls_AG_NEWS/Makefile
@@ -17,4 +17,5 @@ prepare-data:
 	[ -d models ] ||  mkdir -p models
 	[ -f models/model.i ] || ([ -z $(CN) ] && curl -o models/model.i https://starwhale-examples.s3.us-west-1.amazonaws.com/model/ag_news/models/model.i) || curl -o models/model.i https://starwhale-examples.oss-cn-beijing.aliyuncs.com/model/ag_news/models/model.i
 	[ -f models/vocab.i ] || ([ -z $(CN) ] && curl -o models/vocab.i https://starwhale-examples.s3.us-west-1.amazonaws.com/model/ag_news/models/vocab.i) || curl -o models/vocab.i https://starwhale-examples.oss-cn-beijing.aliyuncs.com/model/ag_news/models/vocab.i
+	python3 -m pip install -r requirements.txt
 
diff --git a/example/text_cls_AG_NEWS/requirements.txt b/example/text_cls_AG_NEWS/requirements.txt
@@ -0,0 +1,2 @@
+torch
+torchtext
diff --git a/example/ucf101/requirements.txt b/example/ucf101/requirements.txt
@@ -1 +1,3 @@
 torch
+numpy
+opencv-python-headless
diff --git a/scripts/client_test/cli_test.py b/scripts/client_test/cli_test.py
@@ -306,7 +306,7 @@ def test_expl(self, expl_name: str) -> None:
         workdir_ = str(expl["workdir"])
 
         p = subprocess.Popen(
-            ["make", "prepare-e2e-data"],
+            ["make", "prepare-data"],
             stdout=subprocess.PIPE,
             stderr=subprocess.STDOUT,
             cwd=workdir_,