diff --git a/lib/python/setup.py b/lib/python/setup.py
index 9cacc0da3..90006722d 100644
--- a/lib/python/setup.py
+++ b/lib/python/setup.py
@@ -36,9 +36,9 @@
         "boto3",
         "cloudpickle",
         "diskcache",
-        "mlflow==2.12.1",
+        "mlflow==2.22.2",
         "paho-mqtt",
-        "protobuf==3.20.3",
+        "protobuf==4.25.8",
         "grpcio==1.53.2",
         "pydantic<2.0",
         "gpustat",