fix: handle different types of input to cross sectional area

seung-lab · Aug 2, 2024 · 3a289bc · 3a289bc
1 parent a1d20f8
commit 3a289bc
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/kimimaro/utility.py b/kimimaro/utility.py
@@ -249,7 +249,14 @@ def cross_sectional_area_helper(skel, binimg, roi):
     cross_sectional_area_helper
   )
 
-  for skel in skeletons.values():
+  if isinstance(skeletons, Skeleton):
+    skelitr = [ skeletons ]
+  elif isinstance(skeletons, dict):
+    skelitr = skeletons.values()
+  else:
+    skelitr = iter(skeletons)
+
+  for skel in skelitr:
     add_property(skel, xs_prop)
     add_property(skel, xs_contact_prop)