import some type annotations to the generated bmi

csdms · Jan 9, 2024 · ade3dad · ade3dad
1 parent d34409b
commit ade3dad
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/bmipy/_template.py b/src/bmipy/_template.py
@@ -19,7 +19,10 @@ def render(self) -> str:
         prefix = f"""\
 from __future__ import annotations
 
+from typing import Any
+
 import numpy as np
+from numpy.typing import NDArray
 
 from bmipy.bmi import Bmi