Skip to content

Commit 0ebf0da

Browse files
authored
Update SMHP CLI eval validation (#333)
1 parent 2f79372 commit 0ebf0da

File tree

1 file changed

+21
-1
lines changed
  • src/hyperpod_cli/validators/recipe_models/nova_evaluation

1 file changed

+21
-1
lines changed

src/hyperpod_cli/validators/recipe_models/nova_evaluation/model.py

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,9 @@ class RunConfig(BaseModel):
1111
replicas: Optional[int|str] = None
1212
data_s3_path: Optional[str] = None
1313
output_s3_path: Optional[str] = None
14+
mlflow_tracking_uri: Optional[str] = None
15+
mlflow_experiment_name: Optional[str] = None
16+
mlflow_run_name: Optional[str] = None
1417

1518

1619
class EvaluationConfig(BaseModel):
@@ -29,11 +32,28 @@ class InferenceConfig(BaseModel):
2932
top_p: Optional[float] = None
3033
temperature: Optional[float] = None
3134
max_new_tokens: Optional[int|str] = None
35+
max_model_len: Optional[int|str] = None
36+
top_logprobs: Optional[int] = None
37+
reasoning_effort: Optional[str] = None
38+
39+
40+
class RlEnvConfig(BaseModel):
41+
model_config = ConfigDict(extra="allow")
42+
43+
44+
class ProcessorConfig(BaseModel):
45+
model_config = ConfigDict(extra="allow")
3246

3347

3448
class NovaEvaluationRecipeSchema(BaseModel):
3549
model_config = ConfigDict(extra="forbid")
3650

51+
display_name: Optional[str] = None
52+
versions: Optional[list] = None
53+
instance_types: Optional[list] = None
54+
3755
run: RunConfig
3856
evaluation: EvaluationConfig
39-
inference: Optional[InferenceConfig] = None
57+
inference: Optional[InferenceConfig] = None
58+
rl_env: Optional[RlEnvConfig] = None
59+
processor: Optional[ProcessorConfig] = None

0 commit comments

Comments
 (0)