From ade3dad373aab3b9b6c97852681cbd7d7d5b22d4 Mon Sep 17 00:00:00 2001
From: mcflugen <mcflugen@gmail.com>
Date: Mon, 8 Jan 2024 21:03:20 -0700
Subject: [PATCH] import some type annotations to the generated bmi

---
 src/bmipy/_template.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/bmipy/_template.py b/src/bmipy/_template.py
index 3a7be02..acb9981 100644
--- 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