@@ -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
1619class 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
3448class 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