Skip to content

Commit

Permalink
Temporary pinned mlflow version to fix the failing CI (#3278)
Browse files Browse the repository at this point in the history
* Pinned mlflow version

* Fixed mypy failure on the CI (#3279)
  • Loading branch information
vfdev-5 authored Sep 3, 2024
1 parent aa3e3e1 commit e21baf9
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions ignite/handlers/wandb_logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
2 changes: 1 addition & 1 deletion requirements-dev.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit e21baf9

Please sign in to comment.