Apply suggestions from code review

Co-authored-by: Hofer-Julian <[email protected]>
Deltares · Feb 12, 2024 · b8c3bc1 · b8c3bc1
1 parent 5996c12
commit b8c3bc1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/core/src/util.jl b/core/src/util.jl
@@ -378,7 +378,7 @@ function get_level(
     storage::Union{AbstractArray, Number} = 0,
 )::Union{Real, Nothing}
     (; basin, level_boundary) = p
-    return if node_id.type == NodeType.Basin
+   if node_id.type == NodeType.Basin
         _, i = id_index(basin.node_id, node_id)
         current_level = get_tmp(basin.current_level, storage)
         current_level[i]

diff --git a/python/ribasim/ribasim/model.py b/python/ribasim/ribasim/model.py
@@ -332,7 +332,7 @@ def _add_node_types(self):
         )
 
     @classmethod
-    def read(cls, filepath: FilePath) -> "Model":
+    def read(cls, filepath: FilePath) -> Self:
         """Read model from TOML file."""
         return cls(filepath=filepath)  # type: ignore