Skip to content

Commit

Permalink
Replace depricated root_validator with model_validator
Browse files Browse the repository at this point in the history
  • Loading branch information
KristofferSkare committed Mar 18, 2024
1 parent 5f8728b commit 661ff29
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/mlfmu/types/fmu_component.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
from typing import Any, Dict, List, Optional, Tuple, Union
from uuid import UUID

from pydantic import BaseModel, ConfigDict, StringConstraints, root_validator
from pydantic import BaseModel, ConfigDict, StringConstraints, model_validator
from pydantic.fields import Field
from typing_extensions import Annotated

Expand Down Expand Up @@ -101,7 +101,8 @@ class InternalState(BaseModelConfig):
examples=["10", "10:20", "30"],
)

@root_validator(allow_reuse=True, skip_on_failure=True)
@model_validator(mode="before")
@classmethod
def check_only_one_initialization(cls, values: Dict[str, Any]):
init_var = "initialization_variable" in values and values["initialization_variable"] is not None
name = "name" in values and values["name"] is not None
Expand Down

0 comments on commit 661ff29

Please sign in to comment.