fix: typing

JBris · Sep 9, 2024 · f3ce1f5 · f3ce1f5
1 parent 650b006
commit f3ce1f5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/deeprootgen/model/hgraph.py b/deeprootgen/model/hgraph.py
@@ -5,8 +5,6 @@
 
 """
 
-# mypy: ignore-errors
-
 from typing import List
 
 import networkx as nx
@@ -56,11 +54,11 @@ def add_child_node(
         if new_organ:
             organ_id = self.G.increment_organ_id()
             segment_rank = 1
-            order = self.node_data.order + 1
+            order = self.node_data.order + 1  # type: ignore
         else:
-            organ_id = self.node_data.organ_id
-            segment_rank = self.node_data.segment_rank + 1
-            order = self.node_data.order
+            organ_id = self.node_data.organ_id  # type: ignore
+            segment_rank = self.node_data.segment_rank + 1  # type: ignore
+            order = self.node_data.order  # type: ignore
 
         child_data.parent_id = self.node_data.node_id
         child_data.organ_id = organ_id

diff --git a/deeprootgen/model/root.py b/deeprootgen/model/root.py
@@ -630,7 +630,7 @@ def __init__(
         """
         self.soil: Soil = Soil()
         self.G: RootSystemGraph = RootSystemGraph()
-        self.organs: Dict[str, List[RootOrgan]] = {}
+        self.organs: Dict[int, List[RootOrgan]] = {}
         self.simulation_tag = simulation_tag
         self.visualise = visualise
         self.rng = default_rng(random_seed)