bug: integrate upds to whole-brain (#88)

Co-authored-by: anna-grim <[email protected]>

src/deep_neurographs/neurograph.py

@@ -154,6 +154,7 @@ def generate_proposals(
 
         """
         self.init_kdtree()
+        self.reset_proposals()
         self.proposals = dict()
         existing_connections = dict()
         doubles = set()
@@ -223,6 +224,10 @@ def generate_proposals(
         if optimize:
             self.run_optimization()
 
+    def reset_proposals(self):
+        for i in self.nodes:
+            self.nodes[i]["proposals"] = set()
+
     def set_proposals_per_leaf(self, proposals_per_leaf):
         self.proposals_per_leaf = proposals_per_leaf
 

src/deep_neurographs/utils.py

@@ -360,6 +360,10 @@ def open_tensorstore(path, driver):
     ).result()
     if driver == "neuroglancer_precomputed":
         return arr[ts.d["channel"][0]]
+    elif driver == "zarr":
+        arr = arr[0, 0, :, :, :]
+        arr = arr[ts.d[0].transpose[2]]
+        arr = arr[ts.d[0].transpose[1]]
     return arr