Fix mypy warnings

Deltares · Mar 13, 2024 · 5d5bd8b · 5d5bd8b
1 parent 6281c30
commit 5d5bd8b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/python/ribasim/ribasim/config.py b/python/ribasim/ribasim/config.py
@@ -132,7 +132,7 @@ def add(self, node: Node, tables: Sequence[TableModel[Any]] | None = None) -> No
             node_type=self.__class__.__name__,
         )
         self.node.df = (
-            node_table  # type: ignore
+            node_table
             if self.node.df is None
             else pd.concat([self.node.df, node_table])
         )

diff --git a/python/ribasim/ribasim/geometry/edge.py b/python/ribasim/ribasim/geometry/edge.py
@@ -76,7 +76,7 @@ def add(
         if self.df is None:
             self.df = table_to_append
         else:
-            self.df = pd.concat([self.df, table_to_append])  # type: ignore
+            self.df = pd.concat([self.df, table_to_append])
 
     def get_where_edge_type(self, edge_type: str) -> NDArray[np.bool_]:
         assert self.df is not None

diff --git a/python/ribasim/ribasim/input_base.py b/python/ribasim/ribasim/input_base.py
@@ -361,10 +361,9 @@ def _write_table(self, path: FilePath) -> None:
         ----------
         path : FilePath
         """
-
-        gdf = gpd.GeoDataFrame(data=self.df)
-        gdf = gdf.set_geometry("geometry")
-        gdf.to_file(path, layer=self.tablename(), driver="GPKG", mode="a")
+        if self.df is None:
+            return
+        self.df.to_file(path, layer=self.tablename(), driver="GPKG", mode="a")
 
     def sort(self):
         if self.df is not None: