Move installation command of timm library to setup.py

huggingface · Jul 25, 2024 · 8e088af · 8e088af
1 parent 0a02cd3
commit 8e088af
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -61,10 +61,10 @@ fast_tests_feature_extraction:
 fast_test_videomae:
 	python -m pip install .[tests]
 	python -m pytest tests/test_video_mae.py
+
 # Run unit and integration tests related to Image segmentation
 fast_tests_object_detection:
 	python -m pip install .[tests]
-	python -m pip install timm
 	python -m pytest tests/test_object_detection.py
 
 # Run single-card non-regression tests

diff --git a/setup.py b/setup.py
@@ -47,6 +47,7 @@
     "safetensors",
     "pytest < 8.0.0",
     "torchsde",
+    "timm",
 ]
 
 QUALITY_REQUIRES = [