snowflake-ml-python 1.14.0__py3-none-any.whl → 1.16.0__py3-none-any.whl

snowflake/ml/modeling/neighbors/kernel_density.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/local_outlier_factor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/nearest_centroid.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/nearest_neighbors.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/neighborhood_components_analysis.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/radius_neighbors_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/radius_neighbors_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neural_network/bernoulli_rbm.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neural_network/mlp_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neural_network/mlp_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/preprocessing/polynomial_features.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/semi_supervised/label_propagation.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/semi_supervised/label_spreading.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/svm/linear_svc.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/svm/linear_svr.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/svm/nu_svc.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/svm/nu_svr.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/svm/svc.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/svm/svr.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/tree/decision_tree_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/tree/decision_tree_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/tree/extra_tree_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/tree/extra_tree_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/xgboost/xgb_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/xgboost/xgb_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/xgboost/xgbrf_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/xgboost/xgbrf_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.8')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/registry/_manager/model_manager.py

@@ -1,8 +1,8 @@
+import logging
 from types import ModuleType
 from typing import TYPE_CHECKING, Any, Optional, Union
 
 import pandas as pd
-from absl.logging import logging
 
 from snowflake.ml._internal import platform_capabilities, telemetry
 from snowflake.ml._internal.exceptions import error_codes, exceptions
