diff --git a/tests/notebooks/e2e-wine/requirements.in b/tests/notebooks/e2e-wine/requirements.in
index 7ec99c2..ee1679b 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.11.3
 numpy
 pyarrow
 requests
diff --git a/tests/notebooks/e2e-wine/requirements.txt b/tests/notebooks/e2e-wine/requirements.txt
index 32d9ca5..167de73 100644
--- a/tests/notebooks/e2e-wine/requirements.txt
+++ b/tests/notebooks/e2e-wine/requirements.txt
@@ -8,6 +8,8 @@ absl-py==1.4.0
     # via kfp
 alembic==1.12.0
     # via mlflow
+aniso8601==9.0.1
+    # via graphene
 attrs==23.1.0
     # via
     #   jsonschema
@@ -32,7 +34,6 @@ charset-normalizer==3.2.0
     # via requests
 click==8.1.7
     # via
-    #   databricks-cli
     #   flask
     #   kfp
     #   mlflow
@@ -41,15 +42,12 @@ cloudpickle==2.2.1
     # via
     #   kfp
     #   mlflow
-    #   shap
 configparser==6.0.0
     # via minio
 contourpy==1.1.1
     # via matplotlib
 cycler==0.11.0
     # via matplotlib
-databricks-cli==0.17.7
-    # via mlflow
 deprecated==1.2.14
     # via kfp
 docker==6.1.3
@@ -97,6 +95,14 @@ google-resumable-media==2.6.0
     # via google-cloud-storage
 googleapis-common-protos==1.60.0
     # via google-api-core
+graphene==3.3
+    # via mlflow
+graphql-core==3.2.3
+    # via
+    #   graphene
+    #   graphql-relay
+graphql-relay==3.2.0
+    # via graphene
 greenlet==2.0.2
     # via sqlalchemy
 gunicorn==20.1.0
@@ -113,7 +119,6 @@ importlib-metadata==5.2.0
     #   flask
     #   markdown
     #   mlflow
-    #   numba
 importlib-resources==6.0.1
     # via
     #   alembic
@@ -146,8 +151,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,36 +164,27 @@ matplotlib==3.7.3
     # via mlflow
 minio==6.0.2
     # via -r requirements.in
-mlflow==2.1.1
+mlflow==2.11.3
     # 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
-    #   requests-oauthlib
+    # via requests-oauthlib
 packaging==22.0
     # via
     #   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
@@ -214,8 +208,6 @@ pyasn1-modules==0.3.0
     # via google-auth
 pydantic==1.10.12
     # via kfp
-pyjwt==2.8.0
-    # via databricks-cli
 pyparsing==3.1.1
     # via
     #   httplib2
@@ -247,7 +239,6 @@ referencing==0.30.2
 requests==2.31.0
     # via
     #   -r requirements.in
-    #   databricks-cli
     #   docker
     #   google-api-core
     #   google-cloud-storage
@@ -271,25 +262,18 @@ 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
     #   fire
     #   google-api-python-client
     #   kfp-server-api
     #   kubernetes
     #   python-dateutil
     #   querystring-parser
-slicer==0.0.7
-    # via shap
 smmap==5.0.1
     # via gitdb
 sqlalchemy==1.4.49
@@ -301,17 +285,13 @@ sqlparse==0.4.4
 strip-hints==0.1.10
     # via kfp
 tabulate==0.9.0
-    # via
-    #   databricks-cli
-    #   kfp
+    # via kfp
 tenacity==8.2.3
     # via -r requirements.in
 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
@@ -327,7 +307,6 @@ uritemplate==3.0.1
 urllib3==1.26.16
     # via
     #   botocore
-    #   databricks-cli
     #   docker
     #   google-auth
     #   kfp
diff --git a/tests/notebooks/mlflow-kserve/requirements.in b/tests/notebooks/mlflow-kserve/requirements.in
index 83046f5..ea7c0b8 100644
--- a/tests/notebooks/mlflow-kserve/requirements.in
+++ b/tests/notebooks/mlflow-kserve/requirements.in
@@ -2,7 +2,7 @@ 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.11.3
 numpy
 pyarrow
 requests
diff --git a/tests/notebooks/mlflow-kserve/requirements.txt b/tests/notebooks/mlflow-kserve/requirements.txt
index bc8e22b..13ce966 100644
--- a/tests/notebooks/mlflow-kserve/requirements.txt
+++ b/tests/notebooks/mlflow-kserve/requirements.txt
@@ -18,6 +18,8 @@ aiosignal==1.3.1
     #   ray
 alembic==1.12.0
     # via mlflow
