From a03e92dfc9befe28fd569aff362a183c54845823 Mon Sep 17 00:00:00 2001 From: Gal Topper Date: Sun, 14 Jan 2024 21:18:40 +0800 Subject: [PATCH] [Platform] Return result of drain and termination callbacks (#60) --- nuclio_sdk/platform.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nuclio_sdk/platform.py b/nuclio_sdk/platform.py index acd2537..5f966d1 100644 --- a/nuclio_sdk/platform.py +++ b/nuclio_sdk/platform.py @@ -151,6 +151,6 @@ def _on_signal(self, callback_type="termination"): :arg callback_type:str - callback type, can be "termination" or "drain" """ if callback_type == "termination" and self._termination_callback: - self._termination_callback() + return self._termination_callback() elif callback_type == "drain" and self._drain_callback: - self._drain_callback() + return self._drain_callback()