From e21baf9878237dd8a7fc5ee1064403e26e1afb6b Mon Sep 17 00:00:00 2001 From: vfdev Date: Tue, 3 Sep 2024 19:12:12 +0200 Subject: [PATCH] Temporary pinned mlflow version to fix the failing CI (#3278) * Pinned mlflow version * Fixed mypy failure on the CI (#3279) --- ignite/handlers/wandb_logger.py | 4 ++-- requirements-dev.txt | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/ignite/handlers/wandb_logger.py b/ignite/handlers/wandb_logger.py index 621ff9d3ebd..ec3871ae9f8 100644 --- a/ignite/handlers/wandb_logger.py +++ b/ignite/handlers/wandb_logger.py @@ -135,13 +135,13 @@ def __init__(self, *args: Any, **kwargs: Any): ) if kwargs.get("init", True): kwargs.pop("init", None) - wandb.init(*args, **kwargs) + self._wandb.init(*args, **kwargs) def __getattr__(self, attr: Any) -> Any: return getattr(self._wandb, attr) def close(self) -> None: - self._wandb.finish() + self._wandb.finish() # type: ignore[attr-defined] def _create_output_handler(self, *args: Any, **kwargs: Any) -> "OutputHandler": return OutputHandler(*args, **kwargs) diff --git a/requirements-dev.txt b/requirements-dev.txt index d475e556cdf..732a0544b71 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -17,7 +17,7 @@ tensorboardX visdom polyaxon wandb -mlflow +mlflow<2.16.0 # https://github.com/pytorch/ignite/issues/3277 neptune-client>=0.16.17 tensorboard torchvision