[Platform] Return result of drain and termination callbacks (#60)

nuclio · Jan 14, 2024 · a03e92d · a03e92d
1 parent d153f3c
commit a03e92d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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()