fix: serialising Enums

Author: theissenhelen

src/anemoi/training/schemas/data.py

@@ -36,7 +36,7 @@ class RemapperSchema(BaseModel):
     default: str = Field(description="Remapper default method to apply.", examples=["none", "cos_sin"])
 
 
-class Target(Enum):
+class Target(str, Enum):
     normalizer = "anemoi.models.preprocessing.normalizer.InputNormalizer"
     imputer = "anemoi.models.preprocessing.imputer.InputImputer"
     remapper = "anemoi.models.preprocessing.remapper.Remapper"

src/anemoi/training/schemas/training.py

@@ -78,7 +78,7 @@ class LossScalingSchema(BaseModel):
     sfc: dict[str, NonNegativeFloat]
 
 
-class PressureLevelScalerTargets(Enum):
+class PressureLevelScalerTargets(str, Enum):
 
     relu_scaler = "anemoi.training.data.scaling.ReluPressureLevelScaler"
     linear_scaler = "anemoi.training.data.scaling.LinearPressureLevelScaler"