Add mednet dependencies for Gitlab/MLFLow integration, related with #60

conda/meta.yaml

@@ -41,6 +41,8 @@ requirements:
     - torchvision  {{ torchvision }}
     - tqdm {{ tqdm }}
     - tensorboard {{ tensorboard }}
+    - python-gitlab {{ python-gitlab }}
+    - mlflow {{ mlflow }}
     - lightning {{ lightning }}
     - lightning >=2.2.0
   run:
@@ -62,6 +64,8 @@ requirements:
     - {{ pin_compatible('torchvision') }}
     - {{ pin_compatible('tqdm') }}
     - {{ pin_compatible('tensorboard') }}
+    - {{ pin_compatible('python-gitlab') }}
+    - {{ pin_compatible('mlflow') }}
     - {{ pin_compatible('lightning', max_pin='x.x') }}
     - lightning >=2.2.0
 

pyproject.toml

@@ -47,6 +47,8 @@ dependencies = [
   "tensorboard",
   "grad-cam>=1.4.8",
   "versioningit",
+  "python-gitlab",
+  "mlflow",
 ]
 
 [project.urls]
@@ -95,6 +97,8 @@ lightning = ">=2.2.0"
 tensorboard = "*"
 grad-cam = ">=1.4.8"
 versioningit = "*"
+python-gitlab = "*"
+mlflow = "*"
 
 [tool.pixi.pypi-dependencies]
 mednet = { path = ".", editable = true, extras = ["qa", "doc", "test"] }