Validate served model name consistency
This commit is contained in:
@@ -576,7 +576,7 @@ class StudySpec:
|
||||
|
||||
@classmethod
|
||||
def from_dict(cls, data: Mapping[str, Any]) -> "StudySpec":
|
||||
return cls(
|
||||
study = cls(
|
||||
study_id=_require_str(data.get("study_id"), context="study_id"),
|
||||
hardware=HardwareSpec.from_dict(
|
||||
_require_mapping(data.get("hardware"), context="hardware")
|
||||
@@ -595,6 +595,14 @@ class StudySpec:
|
||||
if data.get("capability_profile_path")
|
||||
else None,
|
||||
)
|
||||
served_model_name = str(study.engine.base_flags.get("served-model-name") or "").strip()
|
||||
if served_model_name and served_model_name != study.model.served_model_name:
|
||||
raise SpecError(
|
||||
"model.served_model_name must match engine.base_flags['served-model-name'] "
|
||||
f"when both are set. Got model.served_model_name={study.model.served_model_name!r} "
|
||||
f"and engine.base_flags['served-model-name']={served_model_name!r}."
|
||||
)
|
||||
return study
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
|
||||
Reference in New Issue
Block a user