Bump dask-ml dependency (#378)

* Bump dask-ml to 2022.1.22

* Un-xfail affected ML tests

Author: charlesbluca

continuous_integration/environment-3.7-jdk11-dev.yaml

@@ -7,7 +7,7 @@ dependencies:
 - antlr4-python3-runtime>=4.9.2
 - black=19.10b0
 - ciso8601>=2.2.0
-- dask-ml>=2021.11.16
+- dask-ml>=2022.1.22
 - dask>=2021.11.1
 - fastapi>=0.61.1
 - fs>=2.4.11

continuous_integration/environment-3.7-jdk8-dev.yaml

@@ -7,7 +7,7 @@ dependencies:
 - antlr4-python3-runtime>=4.9.2
 - black=19.10b0
 - ciso8601>=2.2.0
-- dask-ml>=2021.11.16
+- dask-ml>=2022.1.22
 - dask>=2021.11.1
 - fastapi>=0.61.1
 - fs>=2.4.11

continuous_integration/environment-3.8-jdk11-dev.yaml

@@ -7,7 +7,7 @@ dependencies:
 - antlr4-python3-runtime>=4.9.2
 - black=19.10b0
 - ciso8601>=2.2.0
-- dask-ml>=2021.11.16
+- dask-ml>=2022.1.22
 - dask>=2021.11.1
 - fastapi>=0.61.1
 - fs>=2.4.11

continuous_integration/environment-3.8-jdk8-dev.yaml

@@ -7,7 +7,7 @@ dependencies:
 - antlr4-python3-runtime>=4.9.2
 - black=19.10b0
 - ciso8601>=2.2.0
-- dask-ml>=2021.11.16
+- dask-ml>=2022.1.22
 - dask>=2021.11.1
 - fastapi>=0.61.1
 - fs>=2.4.11

continuous_integration/environment-3.9-jdk11-dev.yaml

@@ -7,7 +7,7 @@ dependencies:
 - antlr4-python3-runtime>=4.9.2
 - black=19.10b0
 - ciso8601>=2.2.0
-- dask-ml>=2021.11.16
+- dask-ml>=2022.1.22
 - dask>=2021.11.1
 - fastapi>=0.61.1
 - fs>=2.4.11

continuous_integration/environment-3.9-jdk8-dev.yaml

@@ -7,7 +7,7 @@ dependencies:
 - antlr4-python3-runtime>=4.9.2
 - black=19.10b0
 - ciso8601>=2.2.0
-- dask-ml>=2021.11.16
+- dask-ml>=2022.1.22
 - dask>=2021.11.1
 - fastapi>=0.61.1
 - fs>=2.4.11

docker/conda.txt

@@ -16,7 +16,7 @@ uvicorn>=0.11.3
 pyarrow>=0.15.1
 prompt_toolkit>=3.0.8
 pygments>=2.7.1
-dask-ml>=2021.11.16
+dask-ml>=2022.1.22
 scikit-learn>=0.24.2
 intake>=0.6.0
 pre-commit>=2.11.1

docker/main.dockerfile

@@ -16,7 +16,7 @@ RUN conda config --add channels conda-forge \
     "pyarrow>=0.15.1" \
     "prompt_toolkit>=3.0.8" \
     "pygments>=2.7.1" \
-    "dask-ml>=2021.11.16" \
+    "dask-ml>=2022.1.22" \
     "scikit-learn>=0.24.2" \
     "intake>=0.6.0" \
     && conda clean -ay

setup.py

@@ -94,7 +94,7 @@ def run(self):
             "mock>=4.0.3",
             "sphinx>=3.2.1",
             "pyarrow>=0.15.1",
-            "dask-ml>=2021.11.16",
+            "dask-ml>=2022.1.22",
             "scikit-learn>=0.24.2",
             "intake>=0.6.0",
             "pre-commit",

tests/integration/test_model.py

@@ -165,9 +165,6 @@ def test_clustering_and_prediction(c, training_df):
     check_trained_model(c)
 
 
-@pytest.mark.xfail(
-    reason="Upstream breakage in dask needs to be fixed by next dask-ml release"
-)
 def test_iterative_and_prediction(c, training_df):
     c.sql(
         """
@@ -187,9 +184,6 @@ def test_iterative_and_prediction(c, training_df):
     check_trained_model(c)
 
 
-@pytest.mark.xfail(
-    reason="Upstream breakage in dask needs to be fixed by next dask-ml release"
-)
 def test_show_models(c, training_df):
     c.sql(
         """