Skip to content

Commit b527dcc

Browse files
committed
Add extra=forbid to pydantic basemodels in ert config and cleanup
# Conflicts: # src/ert/config/derived_response_config.py
1 parent a139c58 commit b527dcc

12 files changed

Lines changed: 8 additions & 23 deletions

src/ert/config/derived_response_config.py

Whitespace-only changes.

src/ert/config/distribution.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,7 @@
1313
T = TypeVar("T", bound="TransSettingsValidation")
1414

1515

16-
class TransSettingsValidation(BaseModel):
17-
model_config = {"extra": "forbid"}
18-
16+
class TransSettingsValidation(BaseModel, extra="forbid"):
1917
@classmethod
2018
def create(cls, *args: Any, **kwargs: Any) -> Self:
2119
return cls(*args, **kwargs)

src/ert/config/ensemble_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
logger = logging.getLogger(__name__)
2222

2323

24-
class EnsembleConfig(BaseModel):
24+
class EnsembleConfig(BaseModel, extra="forbid"):
2525
response_configs: dict[str, KnownErtResponseTypes] = Field(default_factory=dict)
2626
derived_response_configs: dict[str, KnownDerivedResponseTypes] = Field(
2727
default_factory=dict

src/ert/config/ert_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -785,7 +785,7 @@ def log_observation_keys(
785785
USER_CONFIG_SCHEMA = init_user_config_schema()
786786

787787

788-
class ErtConfig(BaseModel):
788+
class ErtConfig(BaseModel, extra="forbid"):
789789
DEFAULT_ENSPATH: ClassVar[str] = "storage"
790790
DEFAULT_RUNPATH_FILE: ClassVar[str] = ".ert_runpath_list"
791791
PREINSTALLED_FORWARD_MODEL_STEPS: ClassVar[

src/ert/config/everest_control.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
import numpy as np
1212
import polars as pl
1313
import xarray as xr
14-
from pydantic import BaseModel, ConfigDict, Field, model_validator
14+
from pydantic import BaseModel, Field, model_validator
1515
from ropt.workflow import find_sampler_plugin
1616

1717
from .parameter_config import LocalizationType, ParameterCardinality, ParameterConfig
@@ -26,7 +26,7 @@
2626
MutableDataType = MutableMapping[str, Number | MutableMapping[str, Number]]
2727

2828

29-
class SamplerConfig(BaseModel):
29+
class SamplerConfig(BaseModel, extra="forbid"):
3030
backend: str | None = Field(
3131
default=None,
3232
description=dedent(
@@ -86,7 +86,6 @@ class SamplerConfig(BaseModel):
8686
"""
8787
),
8888
)
89-
model_config = ConfigDict(extra="forbid")
9089

9190
@model_validator(mode="after")
9291
def validate_backend_and_method(self) -> Self:

src/ert/config/forward_model_step.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def _get_source_package() -> str:
112112
return "not found"
113113

114114

115-
class ForwardModelStepDocumentation(BaseModel):
115+
class ForwardModelStepDocumentation(BaseModel, extra="forbid"):
116116
config_file: str | None = Field(default=None)
117117
source_package: str = Field(default_factory=_get_source_package)
118118
source_function_name: str = Field(default="ert")

src/ert/config/model_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@
3131
# and used space in percentage is greater than FULL_DISK_PERCENTAGE_THRESHOLD
3232

3333

34-
class ModelConfig(BaseModel):
34+
class ModelConfig(BaseModel, extra="forbid"):
3535
num_realizations: int = 1
3636
runpath_format_string: str = DEFAULT_RUNPATH
3737
jobname_format_string: str = DEFAULT_JOBNAME_FORMAT

src/ert/config/parameter_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ class ParameterCardinality(StrEnum):
4545
one_config_per_realization_dataset = auto()
4646

4747

48-
class ParameterConfig(BaseModel):
48+
class ParameterConfig(BaseModel, extra="forbid"):
4949
type: str
5050
name: str
5151
forward_init: bool

src/ert/run_models/everest_run_model.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -301,8 +301,6 @@ def create(
301301
keys=_extract_summary_keys(everest_config), input_files=[eclbase]
302302
)
303303
)
304-
else:
305-
eclbase = None
306304

307305
runpath_config = ErtModelConfig(
308306
num_realizations=len(everest_config.model.realizations)
@@ -314,9 +312,6 @@ def create(
314312
/ "realization_<REALIZATION_ID>"
315313
/ "<SIM_DIR>"
316314
),
317-
eclbase_format_string=eclbase
318-
if eclbase is not None
319-
else DEFAULT_ECLBASE_FORMAT,
320315
)
321316

322317
simulator_config = everest_config.simulator

tests/ert/unit_tests/config/test_field.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,6 @@ def create_dummy_field(nx, ny, nz, mask):
102102
forward_init_file="no_nees",
103103
output_file="no_nees",
104104
grid_file="no_nees",
105-
mask_file=Path("grid_mask.npy"),
106105
)
107106

108107

0 commit comments

Comments
 (0)