+aniso8601==9.0.1
+    # via graphene
 anyio==4.0.0
     # via
     #   httpcore
@@ -56,7 +58,6 @@ charset-normalizer==3.2.0
     #   requests
 click==8.1.7
     # via
-    #   databricks-cli
     #   flask
     #   mlflow
     #   ray
@@ -64,9 +65,7 @@ click==8.1.7
 cloudevents==1.10.0
     # via kserve
 cloudpickle==2.2.1
-    # via
-    #   mlflow
-    #   shap
+    # via mlflow
 colorful==0.5.5
     # via ray
 configparser==6.0.0
@@ -77,8 +76,6 @@ cuda-python==12.3.0
     # via tritonclient
 cycler==0.11.0
     # via matplotlib
-databricks-cli==0.17.7
-    # via mlflow
 deprecation==2.1.0
     # via cloudevents
 distlib==0.3.7
@@ -120,6 +117,14 @@ googleapis-common-protos==1.60.0
     # via google-api-core
 gpustat==1.1.1
     # via ray
+graphene==3.3
+    # via mlflow
+graphql-core==3.2.3
+    # via
+    #   graphene
+    #   graphql-relay
+graphql-relay==3.2.0
+    # via graphene
 greenlet==2.0.2
     # via sqlalchemy
 grpcio==1.51.3
@@ -150,7 +155,6 @@ importlib-metadata==5.2.0
     #   flask
     #   markdown
     #   mlflow
-    #   numba
 importlib-resources==6.0.1
     # via
     #   alembic
@@ -181,8 +185,6 @@ kubernetes==25.3.0
     # via
     #   -r requirements.in
     #   kserve
-llvmlite==0.40.1
-    # via numba
 mako==1.2.4
     # via alembic
 markdown==3.4.4
@@ -196,7 +198,7 @@ matplotlib==3.7.3
     # via mlflow
 minio==6.0.2
     # via -r requirements.in
-mlflow==2.1.1
+mlflow==2.11.3
     # via -r requirements.in
 msgpack==1.0.7
     # via ray
@@ -204,8 +206,6 @@ multidict==6.0.4
     # via
     #   aiohttp
     #   yarl
-numba==0.57.1
-    # via shap
 numpy==1.24.4
     # via
     #   -r requirements.in
@@ -213,20 +213,16 @@ numpy==1.24.4
     #   kserve
     #   matplotlib
     #   mlflow
-    #   numba
     #   pandas
     #   pyarrow
     #   ray
     #   scikit-learn
     #   scipy
-    #   shap
     #   tritonclient
 nvidia-ml-py==12.535.108
     # via gpustat
 oauthlib==3.2.2
-    # via
-    #   databricks-cli
-    #   requests-oauthlib
+    # via requests-oauthlib
 opencensus==0.11.3
     # via ray
 opencensus-context==0.1.3
@@ -239,12 +235,10 @@ packaging==22.0
     #   docker
     #   matplotlib
     #   mlflow
-    #   shap
 pandas==1.5.3
     # via
     #   kserve
     #   mlflow
-    #   shap
 pillow==10.0.1
     # via matplotlib
 pkgutil-resolve-name==1.3.10
@@ -258,6 +252,7 @@ prometheus-client==0.13.1
 protobuf==3.20.3
     # via
     #   google-api-core
+    #   googleapis-common-protos
     #   kserve
     #   mlflow
     #   ray
@@ -281,8 +276,6 @@ pydantic==1.10.12
     # via
     #   fastapi
     #   ray
-pyjwt==2.8.0
-    # via databricks-cli
 pyparsing==3.1.1
     # via matplotlib
 python-dateutil==2.8.2
@@ -319,7 +312,6 @@ referencing==0.30.2
 requests==2.31.0
     # via
     #   -r requirements.in
-    #   databricks-cli
     #   docker
     #   google-api-core
     #   kubernetes
@@ -342,24 +334,17 @@ 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
     #   blessed
-    #   databricks-cli
     #   kserve
     #   kubernetes
     #   python-dateutil
     #   querystring-parser
-slicer==0.0.7
-    # via shap
 smart-open==6.4.0
     # via ray
 smmap==5.0.1
