diff --git a/tests/templates/conftest.py b/tests/templates/conftest.py index cf347949..26fe87bd 100644 --- a/tests/templates/conftest.py +++ b/tests/templates/conftest.py @@ -4,7 +4,7 @@ from jinja2 import Environment, FileSystemLoader -from coffee.helm import HelmSut +from coffee.helm_runner import HelmSut from coffee.benchmark import ( GeneralChatBotBenchmarkDefinition, BiasHarmDefinition,