From 07dafa9c671899a16106342c13a2f97de70024b1 Mon Sep 17 00:00:00 2001 From: Akhilesh Halageri Date: Sat, 2 Sep 2023 19:42:49 +0000 Subject: [PATCH] fix: missing timestamps --- pychunkedgraph/graph/edits.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/pychunkedgraph/graph/edits.py b/pychunkedgraph/graph/edits.py index 9c44a1c5d..2f30c1016 100644 --- 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))