fix: missing timestamps

CAVEconnectome · Sep 2, 2023 · 07dafa9 · 07dafa9
1 parent 5f383ae
commit 07dafa9
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pychunkedgraph/graph/edits.py b/pychunkedgraph/graph/edits.py
@@ -391,7 +391,7 @@ def remove_edges(
         new_cx_edges_d = cx_edges_d.get(new_id, {})
         for layer, edges in new_cx_edges_d.items():
             svs = np.unique(edges)
-            parents = cg.get_parents(svs)
+            parents = cg.get_parents(svs, time_stamp=parent_ts)
             temp_map = dict(zip(svs, parents))
 
             edges = fastremap.remap(edges, temp_map, preserve_missing_labels=True)
@@ -520,7 +520,10 @@ def _update_cross_edge_cache(self, parent, children):
         parent_layer = self.cg.get_chunk_layer(parent)
         edge_nodes = np.unique(np.concatenate([*cx_edges_d.values(), types.empty_2d]))
         edge_parents = self.cg.get_roots(
-            edge_nodes, stop_layer=parent_layer, ceil=False
+            edge_nodes,
+            stop_layer=parent_layer,
+            ceil=False,
+            time_stamp=self._last_successful_ts,
         )
         edge_parents_d = dict(zip(edge_nodes, edge_parents))