Skip to content

Commit bcf8ccb

Browse files
committed
Avoid protected pydantic ns in cfg schema
using "model" seems really problematic -- wth, pydantic!!
1 parent 20f16f4 commit bcf8ccb

3 files changed

Lines changed: 3 additions & 3 deletions

File tree

notebooks/pdp/config-v2-TEMPLATE.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ timeout_minutes = 10
4545
[models.graduation]
4646
experiment_id = "EXPERIMENT_ID"
4747
run_id = "RUN_ID"
48-
model_type = "sklearn"
48+
framework = "sklearn"
4949

5050
[inference]
5151
num_top_features = 5

src/student_success_tool/configs/schemas/pdp_v2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ class DatasetConfig(pyd.BaseModel):
186186
class ModelConfig(pyd.BaseModel):
187187
experiment_id: str
188188
run_id: str
189-
model_type: t.Optional[t.Literal["sklearn", "xgboost", "lightgbm"]] = None
189+
framework: t.Optional[t.Literal["sklearn", "xgboost", "lightgbm"]] = None
190190

191191
@pyd.computed_field # type: ignore[misc]
192192
@property

tests/configs/schemas/test_pdp_v2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def template_cfg_str():
6060
[models.graduation]
6161
experiment_id = "EXPERIMENT_ID"
6262
run_id = "RUN_ID"
63-
model_type = "sklearn"
63+
framework = "sklearn"
6464
6565
[inference]
6666
num_top_features = 5

0 commit comments

Comments
 (0)