improve type annotation

fabm-model · Oct 21, 2024 · 9ed3b9e · 9ed3b9e
1 parent 13332fa
commit 9ed3b9e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/pyfabm/__init__.py b/src/pyfabm/__init__.py
@@ -1409,7 +1409,7 @@ def _update_configuration(self, settings: Optional[Tuple] = None):
             + self.horizontal_dependencies
             + self.scalar_dependencies
         )
-        self.variables = (
+        self.variables: NamedObjectList[VariableFromPointer] = (
             self.state_variables + self.diagnostic_variables + self.dependencies
         )
 
@@ -1422,7 +1422,7 @@ def _update_configuration(self, settings: Optional[Tuple] = None):
 
         self.itime = -1.0
 
-    def getRates(self, t: float = None, surface: bool = True, bottom: bool = True):
+    def getRates(self, t: Optional[float] = None, surface: bool = True, bottom: bool = True):
         """Returns the local rate of change in state variables,
         given the current state and environment.
         """
@@ -1459,7 +1459,7 @@ def getRates(self, t: float = None, surface: bool = True, bottom: bool = True):
 
     def get_sources(
         self,
-        t: float = None,
+        t: Optional[float] = None,
         out: Optional[Tuple[np.ndarray, np.ndarray, np.ndarray]] = None,
     ) -> Tuple[np.ndarray, np.ndarray, np.ndarray]:
         if t is None: