Skip to content

Commit 45c0170

Browse files
committed
Rfc the server state into a more concrete class, init the scheduler into the lifespan and thus we can extend to more state in the future
1 parent 9701c73 commit 45c0170

File tree

3 files changed

+501
-396
lines changed

3 files changed

+501
-396
lines changed

lmms_eval/entrypoints/__init__.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,20 @@
11
from lmms_eval.entrypoints.client import AsyncEvalClient, EvalClient
22
from lmms_eval.entrypoints.http_server import launch_server
3+
from lmms_eval.entrypoints.job_scheduler import (
4+
EvaluateRequest,
5+
JobInfo,
6+
JobScheduler,
7+
JobStatus,
8+
)
39
from lmms_eval.entrypoints.server_args import ServerArgs
410

5-
__all__ = ["ServerArgs", "launch_server", "EvalClient", "AsyncEvalClient"]
11+
__all__ = [
12+
"ServerArgs",
13+
"launch_server",
14+
"EvalClient",
15+
"AsyncEvalClient",
16+
"JobScheduler",
17+
"JobStatus",
18+
"JobInfo",
19+
"EvaluateRequest",
20+
]

0 commit comments

Comments
 (0)