From fe1cb8f550b8e31f1bddd26479fb1d291c9206d1 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 15 Jul 2024 20:07:07 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- nemo_skills/inference/prompt/nemotron/eval_map.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/nemo_skills/inference/prompt/nemotron/eval_map.py b/nemo_skills/inference/prompt/nemotron/eval_map.py index 51d401aa8..4f27c8928 100644 --- a/nemo_skills/inference/prompt/nemotron/eval_map.py +++ b/nemo_skills/inference/prompt/nemotron/eval_map.py @@ -16,11 +16,10 @@ # can have a default key that will be used if benchmark name is not in dict EVAL_MAP = { - 'base': { + 'base': { 'default': 'nemotron/fewshot', }, 'instruct': { # nemotron-instruct 'default': 'nemotron/zeroshot', }, - }