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 df22f9495d0d7a9d25a97cd329559c4f0bcb2cd3..9d3e6f969e57442f3b5303f57ee9c6cff21f846d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -47,6 +47,8 @@ dependencies = [ "tensorboard", "grad-cam>=1.4.8", "versioningit", + "python-gitlab", + "mlflow", ] [project.urls] @@ -95,6 +97,8 @@ lightning = ">=2.2.0" tensorboard = "*" grad-cam = ">=1.4.8" versioningit = "*" +python-gitlab = "*" +mlflow = "*" [tool.pixi.pypi-dependencies] mednet = { path = ".", editable = true, extras = ["qa", "doc", "test"] }