diff --git a/tests/notebooks/e2e-wine/requirements.in b/tests/notebooks/e2e-wine/requirements.in
index 7ec99c2..4c6a2ee 100644
--- a/tests/notebooks/e2e-wine/requirements.in
+++ b/tests/notebooks/e2e-wine/requirements.in
@@ -5,7 +5,7 @@ kfp==1.8.22
 # pin to <7.0 to avoid breaking changes in sdk
 minio<7.0
 # pin the client to match the version of the deployed MLflow server
-mlflow==2.1.1
+mlflow==2.7.1
 numpy
 pyarrow
 requests
diff --git a/tests/notebooks/e2e-wine/requirements.txt b/tests/notebooks/e2e-wine/requirements.txt
index 32d9ca5..4359433 100644
--- a/tests/notebooks/e2e-wine/requirements.txt
+++ b/tests/notebooks/e2e-wine/requirements.txt
@@ -41,7 +41,6 @@ cloudpickle==2.2.1
     # via
     #   kfp
     #   mlflow
-    #   shap
 configparser==6.0.0
     # via minio
 contourpy==1.1.1
@@ -113,7 +112,6 @@ importlib-metadata==5.2.0
     #   flask
     #   markdown
     #   mlflow
-    #   numba
 importlib-resources==6.0.1
     # via
     #   alembic
@@ -146,8 +144,6 @@ kiwisolver==1.4.5
     # via matplotlib
 kubernetes==25.3.0
     # via kfp
-llvmlite==0.40.1
-    # via numba
 mako==1.2.4
     # via alembic
 markdown==3.4.4
@@ -161,22 +157,18 @@ matplotlib==3.7.3
     # via mlflow
 minio==6.0.2
     # via -r requirements.in
-mlflow==2.1.1
+mlflow==2.7.1
     # via -r requirements.in
-numba==0.57.1
-    # via shap
 numpy==1.24.4
     # via
     #   -r requirements.in
     #   contourpy
     #   matplotlib
     #   mlflow
-    #   numba
     #   pandas
     #   pyarrow
     #   scikit-learn
     #   scipy
-    #   shap
 oauthlib==3.2.2
     # via
     #   databricks-cli
@@ -186,11 +178,8 @@ packaging==22.0
     #   docker
     #   matplotlib
     #   mlflow
-    #   shap
 pandas==1.5.3
-    # via
-    #   mlflow
-    #   shap
+    # via mlflow
 pillow==10.0.1
     # via matplotlib
 pkgutil-resolve-name==1.3.10
@@ -271,14 +260,10 @@ scikit-learn==1.1.3
     # via
     #   -r requirements.in
     #   mlflow
-    #   shap
 scipy==1.10.1
     # via
     #   mlflow
     #   scikit-learn
-    #   shap
-shap==0.42.1
-    # via mlflow
 six==1.16.0
     # via
     #   databricks-cli
@@ -288,8 +273,6 @@ six==1.16.0
     #   kubernetes
     #   python-dateutil
     #   querystring-parser
-slicer==0.0.7
-    # via shap
 smmap==5.0.1
     # via gitdb
 sqlalchemy==1.4.49
@@ -310,8 +293,6 @@ termcolor==2.3.0
     # via fire
 threadpoolctl==3.2.0
     # via scikit-learn
-tqdm==4.66.1
-    # via shap
 typer==0.9.0
     # via kfp
 typing-extensions==4.8.0
diff --git a/tests/notebooks/mlflow/requirements.in b/tests/notebooks/mlflow/requirements.in
index 946ff3d..f5a643c 100644
--- a/tests/notebooks/mlflow/requirements.in
+++ b/tests/notebooks/mlflow/requirements.in
@@ -2,5 +2,5 @@ boto3
 # pin to <7.0 to avoid breaking changes in sdk
 minio<7.0
 # pin the client to match the version of the deployed MLflow server
-mlflow==2.1.1
+mlflow==2.7.1
 tenacity
diff --git a/tests/notebooks/mlflow/requirements.txt b/tests/notebooks/mlflow/requirements.txt
index 97a7a72..c145b85 100644
--- a/tests/notebooks/mlflow/requirements.txt
+++ b/tests/notebooks/mlflow/requirements.txt
@@ -26,9 +26,7 @@ click==8.1.7
     #   flask
     #   mlflow
 cloudpickle==2.2.1
-    # via
-    #   mlflow
-    #   shap
+    # via mlflow
 configparser==6.0.0
     # via minio
 contourpy==1.1.1
@@ -61,7 +59,6 @@ importlib-metadata==5.2.0
     #   flask
     #   markdown
     #   mlflow
-    #   numba
 importlib-resources==6.0.1
     # via
     #   alembic
@@ -80,8 +77,6 @@ joblib==1.3.2
     # via scikit-learn
 kiwisolver==1.4.5
     # via matplotlib
-llvmlite==0.40.1
-    # via numba
 mako==1.2.4
     # via alembic
 markdown==3.4.4
@@ -95,21 +90,17 @@ matplotlib==3.7.3
     # via mlflow
 minio==6.0.2
     # via -r requirements.in
-mlflow==2.1.1
+mlflow==2.7.1
     # via -r requirements.in
-numba==0.57.1
-    # via shap
 numpy==1.24.4
     # via
     #   contourpy
     #   matplotlib
     #   mlflow
-    #   numba
     #   pandas
     #   pyarrow
     #   scikit-learn
     #   scipy
-    #   shap
 oauthlib==3.2.2
     # via databricks-cli
 packaging==22.0
@@ -117,11 +108,8 @@ packaging==22.0
     #   docker
     #   matplotlib
     #   mlflow
-    #   shap
 pandas==1.5.3
-    # via
-    #   mlflow
-    #   shap
+    # via mlflow
 pillow==10.0.1
     # via matplotlib
 protobuf==4.24.3
@@ -155,23 +143,16 @@ requests==2.31.0
 s3transfer==0.6.2
     # via boto3
 scikit-learn==1.3.0
-    # via
-    #   mlflow
-    #   shap
+    # via mlflow
 scipy==1.10.1
     # via
     #   mlflow
     #   scikit-learn
-    #   shap
-shap==0.42.1
-    # via mlflow
 six==1.16.0
     # via
     #   databricks-cli
     #   python-dateutil
     #   querystring-parser
-slicer==0.0.7
-    # via shap
 smmap==5.0.1
     # via gitdb
 sqlalchemy==1.4.49
@@ -186,8 +167,6 @@ tenacity==8.2.3
     # via -r requirements.in
 threadpoolctl==3.2.0
     # via scikit-learn
-tqdm==4.66.1
-    # via shap
 typing-extensions==4.8.0
     # via alembic
 urllib3==1.26.16