From 44aef29f12ec80e4bc911a0c2492d9509e2f7980 Mon Sep 17 00:00:00 2001 From: Philipp van Kempen Date: Thu, 9 Jan 2025 10:29:29 +0100 Subject: [PATCH] do not skip postprocess stage if run_per_stage=0 (fixes #185) --- mlonmcu/session/run.py | 4 ++-- mlonmcu/session/session.py | 3 +++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/mlonmcu/session/run.py b/mlonmcu/session/run.py index 20b869837..36a850233 100644 --- a/mlonmcu/session/run.py +++ b/mlonmcu/session/run.py @@ -1039,12 +1039,12 @@ def load(self): self.completed[RunStage.LOAD] = True self.unlock() - def process(self, until=RunStage.RUN, skip=None, export=False): + def process(self, until=RunStage.RUN, start=None, skip=None, export=False): """Process the run until a given stage.""" skip = skip if skip is not None else [] if until == RunStage.DONE: until = RunStage.DONE - 1 - start = self.next_stage # self.stage hold the max finished stage + start = self.next_stage if start is None else start if until < start: logger.debug("%s Nothing to do", self.prefix) return self.get_report() diff --git a/mlonmcu/session/session.py b/mlonmcu/session/session.py index 6c25167de..2f8fff0f3 100644 --- a/mlonmcu/session/session.py +++ b/mlonmcu/session/session.py @@ -203,6 +203,9 @@ def _close_progress(pbar): def _process(pbar, run, until, skip): """Helper function to invoke the run.""" run.process(until=until, skip=skip, export=export) + if not per_stage and run.has_stage(RunStage.POSTPROCESS) and RunStage.POSTPROCESS not in skip: + # run.postprocess() + run.process(until=RunStage.POSTPROCESS, start=RunStage.POSTPROCESS, skip=skip, export=export) if progress: _update_progress(pbar)