From dac3156b4abf55e6a3847c18deb8f41e645b84e0 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.

---
 pyproject.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index b6679330..ab6b90f0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -53,7 +53,7 @@ dependencies = [
   "grad-cam~=1.5.0",
   "versioningit",
   "python-gitlab",
-  "mlflow",
+  "mlflow-skinny",
 ]
 
 [project.urls]
@@ -109,7 +109,7 @@ tensorboard = "*"
 grad-cam = "~=1.5.0"
 versioningit = "*"
 python-gitlab = "*"
-mlflow = "*"
+mlflow-skinny = "*"
 
 [tool.pixi.feature.self.pypi-dependencies]
 mednet = { path = ".", editable = true }
-- 
GitLab