From 61ebe00f64e9e1f6ead1423d6c172fefb963f087 Mon Sep 17 00:00:00 2001 From: Albert Torosyan Date: Mon, 26 Jun 2023 13:59:28 +0400 Subject: [PATCH 1/2] Set upper limit for aim version --- metaseq/logging/progress_bar/aim_progress_bar.py | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/metaseq/logging/progress_bar/aim_progress_bar.py b/metaseq/logging/progress_bar/aim_progress_bar.py index 4b1ad7dd9..aae3d828b 100644 --- a/metaseq/logging/progress_bar/aim_progress_bar.py +++ b/metaseq/logging/progress_bar/aim_progress_bar.py @@ -12,7 +12,7 @@ try: import functools - from aim import Repo as AimRepo + from aim.sdk import Repo as AimRepo @functools.lru_cache() def get_aim_run(repo, run_hash): diff --git a/setup.py b/setup.py index 197bd41df..388cbc62a 100644 --- a/setup.py +++ b/setup.py @@ -301,7 +301,7 @@ def do_setup(): "dev": [ "flake8", "black==22.3.0", - "aim>=3.9.4", + "aim>=3.9.4,<4.0.0", "azure-storage-blob", "mypy", ], From 8d6360f875c385acd6feeb611647e5859abe612f Mon Sep 17 00:00:00 2001 From: Albert Torosyan Date: Tue, 27 Jun 2023 11:27:08 +0400 Subject: [PATCH 2/2] [fix] Change the warning message when Aim is not installed --- metaseq/logging/progress_bar/aim_progress_bar.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metaseq/logging/progress_bar/aim_progress_bar.py b/metaseq/logging/progress_bar/aim_progress_bar.py index aae3d828b..d24383338 100644 --- a/metaseq/logging/progress_bar/aim_progress_bar.py +++ b/metaseq/logging/progress_bar/aim_progress_bar.py @@ -33,7 +33,7 @@ def __init__(self, wrapped_bar, aim_repo, aim_run_hash, aim_param_checkpoint_dir if get_aim_run is None: self.run = None - logger.warning("Aim not found, please install with: pip install aim") + logger.warning("Aim not found, please install with: pip install \"aim<4.0.0\"") else: logger.info(f"Storing logs at Aim repo: {aim_repo}") assert AimRepo is not None