chore: check that the emitter type is oil_volume

equinor · Apr 19, 2024 · 1f9e00b · 1f9e00b
1 parent 8cca29a
commit 1f9e00b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/libecalc/presentation/exporter/queries.py b/src/libecalc/presentation/exporter/queries.py
@@ -19,6 +19,9 @@
 from libecalc.core.result import GeneratorSetResult
 from libecalc.dto.utils.validators import convert_expression
 from libecalc.expression import Expression
+from libecalc.presentation.yaml.yaml_types.emitters.yaml_venting_emitter import (
+    YamlVentingType,
+)
 
 
 class Query(abc.ABC):
@@ -179,7 +182,7 @@ def query(
                     if (
                         self.consumer_categories is None
                         or venting_emitter.user_defined_category in self.consumer_categories
-                    ):
+                    ) and venting_emitter.type == YamlVentingType.OIL_VOLUME:
                         oil_volumes_values = Expression.evaluate(
                             convert_expression(venting_emitter.volume.rate.value),
                             variables=installation_graph.variables_map.variables,