Skip to content

Commit

Permalink
Make mypy happy.
Browse files Browse the repository at this point in the history
  • Loading branch information
evetion committed Mar 13, 2024
1 parent bfc882e commit c119d2c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/ribasim/ribasim/geometry/edge.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ def add(
if geometry is None
else [geometry]
)
table_to_append = GeoDataFrame(
table_to_append = GeoDataFrame[EdgeSchema](
data={
"from_node_type": pd.Series([from_node.node_type], dtype=str),
"from_node_id": pd.Series([from_node.node_id], dtype=int),
Expand All @@ -79,7 +79,7 @@ def add(
if self.df is None:
self.df = table_to_append
else:
self.df = pd.concat([self.df, table_to_append])
self.df = GeoDataFrame[EdgeSchema](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
Expand Down

0 comments on commit c119d2c

Please sign in to comment.