From cd70b2b3ad40dbd15d39c09c4445260dea41c1e8 Mon Sep 17 00:00:00 2001
From: Gokhan Ozbulak <gokhan.ozbulak@idiap.ch>
Date: Tue, 23 Apr 2024 12:48:02 +0200
Subject: [PATCH] Swithcing to lightweight package for MLFlow #60.
---
conda/meta.yaml | 4 ++--
pyproject.toml | 4 ++--
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/conda/meta.yaml b/conda/meta.yaml
index 69ff5efe..50d8e72d 100644
--- a/conda/meta.yaml
+++ b/conda/meta.yaml
@@ -42,7 +42,7 @@ requirements:
- tqdm {{ tqdm }}
- tensorboard {{ tensorboard }}
- python-gitlab {{ python_gitlab }}
- - mlflow {{ mlflow }}
+ - mlflow-skinny {{ mlflow_skinny }}
- lightning {{ lightning }}
- lightning >=2.2.0
run:
@@ -65,7 +65,7 @@ requirements:
- {{ pin_compatible('tqdm') }}
- {{ pin_compatible('tensorboard') }}
- {{ pin_compatible('python-gitlab') }}
- - {{ pin_compatible('mlflow') }}
+ - {{ pin_compatible('mlflow-skinny') }}
- {{ pin_compatible('lightning', max_pin='x.x') }}
- lightning >=2.2.0
diff --git a/pyproject.toml b/pyproject.toml
index 18129c98..730ea1c7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -48,7 +48,7 @@ dependencies = [
"grad-cam>=1.4.8",
"versioningit",
"python-gitlab",
- "mlflow",
+ "mlflow-skinny",
]
[project.urls]
@@ -97,7 +97,7 @@ tensorboard = "*"
grad-cam = ">=1.4.8"
versioningit = "*"
python-gitlab = "*"
-mlflow = "*"
+mlflow-skinny = "*"
[tool.pixi.feature.self.pypi-dependencies]
mednet = { path = ".", editable = true }
--
GitLab