From 7ca32b229f2bb757770e74413154021d7f708666 Mon Sep 17 00:00:00 2001 From: maffettone Date: Thu, 11 Apr 2024 16:59:33 -0700 Subject: [PATCH] fix: syntax error on stop in run bundle --- pdf_agents/agents.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pdf_agents/agents.py b/pdf_agents/agents.py index ffb9421..d86783f 100644 --- a/pdf_agents/agents.py +++ b/pdf_agents/agents.py @@ -400,9 +400,10 @@ def start(self, *args, **kwargs): super().start(*args, **kwargs) self._report_producer("start", self._compose_run_bundle.start_doc) - def stop(self, *args, **kwargs): - super().stop(*args, **kwargs) - self._report_producer("stop", self._compose_run_bundle.stop_doc) + def stop(self, exit_status="success", reason=""): + super().stop(exit_status, reason) + stop_doc = self._compose_run_bundle.compose_stop(exit_status=exit_status, reason=reason) + self._report_producer("stop", stop_doc) def _write_event(self, stream, doc, uid=None): """Add event to builder as event page, and publish to catalog