omnata-plugin-runtime 0.6.3a169__tar.gz → 0.6.4a171__tar.gz

{omnata_plugin_runtime-0.6.3a169 → omnata_plugin_runtime-0.6.4a171}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.6.3a169
+Version: 0.6.4a171
 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.6.3a169 → omnata_plugin_runtime-0.6.4a171}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.6.3-a169"
+version = "0.6.4-a171"
 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.6.3a169 → omnata_plugin_runtime-0.6.4a171}/src/omnata_plugin_runtime/omnata_plugin.py

@@ -2357,6 +2357,8 @@ def omnata_udtf(
                                     data_type='OBJECT',
                                     description='The connection object, obtained from calling PLUGIN.PLUGIN_CONNECTION.')] + params
             cls._omnata_udtf_params = params_new
+        else:
+            cls._omnata_udtf_params = params
         if len(cls._omnata_udtf_params) != len(function_params) -1:
             raise ValueError(f"You must document all the parameters of the 'process' function in the @omnata_udtf decorator in the same order ('connection_parameters' will be included automatically).")
 
@@ -2471,6 +2473,8 @@ def omnata_udf(
                                     data_type='OBJECT',
                                     description='The connection object, obtained from calling PLUGIN.PLUGIN_CONNECTION.')] + params
             func._omnata_udf_params = params_new
+        else:
+            func._omnata_udf_params = params
         if len(func._omnata_udf_params) != len(function_params):
             raise ValueError(f"You must document all the parameters of the function in the @omnata_udf decorator in the same order ('connection_parameters' will be included automatically).")
 

{omnata_plugin_runtime-0.6.3a169 → omnata_plugin_runtime-0.6.4a171}/src/omnata_plugin_runtime/plugin_entrypoints.py

@@ -60,8 +60,8 @@ class PluginEntrypoint:
         self._plugin_instance: OmnataPlugin = class_obj()
         self._plugin_instance._session = session  # pylint: disable=protected-access
         # logging defaults
-        logger = logging.getLogger("snowflake")
-        logger.setLevel(logging.WARN) # we don't want snowflake queries being logged by default
+        snowflake_logger = logging.getLogger("snowflake")
+        snowflake_logger.setLevel(logging.WARN) # we don't want snowflake queries being logged by default
         # the sync engine can tell the plugin to override log level via a session variable
         if session is not None:
             try:
@@ -70,10 +70,10 @@ class PluginEntrypoint:
                 if result is not None:
                     log_level_overrides:Dict[str,str] = json.loads(result)
                     for logger_name,level in log_level_overrides.items():
-                        logger = logging.getLogger(logger_name)
-                        logger.setLevel(level)
-                        logger.propagate = False
-                        for handler in logger.handlers:
+                        logger_override = logging.getLogger(logger_name)
+                        logger_override.setLevel(level)
+                        logger_override.propagate = False
+                        for handler in logger_override.handlers:
                             handler.setLevel(level)
             except Exception as e:
                 logger.error(f"Error setting log level overrides: {str(e)}")