diff --git a/pyproject.toml b/pyproject.toml
index 063d30c795d1d0062328270fe28ed9351a51ba4b..b667933076844ec6c023d6b513f212ce0cb6c4a3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -52,6 +52,8 @@ dependencies = [
   "tensorboard",
   "grad-cam~=1.5.0",
   "versioningit",
+  "python-gitlab",
+  "mlflow",
 ]
 
 [project.urls]
@@ -106,6 +108,8 @@ lightning = "~=2.2.1"
 tensorboard = "*"
 grad-cam = "~=1.5.0"
 versioningit = "*"
+python-gitlab = "*"
+mlflow = "*"
 
 [tool.pixi.feature.self.pypi-dependencies]
 mednet = { path = ".", editable = true }