Skip to content

Commit ff0bb85

Browse files
committed
Add extra=forbid to pydantic basemodels in ert run models and cleanup
1 parent 0a6bf58 commit ff0bb85

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

src/ert/run_models/event.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ class RunModelStatusEvent(RunModelEvent):
3333
detail: bool = False
3434

3535

36-
class EverestStatusEvent(BaseModel):
36+
class EverestStatusEvent(BaseModel, extra="forbid"):
3737
batch: int | None
3838
event_type: Literal["EverestStatusEvent"] = "EverestStatusEvent"
3939
everest_event: Literal[
@@ -42,7 +42,7 @@ class EverestStatusEvent(BaseModel):
4242
]
4343

4444

45-
class EverestBatchResultEvent(BaseModel):
45+
class EverestBatchResultEvent(BaseModel, extra="forbid"):
4646
batch: int
4747
event_type: Literal["EverestBatchResultEvent"] = "EverestBatchResultEvent"
4848
everest_event: Literal["OPTIMIZATION_RESULT",]
@@ -89,7 +89,7 @@ def write_as_csv(self, output_path: Path | None) -> None:
8989
self.data.to_csv("Report", output_path / str(self.run_id))
9090

9191

92-
class RunPathCreationEvent(BaseModel):
92+
class RunPathCreationEvent(BaseModel, extra="forbid"):
9393
pass
9494

9595

src/ert/run_models/initial_ensemble_run_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def str_to_dtype(dtype_str: str) -> pl.DataType:
3434
return dtype
3535

3636

37-
class DictEncodedDataFrame(BaseModel):
37+
class DictEncodedDataFrame(BaseModel, extra="forbid"):
3838
type: Literal["dicts"]
3939
data: list[dict[str, Any]]
4040
datatypes: dict[str, str]

0 commit comments

Comments
 (0)