fixed performance gap

AllenInstitute · Oct 11, 2024 · 827967d · 827967d
1 parent 7e0114f
commit 827967d
Show file tree

Hide file tree

Showing 15 changed files with 937 additions and 1,370 deletions.
diff --git a/src/deep_neurographs/generate_proposals.py b/src/deep_neurographs/generate_proposals.py
@@ -319,8 +319,8 @@ def run_trimming(neurograph, proposals, radius, progress_bar):
 
 def trim_endpoints(neurograph, i, j, radius):
     # Initializations
-    branch_i = neurograph.get_branch(i)
-    branch_j = neurograph.get_branch(j)
+    branch_i = neurograph.branch(i)
+    branch_j = neurograph.branch(j)
 
     # Check both orderings
     idx_i, idx_j = trim_endpoints_ordered(branch_i, branch_j)
@@ -394,8 +394,8 @@ def trim_to_idx(neurograph, i, idx):
 
     """
     # Update node
-    branch_xyz = neurograph.get_branch(i, key="xyz")
-    branch_radii = neurograph.get_branch(i, key="radius")
+    branch_xyz = neurograph.branch(i, key="xyz")
+    branch_radii = neurograph.branch(i, key="radius")
     neurograph.nodes[i]["xyz"] = branch_xyz[idx]
     neurograph.nodes[i]["radius"] = branch_radii[idx]
 
@@ -484,4 +484,4 @@ def tangent(branch, idx, depth):
 
     """
     end = min(idx + depth, len(branch))
-    return geometry.tangent(branch[idx:end])
+    return geometry.tangent(branch[idx:end])