fix: backwards compatibility of NWMRequest

NOAA-OWP · Jan 30, 2024 · 0c71143 · 0c71143
1 parent 156c2f7
commit 0c71143
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/python/lib/communication/dmod/communication/maas_request/nwm/nwm_request.py b/python/lib/communication/dmod/communication/maas_request/nwm/nwm_request.py
@@ -51,8 +51,14 @@ def __init__(
             super().__init__(**data)
         else:
             data["request_body"] = dict()
-            nwm_inner_request_body = {"config_data_id": config_data_id}
-            data["request_body"]["nwm"] = nwm_inner_request_body
+            if "model" in data:
+                nwm_inner_request_body = data["model"][self.get_model_name()]
+                if config_data_id is not None:
+                    nwm_inner_request_body["config_data_id"] = config_data_id
+                data["request_body"][self.get_model_name()] = nwm_inner_request_body
+            else:
+                data["request_body"][self.get_model_name()] = {"config_data_id": config_data_id}
+
             super().__init__(**data)
 
     @classmethod