feat: add alias to Parameter model

NOAA-OWP · Aug 28, 2024 · 3ed98b1 · 3ed98b1
1 parent 223161b
commit 3ed98b1
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/python/ngen_cal/src/ngen/cal/parameter.py b/python/ngen_cal/src/ngen/cal/parameter.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 
-from pydantic import BaseModel, Field
-from typing import Sequence
+from pydantic import BaseModel, Field, root_validator
+from typing import Sequence, Mapping, Optional
 
 class Parameter(BaseModel, allow_population_by_field_name = True):
     """
@@ -11,5 +11,13 @@ class Parameter(BaseModel, allow_population_by_field_name = True):
     min: float
     max: float
     init: float
+    alias: Optional[str]
+
+    @root_validator
+    def _set_alias(cls, values: dict) -> dict:
+        alias = values.get('alias', None)
+        if alias is None:
+            values['alias'] = values['name']
+        return values
 
 Parameters = Sequence[Parameter]