Override __init__ instead for pydantic data transform

equinor · Jan 4, 2024 · 37f6583 · 37f6583
1 parent c76cd42
commit 37f6583
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/subscript/sunsch/sunsch.py b/src/subscript/sunsch/sunsch.py
@@ -65,9 +65,8 @@ class SunschConfig(BaseModel):
     ] = None
     insert: Optional[List[InsertStatement]] = None
 
-    @model_validator(mode="before")
-    @classmethod
-    def set_date_defaults(cls, config: Any) -> Any:
+    def __init__(self, **config):
+        """Transform the input to provide defaults to required fields"""
         if "startdate" not in config:
             if "starttime" in config:
                 config["startdate"] = config["starttime"]
@@ -82,7 +81,7 @@ def set_date_defaults(cls, config: Any) -> Any:
         if "refdate" not in config:
             config["refdate"] = config["startdate"]
 
-        return config
+        super().__init__(**config)
 
 
 def datetime_from_date(