File tree Expand file tree Collapse file tree 2 files changed +7
-5
lines changed Expand file tree Collapse file tree 2 files changed +7
-5
lines changed Original file line number Diff line number Diff line change 1111
1212try :
1313 from lm_eval .__main__ import setup_parser as lm_eval_setup_parser
14- available = True
14+ from fastapi_utils . tasks import repeat_every
1515except ImportError :
16- available = False
16+ raise ImportError ( "The TrustyAI service was not built with LM-Evaluation-Harness support, use `pip install .[eval]`" )
1717
1818from pydantic import BaseModel , create_model
1919
2020from fastapi import HTTPException
21- from fastapi_utils .tasks import repeat_every
22-
2321from fastapi import APIRouter
2422import subprocess
2523import logging
Original file line number Diff line number Diff line change 2525from src .endpoints .metrics .metrics_info import router as metrics_info_router
2626from src .endpoints .data .data_download import router as data_download_router
2727
28- from src .endpoints .evaluation .lm_evaluation_harness import router as lm_evaluation_harness_router , available as lm_evaluation_harness_available
28+ try :
29+ from src .endpoints .evaluation .lm_evaluation_harness import router as lm_evaluation_harness_router
30+ lm_evaluation_harness_available = True
31+ except ImportError :
32+ lm_evaluation_harness_available = False
2933
3034logging .basicConfig (
3135 level = logging .DEBUG ,
You can’t perform that action at this time.
0 commit comments