[Tests] add test for checking soft dependencies. (#3847) · NicolasHug/diffusers@49949f3 · GitHub
Skip to content

Commit 49949f3

Browse files
authored
[Tests] add test for checking soft dependencies. (huggingface#3847)
* add test for checking soft dependencies. * address patrick's comments. * dependency tests should not run twice. * debugging. * up.
1 parent c7469eb commit 49949f3

5 files changed

Lines changed: 76 additions & 1 deletion

File tree

Lines changed: 32 additions & 0 deletions

‎.github/workflows/pr_tests.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ jobs:
8181
if: ${{ matrix.config.framework == 'pytorch_models' }}
8282
run: |
8383
python -m pytest -n 2 --max-worker-restart=0 --dist=loadfile \
84-
-s -v -k "not Flax and not Onnx" \
84+
-s -v -k "not Flax and not Onnx and not Dependency" \
8585
--make-reports=tests_${{ matrix.config.report }} \
8686
tests/models tests/schedulers tests/others
8787

‎setup.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,8 @@
9494
"jaxlib>=0.1.65",
9595
"Jinja2",
9696
"k-diffusion>=0.0.12",
97+
"torchsde",
98+
"note_seq",
9799
"librosa",
98100
"numpy",
99101
"omegaconf",
@@ -106,6 +108,7 @@
106108
"safetensors",
107109
"sentencepiece>=0.1.91,!=0.1.92",
108110
"scipy",
111+
"onnx",
109112
"regex!=2019.12.17",
110113
"requests",
111114
"tensorboard",

‎src/diffusers/dependency_versions_table.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@
1818
"jaxlib": "jaxlib>=0.1.65",
1919
"Jinja2": "Jinja2",
2020
"k-diffusion": "k-diffusion>=0.0.12",
21+
"torchsde": "torchsde",
22+
"note_seq": "note_seq",
2123
"librosa": "librosa",
2224
"numpy": "numpy",
2325
"omegaconf": "omegaconf",
@@ -30,6 +32,7 @@
3032
"safetensors": "safetensors",
3133
"sentencepiece": "sentencepiece>=0.1.91,!=0.1.92",
3234
"scipy": "scipy",
35+
"onnx": "onnx",
3336
"regex": "regex!=2019.12.17",
3437
"requests": "requests",
3538
"tensorboard": "tensorboard",

‎tests/others/test_dependencies.py‎

Lines changed: 37 additions & 0 deletions

0 commit comments

Comments
 (0)