Suppress Ray reinit error

Yu-Group · Feb 9, 2024 · 55fe9b3 · 55fe9b3
1 parent d5f46cf
commit 55fe9b3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py
@@ -15,7 +15,6 @@
 from sklearn.model_selection import train_test_split
 from sklearn.tree import DecisionTreeClassifier, DecisionTreeRegressor
 from sklearn.utils import resample
-
 from vflow import Vset, build_vset, dict_to_df, init_args
 from vflow.subkey import Subkey as sm
 from vflow.vset import PREV_KEY
@@ -422,7 +421,7 @@ def fun2(a, b=1):
         )
 
         np.random.seed(13)
-        ray.init()
+        ray.init(num_cpus=1, ignore_reinit_error=True)
 
         data = data_vset()