From ac8e912579b510e06a86b460a5e427158b623cc5 Mon Sep 17 00:00:00 2001
From: Nicolas Mounet <nicolas.mounet@cern.ch>
Date: Thu, 29 Apr 2021 09:13:57 +0200
Subject: [PATCH] Proper typing in HVACMechanical and HEPAFilter classes

---
 cara/models.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cara/models.py b/cara/models.py
index eaec0591..0c04eac0 100644
--- a/cara/models.py
+++ b/cara/models.py
@@ -364,7 +364,7 @@ class HEPAFilter(Ventilation):
 
     #: The rate at which the HEPA exchanges air (when switched on)
     # in m^3/h
-    q_air_mech: float
+    q_air_mech: _VectorisedFloat
 
     def air_exchange(self, room: Room, time: float) -> _VectorisedFloat:
         # If the HEPA is off, no air is being exchanged.
@@ -381,7 +381,7 @@ class HVACMechanical(Ventilation):
 
     #: The rate at which the HVAC exchanges air (when switched on)
     # in m^3/h
-    q_air_mech: float
+    q_air_mech: _VectorisedFloat
 
     def air_exchange(self, room: Room, time: float) -> _VectorisedFloat:
         # If the HVAC is off, no air is being exchanged.