diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py index aa8669d..4268d9c 100644 --- 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()