From 3e5b91dd2a0814bfdffecdc59a86ea8630f75fca Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 19 Aug 2024 11:21:56 -0400 Subject: [PATCH] ci: pre-commit autoupdate [pre-commit.ci] (#1061) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * ci: pre-commit autoupdate [pre-commit.ci] updates: - [github.com/astral-sh/ruff-pre-commit: v0.5.6 → v0.5.7](https://github.com/astral-sh/ruff-pre-commit/compare/v0.5.6...v0.5.7) * ci: auto fixes from pre-commit.ci For more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> --- .pre-commit-config.yaml | 2 +- src/openllm/__main__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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')