fix: anisotropy not working for 2d images

seung-lab · Dec 19, 2023 · a9712de · a9712de
1 parent c1a96a1
commit a9712de
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dijkstra3d.pyx b/dijkstra3d.pyx
@@ -237,6 +237,10 @@ def dijkstra(
       voxel_graph
     )
   else:
+    anisotropy = list(anisotropy)
+    while len(anisotropy) < 3:
+      anisotropy.append(float(1))
+
     if bidirectional:
       warnings.warn("bidirectional = True is not currently supported by \
 anisotropy dijkstra3d. Fall back to vanilla dijkstra algorithm.")