chore: create new yaml builders for testing (#708)

* chore: create new yaml builders for testing
equinor · Nov 21, 2024 · d942897 · d942897
1 parent 09929cb
commit d942897
Show file tree

Hide file tree

Showing 3 changed files with 408 additions and 7 deletions.
diff --git a/src/libecalc/presentation/yaml/yaml_types/components/legacy/yaml_electricity_consumer.py b/src/libecalc/presentation/yaml/yaml_types/components/legacy/yaml_electricity_consumer.py
@@ -32,7 +32,7 @@ class YamlElectricityConsumer(YamlBase):
         title="NAME",
         description="Name of the consumer.\n\n$ECALC_DOCS_KEYWORDS_URL/NAME",
     )
-    category: ConsumerUserDefinedCategoryType = CategoryField(...)
+    category: YamlTemporalModel[ConsumerUserDefinedCategoryType] = CategoryField(...)
     energy_usage_model: YamlTemporalModel[YamlElectricityEnergyUsageModel] = Field(
         ...,
         title="ENERGY_USAGE_MODEL",

diff --git a/src/libecalc/presentation/yaml/yaml_types/components/legacy/yaml_fuel_consumer.py b/src/libecalc/presentation/yaml/yaml_types/components/legacy/yaml_fuel_consumer.py
@@ -32,7 +32,7 @@ class YamlFuelConsumer(YamlBase):
         title="NAME",
         description="Name of the consumer.\n\n$ECALC_DOCS_KEYWORDS_URL/NAME",
     )
-    category: ConsumerUserDefinedCategoryType = CategoryField(...)
+    category: YamlTemporalModel[ConsumerUserDefinedCategoryType] = CategoryField(...)
     energy_usage_model: YamlTemporalModel[YamlFuelEnergyUsageModel] = Field(
         ...,
         title="ENERGY_USAGE_MODEL",