Pydantic for Aeroval setupclasses.py

pyaerocom/aeroval/setupclasses.py

@@ -1,10 +1,15 @@
-from __future__ import annotations
-
 import logging
 import os
+import sys
 from getpass import getuser
 from pathlib import Path
-from typing import Literal, Self
+
+if sys.version_info < (3, 10):
+    from typing_extensions import Self
+else:
+    from typing import Self
+
+from typing import Literal
 
 from pydantic import (
     BaseModel,
@@ -175,7 +180,7 @@
     round_floats_precision: int | None = None
 
     if round_floats_precision:
         set_float_serialization_precision(round_floats_precision)
 
 
 class TimeSetup(BaseModel):
@@ -338,7 +343,7 @@
             subset_dict = {k: self.model_extra[k] for k in time_cfg_keys}
             return TimeSetup(**subset_dict)
         else:
             return TimeSetup()
 
     @computed_field
     @property
@@ -346,8 +351,8 @@
         if hasattr(self, "model_extra") and bool(
             cfg_extra_keys := set(self.model_extra) & set(ModelMapsSetup.model_fields)
         ):
             subset_dict = {k: self.model_extra[k] for k in cfg_extra_keys}
             return ModelMapsSetup(**subset_dict)
         else:
             return ModelMapsSetup()
 
@@ -371,7 +376,7 @@
             subset_dict = {k: self.model_extra[k] for k in cfg_extra_keys}
             return EvalRunOptions(**subset_dict)
         else:
             return EvalRunOptions()
 
     @computed_field
     @property
@@ -382,7 +387,7 @@
             subset_dict = {k: self.model_extra[k] for k in cfg_extra_keys}
             return StatisticsSetup(**subset_dict)
         else:
             return StatisticsSetup(weighted_stats=True, annual_stats_constrained=False)
 
     ##################################
     ## Non-BaseModel-based attributes
@@ -420,7 +425,7 @@
     @field_validator("obs_cfg")
     def validate_obs_cfg(cls, v):
         if isinstance(v, ObsCollection):
             return v
         return ObsCollection(v)
 
     @field_serializer("obs_cfg")
@@ -432,7 +437,7 @@
     @field_validator("model_cfg")
     def validate_model_cfg(cls, v):
         if isinstance(v, ModelCollection):
             return v
         return ModelCollection(v)
 
     @field_serializer("model_cfg")