readd

Author: chibu

src/automation/tasks/guidellm.py

@@ -1,4 +1,5 @@
 from automation.tasks import BaseTask
+from guidellm.benchmark.scenario import GenerativeTextScenario, get_builtin_scenarios
 from automation.configs import DEFAULT_DOCKER_IMAGE, DEFAULT_RESEARCH_BRANCH
 from typing import Optional, Sequence
 import os
@@ -51,6 +52,8 @@ def __init__(
             packages = self.guidellm_packages
 
         print(packages)
+        print(get_builtin_scenarios())
+        default_scenario = get_builtin_scenarios()[0]
         if "packages" in config_kwargs:
             packages = list(set(packages + config_kwargs.pop("packages")))