@@ -380,17 +365,13 @@ starlette==0.27.0
     #   fastapi
     #   ray
 tabulate==0.9.0
-    # via
-    #   databricks-cli
-    #   kserve
+    # via kserve
 tenacity==8.2.3
     # via -r requirements.in
 threadpoolctl==3.2.0
     # via scikit-learn
 timing-asgi==0.3.1
     # via kserve
-tqdm==4.66.1
-    # via shap
 tritonclient==2.38.0
     # via kserve
 typing-extensions==4.8.0
@@ -401,7 +382,6 @@ typing-extensions==4.8.0
 urllib3==1.26.16
     # via
     #   botocore
-    #   databricks-cli
     #   docker
     #   google-auth
     #   kubernetes
diff --git a/tests/notebooks/mlflow/requirements.in b/tests/notebooks/mlflow/requirements.in
index ac1a3be..38d7985 100644
--- a/tests/notebooks/mlflow/requirements.in
+++ b/tests/notebooks/mlflow/requirements.in
@@ -2,7 +2,7 @@ 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.11.3
 # pin to ensure compatibility with the installed mlflow client
 scikit-learn<1.2
 tenacity
diff --git a/tests/notebooks/mlflow/requirements.txt b/tests/notebooks/mlflow/requirements.txt
index a70e820..551b2eb 100644
--- a/tests/notebooks/mlflow/requirements.txt
+++ b/tests/notebooks/mlflow/requirements.txt
@@ -6,6 +6,8 @@
 #
 alembic==1.12.0
     # via mlflow
+aniso8601==9.0.1
+    # via graphene
 blinker==1.6.2
     # via flask
 boto3==1.28.50
@@ -22,21 +24,16 @@ charset-normalizer==3.2.0
     # via requests
 click==8.1.7
     # via
-    #   databricks-cli
     #   flask
     #   mlflow
 cloudpickle==2.2.1
-    # via
-    #   mlflow
-    #   shap
+    # via mlflow
 configparser==6.0.0
     # via minio
 contourpy==1.1.1
     # via matplotlib
 cycler==0.11.0
     # via matplotlib
-databricks-cli==0.17.7
-    # via mlflow
 docker==6.1.3
     # via mlflow
 entrypoints==0.4
@@ -49,6 +46,14 @@ gitdb==4.0.10
     # via gitpython
 gitpython==3.1.36
     # via mlflow
+graphene==3.3
+    # via mlflow
+graphql-core==3.2.3
+    # via
+    #   graphene
+    #   graphql-relay
+graphql-relay==3.2.0
+    # via graphene
 greenlet==2.0.2
     # via sqlalchemy
 gunicorn==20.1.0
@@ -61,7 +66,6 @@ importlib-metadata==5.2.0
     #   flask
     #   markdown
     #   mlflow
-    #   numba
 importlib-resources==6.0.1
     # via
     #   alembic
@@ -80,8 +84,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,41 +97,30 @@ matplotlib==3.7.3
     # via mlflow
 minio==6.0.2
     # via -r requirements.in
-mlflow==2.1.1
+mlflow==2.11.3
     # 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
     # via
     #   docker
     #   matplotlib
     #   mlflow
-    #   shap
 pandas==1.5.3
-    # via
-    #   mlflow
-    #   shap
+    # via mlflow
 pillow==10.0.1
     # via matplotlib
 protobuf==4.24.3
     # via mlflow
 pyarrow==10.0.1
     # via mlflow
-pyjwt==2.8.0
-    # via databricks-cli
 pyparsing==3.1.1
     # via matplotlib
 python-dateutil==2.8.2
@@ -149,7 +140,6 @@ querystring-parser==1.2.4
     # via mlflow
 requests==2.31.0
     # via
-    #   databricks-cli
     #   docker
     #   mlflow
 s3transfer==0.6.2
@@ -158,21 +148,14 @@ 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
     #   python-dateutil
     #   querystring-parser
-slicer==0.0.7
-    # via shap
 smmap==5.0.1
     # via gitdb
 sqlalchemy==1.4.49
@@ -181,20 +164,15 @@ sqlalchemy==1.4.49
     #   mlflow
 sqlparse==0.4.4
     # via mlflow
-tabulate==0.9.0
-    # via databricks-cli
 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
     # via
     #   botocore
-    #   databricks-cli
     #   docker
     #   minio
     #   requests