diff --git a/setup.py b/setup.py index 3f8d90e08..dddba08f8 100644 --- a/setup.py +++ b/setup.py @@ -47,6 +47,7 @@ setuptools.setup( "spark": [ "pyspark>=3.2.0", "joblibspark>=0.5.0", + "joblib<1.3.0", # temp solution for joblib 1.3.0 issue, no need once https://github.com/joblib/joblib-spark/pull/48 is merged ], "test": [ "lightgbm>=2.3.1", @@ -87,6 +88,7 @@ setuptools.setup( "pydantic", "sympy", "wolframalpha", + "joblib<1.3.0", # temp solution for joblib 1.3.0 issue, no need once https://github.com/joblib/joblib-spark/pull/48 is merged ], "catboost": ["catboost>=0.26"], "blendsearch": ["optuna==2.8.0"], @@ -140,6 +142,7 @@ setuptools.setup( "joblibspark>=0.5.0", "optuna==2.8.0", "pyspark>=3.2.0", + "joblib<1.3.0", # temp solution for joblib 1.3.0 issue, no need once https://github.com/joblib/joblib-spark/pull/48 is merged ], "autozero": ["scikit-learn", "pandas", "packaging"], },