@@ -221,6 +221,7 @@ class ModelManager:
             )
 
         reconciler = model_parameter_reconciler.ModelParameterReconciler(
+            model=model,
             session=self._model_ops._session,
             database_name=self._database_name,
             schema_name=self._schema_name,

snowflake/ml/registry/_manager/model_parameter_reconciler.py

@@ -1,8 +1,8 @@
+import logging
 import warnings
 from dataclasses import dataclass
 from typing import Any, Optional
 
-from absl.logging import logging
 from packaging import requirements
 
 from snowflake.ml import version as snowml_version
@@ -11,6 +11,7 @@ from snowflake.ml._internal.exceptions import error_codes, exceptions
 from snowflake.ml._internal.utils import sql_identifier
 from snowflake.ml.model import target_platform, type_hints as model_types
 from snowflake.ml.model._model_composer.model_manifest import model_manifest_schema
+from snowflake.ml.model.volatility import DEFAULT_VOLATILITY_BY_MODEL_TYPE, Volatility
 from snowflake.snowpark import Session
 from snowflake.snowpark._internal import utils as snowpark_utils
 
@@ -34,6 +35,7 @@ class ModelParameterReconciler:
 
     def __init__(
         self,
+        model: model_types.SupportedModelType,
         session: Session,
         database_name: sql_identifier.SqlIdentifier,
         schema_name: sql_identifier.SqlIdentifier,
@@ -45,6 +47,7 @@ class ModelParameterReconciler:
         python_version: Optional[str] = None,
         statement_params: Optional[dict[str, str]] = None,
     ) -> None:
+        self._model = model
         self._session = session
         self._database_name = database_name
         self._schema_name = schema_name
@@ -67,6 +70,7 @@ class ModelParameterReconciler:
         reconciled_target_platforms = self._reconcile_target_platforms()
         reconciled_options = self._reconcile_explainability_options(reconciled_target_platforms)
         reconciled_options = self._reconcile_relax_version(reconciled_options, reconciled_target_platforms)
+        reconciled_options = self._reconcile_volatility_defaults(reconciled_options)  # ADD THIS LINE
 
         return ReconciledParameters(
             conda_dependencies=self._conda_dependencies,
@@ -221,7 +225,7 @@ class ModelParameterReconciler:
             ).get(env_utils.SNOWPARK_ML_PKG_NAME, [])
 
             if len(snowml_matched_versions) < 1 and not options.get("embed_local_ml_library", False):
-                logging.info(
+                logger.info(
                     f"Local snowflake-ml-python library has version {snowml_version.VERSION},"
                     " which is not available in the Snowflake server, embedding local ML library automatically."
                 )
@@ -292,3 +296,26 @@ class ModelParameterReconciler:
             )
 
         return options
+
+    def _get_default_volatility_for_model(self, model: model_types.SupportedModelType) -> Volatility:
+        """Get default volatility for a model based on its type."""
+        from snowflake.ml.model._packager import model_handler
+
+        handler = model_handler.find_handler(model)
+        # default to IMMUTABLE if no handler found or handler type not in defaults
+        if not handler or handler.HANDLER_TYPE not in DEFAULT_VOLATILITY_BY_MODEL_TYPE:
+            return Volatility.IMMUTABLE
+        return DEFAULT_VOLATILITY_BY_MODEL_TYPE[handler.HANDLER_TYPE]
+
+    def _reconcile_volatility_defaults(self, options: model_types.ModelSaveOption) -> model_types.ModelSaveOption:
+        """Set global default volatility based on model type."""
+
+        # Skip if default_volatility is already explicitly set
+        if "volatility" in options:
+            return options
+
+        # Get default volatility for this model type
+        default_volatility = self._get_default_volatility_for_model(self._model)
+        options["volatility"] = default_volatility
+
+        return options

snowflake/ml/registry/registry.py

@@ -235,6 +235,10 @@ class Registry:
                     - max_batch_size: Maximum batch size that the method could accept in the Snowflake Warehouse.
                         Defaults to None, determined automatically by Snowflake.
                     - function_type: One of supported model method function types (FUNCTION or TABLE_FUNCTION).
+                    - volatility: Volatility level for the function (use Volatility.VOLATILE or Volatility.IMMUTABLE).
+                        Volatility.VOLATILE functions may return different results for the same arguments, while
+                        Volatility.IMMUTABLE functions always return the same result for the same arguments.
+                        Defaults to None (no volatility specified).
         Returns:
             ModelVersion: ModelVersion object corresponding to the model just logged.
         """
@@ -366,6 +370,12 @@ class Registry:
                     Warehouse. It detects any ==x.y.z in specifiers and replaced with >=x.y, <(x+1). Defaults to True.
                 - function_type: Set the method function type globally. To set method function types individually see
                   function_type in model_options.
+                - volatility: Set the volatility for all model methods globally (use Volatility.VOLATILE or
+                  Volatility.IMMUTABLE). Volatility.VOLATILE functions may return different results for the same
+                  arguments, while Volatility.IMMUTABLE functions always return the same result for the same arguments.
+                  Defaults are set automatically based on model type: supported models (sklearn, xgboost, pytorch,
+                  huggingface_pipeline, mlflow, etc.) default to IMMUTABLE, while custom models default to VOLATILE.
+                  Individual method volatility can be set in method_options and will override this global setting.
                 - target_methods: List of target methods to register when logging the model.
                   This option is not used in MLFlow models. Defaults to None, in which case the model handler's
                   default target methods will be used.
@@ -382,6 +392,11 @@ class Registry:
                     - max_batch_size: Maximum batch size that the method could accept in the Snowflake Warehouse.
                         Defaults to None, determined automatically by Snowflake.
                     - function_type: One of supported model method function types (FUNCTION or TABLE_FUNCTION).
+                    - volatility: Volatility level for the function (use Volatility.VOLATILE or Volatility.IMMUTABLE).
+                        Volatility.VOLATILE functions may return different results for the same arguments, while
+                        Volatility.IMMUTABLE functions always return the same result for the same arguments.
+                        This per-method setting overrides any global volatility setting.
+                        Defaults to None (no volatility specified).
 
         Raises:
             ValueError: If extra arguments are specified ModelVersion is provided.

snowflake/ml/utils/authentication.py

@@ -73,3 +73,19 @@ class SnowflakeJWTTokenAuth(auth.AuthBase):
     def __call__(self, r: requests.PreparedRequest) -> requests.PreparedRequest:
         r.headers["Authorization"] = f'Snowflake Token="{self.snowflake_token}"'
         return r
+
+
+class SnowflakePATAuth(auth.AuthBase):
+    """Authentication using Snowflake Programmatic Access Token (PAT)."""
+
+    def __init__(self, pat_token: str) -> None:
+        """Initialize with a PAT token.
+
+        Args:
+            pat_token: The programmatic access token string
+        """
+        self.pat_token = pat_token
+
+    def __call__(self, r: requests.PreparedRequest) -> requests.PreparedRequest:
+        r.headers["Authorization"] = f'Snowflake Token="{self.pat_token}"'
+        return r

snowflake/ml/utils/connection_params.py

@@ -1,13 +1,15 @@
 import configparser
+import logging
 import os
 from typing import Optional, Union
 
-from absl import logging
 from cryptography.hazmat import backends
 from cryptography.hazmat.primitives import serialization
 
 from snowflake import snowpark
 
+logger = logging.getLogger(__name__)
+
 _DEFAULT_CONNECTION_FILE = "~/.snowsql/config"
 
 
@@ -108,7 +110,7 @@ def _load_from_snowsql_config_file(connection_name: str, login_file: str = "") -
     """Loads the dictionary from snowsql config file."""
     snowsql_config_file = login_file if login_file else os.path.expanduser(_DEFAULT_CONNECTION_FILE)
     if not os.path.exists(snowsql_config_file):
-        logging.error(f"Connection name given but snowsql config file is not found at: {snowsql_config_file}")
+        logger.error(f"Connection name given but snowsql config file is not found at: {snowsql_config_file}")
         raise Exception("Snowflake SnowSQL config not found.")
 
     config = configparser.ConfigParser(inline_comment_prefixes="#")
@@ -124,7 +126,7 @@ def _load_from_snowsql_config_file(connection_name: str, login_file: str = "") -
         # See https://docs.snowflake.com/en/user-guide/snowsql-start.html#configuring-default-connection-settings
         connection_name = "connections"
 
-    logging.info(f"Reading {snowsql_config_file} for connection parameters defined as {connection_name}")
+    logger.info(f"Reading {snowsql_config_file} for connection parameters defined as {connection_name}")
     config.read(snowsql_config_file)
     conn_params = dict(config[connection_name])
     # Remap names to appropriate args in Python Connector API

snowflake/ml/version.py

@@ -1,2 +1,2 @@
 # This is parsed by regex in conda recipe meta file. Make sure not to break it.
-VERSION = "1.14.0"
+VERSION = "1.16.0"