sorting/optimizing imports

src/coffee/benchmark.py

@@ -4,7 +4,7 @@
 
 import casefy
 
-from coffee.helm_runner import BbqHelmTest, HelmTest, RealToxicityPromptsHelmTest, HelmSut
+from coffee.helm_runner import BbqHelmTest, HelmSut, HelmTest, RealToxicityPromptsHelmTest
 
 
 def float_to_stars(value: float) -> float:

src/coffee/helm_interface.py

@@ -1,5 +1,5 @@
 import subprocess
-from typing import Iterable, TYPE_CHECKING
+from typing import Iterable
 
 import helm.benchmark.run_specs
 from helm.benchmark.config_registry import (

src/coffee/run.py

@@ -7,11 +7,8 @@
 import termcolor
 
 import coffee
-from coffee.benchmark import (
-    GeneralChatBotBenchmarkDefinition,
-    BenchmarkScore,
-)
-from coffee.helm import HelmSut, CliHelmRunner, HelmResult
+from coffee.benchmark import BenchmarkScore, GeneralChatBotBenchmarkDefinition
+from coffee.helm_runner import CliHelmRunner, HelmResult, HelmSut
 from coffee.static_site_generator import StaticSiteGenerator
 
 

src/coffee/static_site_generator.py

@@ -2,7 +2,6 @@
 import pathlib
 import shutil
 from itertools import groupby
-from typing import Iterator
 
 from jinja2 import Environment, PackageLoader, select_autoescape
 from markupsafe import Markup

src/dubious_helm_cli_wrapper.py

@@ -1,5 +1,5 @@
-from helm.benchmark.run import main
 import helm.benchmark.run_specs
+from helm.benchmark.run import main
 
 helm.benchmark.run_specs.INCLUDE_GENERATIVE_HARMS_METRICS = True
 if __name__ == "__main__":