diff --git a/conda/meta.yaml b/conda/meta.yaml index 69ff5efe1e87bac9630863cea73d696f84c0d695..50d8e72deb18c6d113a24e20ef0b0526555702e2 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 18129c9846d9908374c1543da38b8c1a7fa10c27..730ea1c7a825d91779f7391da23d1a64a91ad854 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 }