feat: only support str limit in toml
Some checks failed
build / build (push) Failing after 2m35s
build / build (pull_request) Failing after 2m33s

This commit is contained in:
张泊明518370910136 2025-03-04 22:44:37 -05:00
parent 719f19232b
commit d6a33d2f20
GPG Key ID: D47306D7062CDA9D

View File

@ -60,20 +60,24 @@ class Files(BaseModel):
class Limit(BaseModel):
mem: int = DEFAULT_MEMORY_LIMIT
cpu: int = DEFAULT_CPU_LIMIT
stderr: int = DEFAULT_FILE_LIMIT
stdout: int = DEFAULT_FILE_LIMIT
stderr: int = DEFAULT_FILE_LIMIT
model_config = ConfigDict(validate_assignment=True)
@field_validator("cpu", mode="before")
@classmethod
def ensure_time(cls, v: Any) -> Time:
if isinstance(v, str):
return Time(v)
raise ValueError("Must be a string")
@field_validator("mem", "stdout", "stderr", mode="before")
@classmethod
def ensure_mem(cls, v: Any) -> Memory:
if isinstance(v, str):
return Memory(v)
raise ValueError("Must be a string")
class Stage(BaseModel):