From b50614fe2781e1769efada1a36d01a472ca5aaa6 Mon Sep 17 00:00:00 2001
From: Gokhan Ozbulak <gokhan.ozbulak@idiap.ch>
Date: Tue, 9 Apr 2024 12:36:12 +0200
Subject: [PATCH] Add mednet dependencies for Gitlab/MLFLow integration,
related with #60
---
conda/meta.yaml | 4 ++++
pyproject.toml | 4 ++++
2 files changed, 8 insertions(+)
diff --git a/conda/meta.yaml b/conda/meta.yaml
index 05cb8ac4..05fdfd74 100644
--- a/conda/meta.yaml
+++ b/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
diff --git a/pyproject.toml b/pyproject.toml
index 785d3790..18129c98 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -47,6 +47,8 @@ dependencies = [
"tensorboard",
"grad-cam>=1.4.8",
"versioningit",
+ "python-gitlab",
+ "mlflow",
]
[project.urls]
@@ -94,6 +96,8 @@ lightning = ">=2.2.0"
tensorboard = "*"
grad-cam = ">=1.4.8"
versioningit = "*"
+python-gitlab = "*"
+mlflow = "*"
[tool.pixi.feature.self.pypi-dependencies]
mednet = { path = ".", editable = true }
--
GitLab