omnata-plugin-runtime 0.3.26__tar.gz → 0.3.27a80__tar.gz

{omnata_plugin_runtime-0.3.26 → omnata_plugin_runtime-0.3.27a80}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.3.26
+Version: 0.3.27a80
 Summary: Classes and common runtime components for building and running Omnata Plugins
 Author: James Weakley
 Author-email: james.weakley@omnata.com

{omnata_plugin_runtime-0.3.26 → omnata_plugin_runtime-0.3.27a80}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.3.26"
+version = "0.3.27-a80"
 description = "Classes and common runtime components for building and running Omnata Plugins"
 authors = ["James Weakley <james.weakley@omnata.com>"]
 readme = "README.md"

{omnata_plugin_runtime-0.3.26 → omnata_plugin_runtime-0.3.27a80}/src/omnata_plugin_runtime/omnata_plugin.py

@@ -289,7 +289,7 @@ class SyncRequest(ABC):
                 self.deadline_reached = True
                 self.apply_deadline_reached()  # pylint: disable=protected-access
                 return
-
+            is_cancelled:bool = False
             with self._snowflake_query_lock:
                 try:
                     # this is not ideal, but "Bind variable in stored procedure is not supported yet"
@@ -297,11 +297,11 @@ class SyncRequest(ABC):
                         f"call {self._source_app_name}.API.PLUGIN_CANCELLATION_CHECK({self._run_id})"
                     ).collect()
                     cancellation_result = handle_proc_result(query_result)
-                    is_cancelled: bool = cancellation_result["is_cancelled"]
-                    if is_cancelled:
-                        self.apply_cancellation()
+                    is_cancelled = cancellation_result["is_cancelled"]
                 except Exception as e:
                     logger.error(f"Error checking cancellation: {e}")
+            if is_cancelled:
+                self.apply_cancellation()
             cancellation_token.wait(20)
         logger.info("cancel checking worker exiting")
 
@@ -1029,7 +1029,7 @@ class InboundSyncRequest(SyncRequest):
             # flatten
             all_dfs:List[pandas.DataFrame] = [x for sublist in all_df_lists for x in sublist]
             combined_length = sum([len(x) for x in all_dfs])
-            # first, don't both if the count is less than 10000, since it's unlikely to be even close
+            # first, don't bother if the count is less than 10000, since it's unlikely to be even close
             if combined_length > 10000:
                 if sum([x.memory_usage(index=True).sum() for x in all_dfs]) > 200000000:
                     logger.info(f"Applying results queue immediately due to combined dataframe size")