Ignore dynamically created imports in mypy.

Deltares · Sep 12, 2023 · 8ba0274 · 8ba0274
1 parent bcb893f
commit 8ba0274
Show file tree

Hide file tree

Showing 13 changed files with 34 additions and 28 deletions.
diff --git a/python/ribasim/ribasim/node_types/basin.py b/python/ribasim/ribasim/node_types/basin.py
@@ -3,7 +3,7 @@
 from pandera.typing import DataFrame
 
 from ribasim.input_base import TableModel
-from ribasim.schemas import (
+from ribasim.schemas import (  # type: ignore
     BasinForcingSchema,
     BasinProfileSchema,
     BasinStateSchema,

diff --git a/python/ribasim/ribasim/node_types/discrete_control.py b/python/ribasim/ribasim/node_types/discrete_control.py
@@ -1,7 +1,10 @@
 from pandera.typing import DataFrame
 
 from ribasim.input_base import TableModel
-from ribasim.schemas import DiscreteControlConditionSchema, DiscreteControlLogicSchema
+from ribasim.schemas import (  # type: ignore
+    DiscreteControlConditionSchema,
+    DiscreteControlLogicSchema,
+)
 
 __all__ = ("DiscreteControl",)
 

diff --git a/python/ribasim/ribasim/node_types/flow_boundary.py b/python/ribasim/ribasim/node_types/flow_boundary.py
@@ -3,7 +3,10 @@
 from pandera.typing import DataFrame
 
 from ribasim.input_base import TableModel
-from ribasim.schemas import FlowBoundaryStaticSchema, FlowBoundaryTimeSchema
+from ribasim.schemas import (  # type: ignore
+    FlowBoundaryStaticSchema,
+    FlowBoundaryTimeSchema,
+)
 
 __all__ = ("FlowBoundary",)
 

diff --git a/python/ribasim/ribasim/node_types/fractional_flow.py b/python/ribasim/ribasim/node_types/fractional_flow.py
@@ -1,7 +1,7 @@
 from pandera.typing import DataFrame
 
 from ribasim.input_base import TableModel
-from ribasim.schemas import FractionalFlowStaticSchema
+from ribasim.schemas import FractionalFlowStaticSchema  # type: ignore
 
 __all__ = ("FractionalFlow",)
 

diff --git a/python/ribasim/ribasim/node_types/level_boundary.py b/python/ribasim/ribasim/node_types/level_boundary.py
@@ -3,7 +3,10 @@
 from pandera.typing import DataFrame
 
 from ribasim.input_base import TableModel
-from ribasim.schemas import LevelBoundaryStaticSchema, LevelBoundaryTimeSchema
+from ribasim.schemas import (  # type: ignore
+    LevelBoundaryStaticSchema,
+    LevelBoundaryTimeSchema,
+)
 
 __all__ = ("LevelBoundary",)
 

diff --git a/python/ribasim/ribasim/node_types/linear_resistance.py b/python/ribasim/ribasim/node_types/linear_resistance.py
@@ -1,7 +1,7 @@
 from pandera.typing import DataFrame
 
 from ribasim.input_base import TableModel
-from ribasim.schemas import LinearResistanceStaticSchema
+from ribasim.schemas import LinearResistanceStaticSchema  # type: ignore
 
 __all__ = ("LinearResistance",)
 

diff --git a/python/ribasim/ribasim/node_types/manning_resistance.py b/python/ribasim/ribasim/node_types/manning_resistance.py
@@ -1,7 +1,7 @@
 from pandera.typing import DataFrame
 
 from ribasim.input_base import TableModel
-from ribasim.schemas import ManningResistanceStaticSchema
+from ribasim.schemas import ManningResistanceStaticSchema  # type: ignore
 
 __all__ = ("ManningResistance",)
 

diff --git a/python/ribasim/ribasim/node_types/outlet.py b/python/ribasim/ribasim/node_types/outlet.py
@@ -1,7 +1,7 @@
 from pandera.typing import DataFrame
 
 from ribasim.input_base import TableModel
-from ribasim.schemas import OutletStaticSchema
+from ribasim.schemas import OutletStaticSchema  # type: ignore
 
 __all__ = ("Outlet",)
 

diff --git a/python/ribasim/ribasim/node_types/pid_control.py b/python/ribasim/ribasim/node_types/pid_control.py
@@ -3,7 +3,7 @@
 from pandera.typing import DataFrame
 
 from ribasim.input_base import TableModel
-from ribasim.schemas import PidControlStaticSchema, PidControlTimeSchema
+from ribasim.schemas import PidControlStaticSchema, PidControlTimeSchema  # type: ignore
 
 __all__ = ("PidControl",)
 

diff --git a/python/ribasim/ribasim/node_types/pump.py b/python/ribasim/ribasim/node_types/pump.py
@@ -1,7 +1,7 @@
 from pandera.typing import DataFrame
 
 from ribasim.input_base import TableModel
-from ribasim.schemas import PumpStaticSchema
+from ribasim.schemas import PumpStaticSchema  # type: ignore
 
 __all__ = ("Pump",)
 

diff --git a/python/ribasim/ribasim/node_types/tabulated_rating_curve.py b/python/ribasim/ribasim/node_types/tabulated_rating_curve.py
@@ -3,7 +3,7 @@
 from pandera.typing import DataFrame
 
 from ribasim.input_base import TableModel
-from ribasim.schemas import (
+from ribasim.schemas import (  # type: ignore
     TabulatedRatingCurveStaticSchema,
     TabulatedRatingCurveTimeSchema,
 )

diff --git a/python/ribasim/ribasim/node_types/terminal.py b/python/ribasim/ribasim/node_types/terminal.py
@@ -1,7 +1,7 @@
 from pandera.typing import DataFrame
 
 from ribasim.input_base import TableModel
-from ribasim.schemas import TerminalStaticSchema
+from ribasim.schemas import TerminalStaticSchema  # type: ignore
 
 __all__ = ("Terminal",)
 

diff --git a/python/ribasim/ribasim/schemas.py b/python/ribasim/ribasim/schemas.py
@@ -2,7 +2,7 @@
 Generate Pandera Schemas from all autogenerated Pydantic Models
 
 These classes have Schema as a postfix, so Ribasim.models.PumpStatic
-becomes Ribasim.node_types.PumpStaticSchema.
+becomes Ribasim.schemas.PumpStaticSchema.
 """
 import inspect
 import sys
@@ -14,22 +14,19 @@
 
 
 def gen_schema(name, cls):
-    classname = f"{name}Schema"
-    setattr(
-        sys.modules[__name__],
-        classname,
-        type(
-            classname,
-            (pa.DataFrameModel,),
-            {
-                "Config": type(
-                    f"{classname}.Config",
-                    (),
-                    {"dtype": PydanticModel(cls), "coerce": True},
-                )
-            },
-        ),
+    cname = f"{name}Schema"
+    ctype = type(
+        cname,
+        (pa.DataFrameModel,),
+        {
+            "Config": type(
+                f"{cname}.Config",
+                (),
+                {"dtype": PydanticModel(cls), "coerce": True},
+            )
+        },
     )
+    setattr(sys.modules[__name__], cname, ctype)
 
 
 for name, cls in inspect.getmembers(models, inspect.isclass):