beartype

Author: wpbonelli

flopy4/__init__.py

@@ -2,9 +2,12 @@
 
 import numpy as np
 from attr import Attribute, fields_dict
+from beartype.claw import beartype_this_package
 from numpy.typing import ArrayLike, NDArray
 from xarray import Dataset, DataTree
 
+beartype_this_package()
+
 
 def _parse_dim_names(shape: str) -> tuple[str, ...]:
     return tuple(

flopy4/mf6/io/parser.py

@@ -1,5 +1,5 @@
+from collections.abc import Iterable
 from os import linesep
-from typing import Iterable
 
 from lark import Lark
 

pyproject.toml

@@ -35,6 +35,7 @@ classifiers = [
 requires-python = ">=3.11"
 dependencies = [
     "attrs",                                # todo: bounds?
+    "beartype",
     "cattrs",                               # todo: bounds?
     "flopy>=3.7.0",
     "Jinja2>=3.0",
@@ -60,7 +61,6 @@ test = [
     "jupyter",
     "jupytext",
     "modflow-devtools",
-    "mypy",
     "pooch>=1.8",
     "pytest!=8.1.0",
     "pytest-dotenv",