fix: lstm pathlib.Path serialization

NOAA-OWP · Sep 27, 2024 · 8d8ff52 · 8d8ff52
1 parent 2823b2c
commit 8d8ff52
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/ngen_conf/src/ngen/config/init_config/lstm.py b/python/ngen_conf/src/ngen/config/init_config/lstm.py
@@ -27,7 +27,10 @@ class LSTM(serde.YamlSerializerDeserializer):
     slope_mean: float
 
     class Config(serde.YamlSerializerDeserializer.Config):
-        field_serializers = {"verbose": lambda b: int(b)}
+        field_serializers = {
+            "verbose": lambda b: int(b),
+            "train_cfg_file": str,
+        }
         # descriptions source: https://github.com/NOAA-OWP/lstm/blob/63116cc6a6bbdb5537868f20ff55cc326795b570/bmi_config_files/README.md
         fields = {
             "area_sqkm": {"description": "allows bmi to adjust a weighted output"},