clean up prints

This commit is contained in:
Xi Yan 2024-12-30 13:55:11 -08:00
parent 421fa980bc
commit 8ec1a04c85

View file

@ -44,11 +44,7 @@ async def construct_stack_for_test(
scoring_fns: Optional[List[ScoringFnInput]] = None, scoring_fns: Optional[List[ScoringFnInput]] = None,
eval_tasks: Optional[List[EvalTaskInput]] = None, eval_tasks: Optional[List[EvalTaskInput]] = None,
) -> TestStack: ) -> TestStack:
from termcolor import cprint
sqlite_file = tempfile.NamedTemporaryFile(delete=False, suffix=".db") sqlite_file = tempfile.NamedTemporaryFile(delete=False, suffix=".db")
cprint(providers, "red")
run_config = dict( run_config = dict(
image_name="test-fixture", image_name="test-fixture",
apis=apis, apis=apis,
@ -62,7 +58,6 @@ async def construct_stack_for_test(
eval_tasks=eval_tasks or [], eval_tasks=eval_tasks or [],
) )
run_config = parse_and_maybe_upgrade_config(run_config) run_config = parse_and_maybe_upgrade_config(run_config)
print("construct_stack_for_test", run_config)
try: try:
remote_config = remote_provider_config(run_config) remote_config = remote_provider_config(run_config)
if not remote_config: if not remote_config: