try fix import ci

volcengine · Dec 6, 2024 · 3c729fd · 3c729fd
1 parent 7aa51c0
commit 3c729fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -42,7 +42,7 @@
     'demo': ['hydra-core', 'transformers', ''],
     'single-controller': ['ray', 'kubernetes'],
     'single-controller-ray': ['ray'],
-    'test': ['fsspec', 'pytest', 'datasets']
+    'test': ['fsspec', 'pytest', 'datasets', 'ray']
 }
 
 from pathlib import Path