diff --git a/aliby/pipeline.py b/aliby/pipeline.py index f41eadf6a455ce2101b9ab0c1ad91d44fd6a7af7..c0bd873ee9aa6815ccdd2a73298ff30d240d724d 100644 --- a/aliby/pipeline.py +++ b/aliby/pipeline.py @@ -263,6 +263,9 @@ class Pipeline(ProcessABC): ) bwriter = BabyWriter(filename) + # State Writer to recover interrupted experiments + swriter = StateWriter(filename) + # Limit extraction parameters during run using the available channels in tiler av_channels = set((*tiler.channels, "general")) config["extraction"]["tree"] = { @@ -316,6 +319,10 @@ class Pipeline(ProcessABC): # ) t = perf_counter() bwriter.write(seg, overwrite=["mother_assign"]) + swriter.write( + data=runner.crawler.tracker_states, + overwrite=swriter.datatypes.keys(), + ) logging.debug(f"Timing:Writing-baby:{perf_counter() - t}s") # TODO add time-skipping for cases when the