diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 81802675..3ecc91f0 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -8,7 +8,7 @@ default_language_version: python: python3.11 # NOTE: sync with .python-version-default repos: - repo: https://github.com/astral-sh/ruff-pre-commit - rev: "v0.5.6" + rev: "v0.5.7" hooks: - id: ruff alias: r diff --git a/src/openllm/__main__.py b/src/openllm/__main__.py index 5d9a8d67..74029acc 100644 --- a/src/openllm/__main__.py +++ b/src/openllm/__main__.py @@ -39,7 +39,7 @@ def _select_bento_name(models: list[BentoInfo], target: DeploymentTarget): model_infos = [(model.repo.name, model.name, can_run(model, target)) for model in models] model_name_groups = defaultdict(lambda: 0.0) for repo, name, score in model_infos: - model_name_groups[(repo, name)] += score + model_name_groups[repo, name] += score table_data = [(name, repo, CHECKED if score > 0 else '') for (repo, name), score in model_name_groups.items()] if not table_data: output('No model found', style='red')