Skip to content

Commit

Permalink
fix schema
Browse files Browse the repository at this point in the history
  • Loading branch information
JPXKQX committed Feb 21, 2025
1 parent e5b1e76 commit 21b976e
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions training/src/anemoi/training/schemas/training.py
Original file line number Diff line number Diff line change
Expand Up @@ -209,14 +209,14 @@ def check_length_of_weights_and_losses(self, values: dict) -> CombinedLossSchema
LossSchemas = Union[BaseLossSchema, HuberLossSchema, CombinedLossSchema]


class ScaleValidationMetrics(BaseModel):
class ScaleValidationMetricsSchema(BaseModel):
"""Configuration for scaling validation metrics.
Here variable scaling is possible due to the metrics being calculated in the same way as the
training loss, within internal model space.
"""

scalars_to_apply: list[str] = Field(example=["variable"])
scalers_to_apply: list[str] = Field(example=["variable"])
"""List of scalars to be applied."""
metrics: list[str]
"""List of metrics to keep in normalised space.."""
Expand Down Expand Up @@ -261,7 +261,7 @@ class TrainingSchema(BaseModel):
"Scalers to use in the computation of the loss and validation scores."
validation_metrics: dict[str, LossSchemas]
"List of validation metrics configurations. These metrics "
scale_validation_metrics: ScaleValidationMetrics
scale_validation_metrics: ScaleValidationMetricsSchema
"""Configuration for scaling validation metrics."""
rollout: Rollout = Field(default_factory=Rollout)
"Rollout configuration."
Expand Down

0 comments on commit 21b976e

Please sign in to comment.