Remove SUT that's not working. Fix filename. Assorted other cleanup.

mlcommons · Feb 15, 2024 · ec7ffdd · ec7ffdd
1 parent b504bfd
commit ec7ffdd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/coffee/newhelm_runner.py b/src/coffee/newhelm_runner.py
@@ -10,6 +10,7 @@ class NewSutDescription:
     def __hash__(self):
         return super().__hash__()
 
+
 class NewhelmSut(NewSutDescription, Enum):
     GPT2 = "gpt2", "OpenAI GPT-2"
     LLAMA_2_7B = "llama-2-7b", "Meta Llama 2, 7b parameters"
diff --git a/src/coffee/run.py b/src/coffee/run.py
@@ -26,9 +26,11 @@ def _make_output_dir():
 @click.group()
 def cli() -> None:
     import newhelm.load_plugins
+
     newhelm.load_plugins.load_plugins()
 
     from newhelm.secrets_registry import SECRETS
+
     SECRETS.set_values(get_or_create_json_file("secrets/default.json"))