diff --git a/continuous_integration/environment-3.10.yaml b/continuous_integration/environment-3.10.yaml
index 6ee044f42..a05c88665 100644
--- a/continuous_integration/environment-3.10.yaml
+++ b/continuous_integration/environment-3.10.yaml
@@ -5,6 +5,7 @@ dependencies:
 - c-compiler
 - dask>=2024.4.1
 - dask-expr>=1.0.11
+- docker-py>=7.1.0
 - fastapi>=0.92.0
 - fugue>=0.7.3
 - httpx>=0.24.1
@@ -12,7 +13,7 @@ dependencies:
 - jsonschema
 - lightgbm
 - maturin>=1.3,<1.4
-- mlflow>=2.9
+- mlflow>=2.10
 - mock
 - numpy>=1.22.4
 - pandas>=2
diff --git a/continuous_integration/environment-3.11.yaml b/continuous_integration/environment-3.11.yaml
index 9edaddbbd..653e5b185 100644
--- a/continuous_integration/environment-3.11.yaml
+++ b/continuous_integration/environment-3.11.yaml
@@ -5,6 +5,7 @@ dependencies:
 - c-compiler
 - dask>=2024.4.1
 - dask-expr>=1.0.11
+- docker-py>=7.1.0
 - fastapi>=0.92.0
 - fugue>=0.7.3
 - httpx>=0.24.1
@@ -12,7 +13,7 @@ dependencies:
 - jsonschema
 - lightgbm
 - maturin>=1.3,<1.4
-- mlflow>=2.9
+- mlflow>=2.10
 - mock
 - numpy>=1.22.4
 - pandas>=2
diff --git a/continuous_integration/environment-3.12.yaml b/continuous_integration/environment-3.12.yaml
index 657e18507..9c4f446f0 100644
--- a/continuous_integration/environment-3.12.yaml
+++ b/continuous_integration/environment-3.12.yaml
@@ -5,6 +5,7 @@ dependencies:
 - c-compiler
 - dask>=2024.4.1
 - dask-expr>=1.0.11
+- docker-py>=7.1.0
 - fastapi>=0.92.0
 - fugue>=0.7.3
 - httpx>=0.24.1
@@ -13,7 +14,7 @@ dependencies:
 - lightgbm
 - maturin>=1.3,<1.4
 # TODO: add once mlflow 3.12 builds are available
-# - mlflow>=2.9
+# - mlflow>=2.10
 - mock
 - numpy>=1.22.4
 - pandas>=2
diff --git a/continuous_integration/environment-3.9.yaml b/continuous_integration/environment-3.9.yaml
index 88eee32b5..7b5353c5d 100644
--- a/continuous_integration/environment-3.9.yaml
+++ b/continuous_integration/environment-3.9.yaml
@@ -5,6 +5,7 @@ dependencies:
 - c-compiler
 - dask=2024.4.1
 - dask-expr=1.0.11
+- docker-py>=7.1.0
 - fastapi=0.92.0
 - fugue=0.7.3
 - httpx=0.24.1
@@ -12,7 +13,7 @@ dependencies:
 - jsonschema
 - lightgbm
 - maturin=1.3
-- mlflow=2.9
+- mlflow=2.10
 - mock
 - numpy=1.22.4
 - pandas=2