Skip to content
This repository was archived by the owner on Nov 1, 2024. It is now read-only.

[fix] Set upper limit for aim version #739

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

[fix] Set upper limit for aim version #739

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
61ebe00
Set upper limit for aim version
alberttorosyan Jun 26, 2023
8d6360f
[fix] Change the warning message when Aim is not installed
alberttorosyan Jun 27, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions metaseq/logging/progress_bar/aim_progress_bar.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand All @@ -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
Expand Down
2 changes: 1 addition & 1 deletion setup.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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",
],
Expand Down