diff --git a/conda/meta.yaml b/conda/meta.yaml index 05cb8ac4907da72cb3552c50559ca8f12a9824ec..05fdfd74a911872359d566bf13f6fbabb62ce993 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 785d3790b07849527158d10ba7a84c105c17eb52..18129c9846d9908374c1543da38b8c1a7fa10c27 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 }