tmp debug meta-ref

This commit is contained in:
Xi Yan 2024-12-30 14:08:25 -08:00
parent b1062d3932
commit e634ceeafc
2 changed files with 3 additions and 4 deletions

View file

@ -186,6 +186,7 @@ async def construct_stack(
run_config, provider_registry or get_provider_registry(), dist_registry run_config, provider_registry or get_provider_registry(), dist_registry
) )
print("impls", impls) print("impls", impls)
print("run_config", run_config)
await register_resources(run_config, impls) await register_resources(run_config, impls)
return impls return impls

View file

@ -9,6 +9,7 @@ import tempfile
from typing import Any, Dict, List, Optional from typing import Any, Dict, List, Optional
from pydantic import BaseModel from pydantic import BaseModel
from termcolor import cprint
from llama_stack.apis.datasets import DatasetInput from llama_stack.apis.datasets import DatasetInput
from llama_stack.apis.eval_tasks import EvalTaskInput from llama_stack.apis.eval_tasks import EvalTaskInput
@ -57,11 +58,8 @@ async def construct_stack_for_test(
scoring_fns=scoring_fns or [], scoring_fns=scoring_fns or [],
eval_tasks=eval_tasks or [], eval_tasks=eval_tasks or [],
) )
from rich.pretty import pprint
pprint(f"run_config BEFORE: {run_config}")
run_config = parse_and_maybe_upgrade_config(run_config) run_config = parse_and_maybe_upgrade_config(run_config)
pprint(f"run_config AFTER: {run_config}") cprint(f"run_config AFTER: {run_config}", "red")
try: try:
remote_config = remote_provider_config(run_config) remote_config = remote_provider_config(run_config)
if not remote_config: if not remote_config: