Skip to content

Commit dddb066

Browse files
committed
Complete Inspect AI integration with proper fallback chain
1 parent 0610100 commit dddb066

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

trainer_with_eval.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@
6767
from hyperparam_utils import get_recommended_lr, get_lr_with_warmup
6868
from logger import StructuredLogger
6969
from checkpoint_manager import CheckpointManager, find_latest_run
70+
from inspect_eval import run_inspect_evaluation, INSPECT_AVAILABLE
7071
except ImportError:
7172
TrainingConfig = None
7273
DataLoader = None
@@ -76,6 +77,8 @@
7677
StructuredLogger = None
7778
CheckpointManager = None
7879
find_latest_run = None
80+
run_inspect_evaluation = None
81+
INSPECT_AVAILABLE = False
7982

8083

8184
def prepare_training_data(
@@ -177,6 +180,7 @@ async def run_evaluations(
177180
tasks: list,
178181
renderer_name: str,
179182
threshold: float,
183+
service_client: Optional[Any] = None,
180184
training_client: Optional[Any] = None,
181185
evalops_client: Optional[Any] = None,
182186
test_suite_id: Optional[str] = None,
@@ -399,6 +403,7 @@ async def async_main(config_path: str, resume: bool = False) -> None:
399403
tasks=tasks,
400404
renderer_name=renderer_name,
401405
threshold=eval_threshold,
406+
service_client=service_client,
402407
training_client=training_client,
403408
evalops_client=evalops_client,
404409
test_suite_id=test_suite_id,

0 commit comments

Comments
 (0)