fix: correct type for total system rate in pump system v2 (#167)

equinor · Sep 12, 2023 · 5559cdd · 5559cdd
1 parent d40558e
commit 5559cdd
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/src/ecalc/libraries/libecalc/common/libecalc/input/yaml_types/components/pump_system.py b/src/ecalc/libraries/libecalc/common/libecalc/input/yaml_types/components/pump_system.py
@@ -21,7 +21,7 @@
 
 
 class OperationalSettings(ConsumerSystemOperationalConditionBase):
-    total_system_rate: Optional[str] = Field(
+    total_system_rate: Optional[ExpressionType] = Field(
         None,
         title="Total system rate",
         description="The total system rate expression."

diff --git a/...t/validation/snapshots/test_validation_json_schemas/test_json_schema_changed/schemas.json b/...t/validation/snapshots/test_validation_json_schemas/test_json_schema_changed/schemas.json
@@ -1079,9 +1079,19 @@
                             "type": "array"
                         },
                         "TOTAL_SYSTEM_RATE": {
+                            "anyOf": [
+                                {
+                                    "type": "string"
+                                },
+                                {
+                                    "type": "integer"
+                                },
+                                {
+                                    "type": "number"
+                                }
+                            ],
                             "description": "The total system rate expression.\n\nShould be used with RATE_FRACTIONS in OPERATIONAL_SETTINGS.",
-                            "title": "Total system rate",
-                            "type": "string"
+                            "title": "Total system rate"
                         }
                     },
                     "title": "OperationalSettings",