snowflake-ml-python 1.14.0__tar.gz → 1.16.0__tar.gz

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.16.0}/CHANGELOG.md

@@ -1,6 +1,55 @@
 # Release History
 
-## 1.14.0
+## 1.16.0
+
+### Bug Fixes
+
+* Registry: Remove redundant pip dependency warnings when `artifact_repository_map` is provided for warehouse model deployments.
+
+### Behavior Changes
+
+### New Features
+
+* Support scikit-learn < 1.8.
+* ML Job: Added support for configuring the runtime image via `runtime_environment`
+  (image tag or full image URL) at submission time.
+  Examples:
+  * @remote(compute_pool, stage_name = 'payload_stage', runtime_environment = '1.8.0')
+  * submit_file('/path/to/repo/test.py', compute_pool, stage_name = 'payload_stage', runtime_environment = '/mydb/myschema/myrepo/myimage:latest')
+* Registry: Ability to mark model methods as `Volatility.VOLATILE` or `Volatility.IMMUTABLE`.
+
+```python
+from snowflake.ml.model.volatility import Volatility
+
+options = {
+    "embed_local_ml_library": True,
+    "relax_version": True,
+    "save_location": "/path/to/my/directory",
+    "function_type": "TABLE_FUNCTION",
+    "volatility": Volatility.IMMUTABLE,
+    "method_options": {
+        "predict": {
+            "case_sensitive": False,
+            "max_batch_size": 100,
+            "function_type": "TABLE_FUNCTION",
+            "volatility": Volatility.VOLATILE,
+        },
+}
+
+````
+
+## 1.15.0 (09-29-2025)
+
+### Bug Fixes
+
+### Behavior Changes
+
+* Registry: Dropping support for deprecated `conversational` task type for Huggingface models.
+  To read more <https://github.com/huggingface/transformers/pull/31165>
+
+### New Features
+
+## 1.14.0 (09-18-2025)
 
 ### Bug Fixes
 

{snowflake_ml_python-1.14.0/snowflake_ml_python.egg-info → snowflake_ml_python-1.16.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.14.0
+Version: 1.16.0
 Summary: The machine learning client library that is used for interacting with Snowflake to build machine learning solutions.
 Author-email: "Snowflake, Inc" <support@snowflake.com>
 License: 
@@ -233,7 +233,6 @@ Classifier: Topic :: Scientific/Engineering :: Information Analysis
 Requires-Python: <3.13,>=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
-Requires-Dist: absl-py<2,>=0.15
 Requires-Dist: anyio<5,>=3.5.0
 Requires-Dist: cachetools<6,>=3.1.1
 Requires-Dist: cloudpickle>=2.0.0
@@ -251,7 +250,7 @@ Requires-Dist: pytimeparse<2,>=1.1.8
 Requires-Dist: pyyaml<7,>=6.0
 Requires-Dist: retrying<2,>=1.3.3
 Requires-Dist: s3fs<2026,>=2024.6.1
-Requires-Dist: scikit-learn<1.7
+Requires-Dist: scikit-learn<1.8
 Requires-Dist: scipy<2,>=1.9
 Requires-Dist: shap<1,>=0.46.0
 Requires-Dist: snowflake-connector-python[pandas]<4,>=3.16.0
@@ -302,21 +301,21 @@ Requires-Dist: torch<3,>=2.0.1; extra == "transformers"
 Requires-Dist: transformers!=4.51.3,<5,>=4.39.3; extra == "transformers"
 Dynamic: license-file
 
-# Snowpark ML
+# Snowflake ML Python
 
-Snowpark ML is a set of tools including SDKs and underlying infrastructure to build and deploy machine learning models.
-With Snowpark ML, you can pre-process data, train, manage and deploy ML models all within Snowflake, using a single SDK,
+Snowflake ML Python is a set of tools including SDKs and underlying infrastructure to build and deploy machine learning models.
+With Snowflake ML Python, you can pre-process data, train, manage and deploy ML models all within Snowflake,
 and benefit from Snowflake’s proven performance, scalability, stability and governance at every stage of the Machine
 Learning workflow.
 
-## Key Components of Snowpark ML
+## Key Components of Snowflake ML Python
 
-The Snowpark ML Python SDK provides a number of APIs to support each stage of an end-to-end Machine Learning development
-and deployment process, and includes two key components.
+The Snowflake ML Python SDK provides a number of APIs to support each stage of an end-to-end Machine Learning development
+and deployment process.
 
-### Snowpark ML Development
+### Snowflake ML Model Development
 
-[Snowpark ML Development](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#ml-modeling)
+[Snowflake ML Model Development](https://docs.snowflake.com/developer-guide/snowflake-ml/overview#ml-modeling)
 provides a collection of python APIs enabling efficient ML model development directly in Snowflake:
 
 1. Modeling API (`snowflake.ml.modeling`) for data preprocessing, feature engineering and model training in Snowflake.
@@ -327,19 +326,16 @@ model development classes based on sklearn, xgboost, and lightgbm.
 1. Framework Connectors: Optimized, secure and performant data provisioning for Pytorch and Tensorflow frameworks in
 their native data loader formats.
 
-1. FileSet API: FileSet provides a Python fsspec-compliant API for materializing data into a Snowflake internal stage
-from a query or Snowpark Dataframe along with a number of convenience APIs.
+### Snowflake ML Ops
 
-### Snowflake MLOps
+Snowflake ML Python contains a suite of MLOps tools. It complements
+the Snowflake Modeling API, and provides end to end development to deployment within Snowflake.
+The Snowflake ML Ops suite consists of:
 
-Snowflake MLOps contains suit of tools and objects to make ML development cycle. It complements
-the Snowpark ML Development API, and provides end to end development to deployment within Snowflake.
-Currently, the API consists of:
-
-1. [Registry](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowflake-model-registry): A python API
+1. [Registry](https://docs.snowflake.com/developer-guide/snowflake-ml/overview#snowflake-model-registry): A python API
   allows secure deployment and management of models in Snowflake, supporting models trained both inside and outside of
   Snowflake.
-2. [Feature Store](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowflake-feature-store): A fully
+2. [Feature Store](https://docs.snowflake.com/developer-guide/snowflake-ml/overview#snowflake-feature-store): A fully
   integrated solution for defining, managing, storing and discovering ML features derived from your data. The
   Snowflake Feature Store supports automated, incremental refresh from batch and streaming data sources, so that
   feature pipelines need be defined only once to be continuously updated with new data.
@@ -348,12 +344,19 @@ Currently, the API consists of:
 
 ## Getting started
 
+Learn about all Snowflake ML feature offerings in the [Developer Guide](https://docs.snowflake.com/developer-guide/snowflake-ml/overview).
+
 ### Have your Snowflake account ready
 
 If you don't have a Snowflake account yet, you can [sign up for a 30-day free trial account](https://signup.snowflake.com/).
 
 ### Installation
 
+Snowflake ML Python is pre-installed in Container Runtime notebook environments.
+[Learn more](https://docs.snowflake.com/en/developer-guide/snowflake-ml/notebooks-on-spcs).
+
+In Snowflake Warehouse notebook environments, snowflake-ml-python can be installed using the "Packages" drop-down menu.
+
 Follow the [installation instructions](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#installing-snowpark-ml)
 in the Snowflake documentation.
 
@@ -363,8 +366,8 @@ or [virtualenv](https://docs.python.org/3/tutorial/venv.html) to create a virtua
 
 ### Conda channels
 
-The [Snowflake Conda Channel](https://repo.anaconda.com/pkgs/snowflake/) contains the official snowpark ML package releases.
-The recommended approach is to install `snowflake-ml-python` this conda channel:
+The [Snowflake Anaconda Channel](https://repo.anaconda.com/pkgs/snowflake/) contains the official snowflake-ml-python package
+releases. To install `snowflake-ml-python` from this conda channel:
 
 ```sh
 conda install \
@@ -373,25 +376,18 @@ conda install \
   snowflake-ml-python
 ```
 
-See [the developer guide](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index) for installation instructions.
+See [the developer guide](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index) for detailed installation instructions.
 
-The latest version of the `snowpark-ml-python` package is also published in a conda channel in this repository. Package versions
-in this channel may not yet be present in the official Snowflake conda channel.
+The snowflake-ml-python package is also published in [conda-forge](https://anaconda.org/conda-forge/snowflake-ml-python).
+To install `snowflake-ml-python` from conda forge:
 
-Install `snowflake-ml-python` from this channel with the following (being sure to replace `<version_specifier>` with the
-desired version, e.g. `1.0.10`):
-
-```bash
+```sh
 conda install \
-  -c https://raw.githubusercontent.com/snowflakedb/snowflake-ml-python/conda/releases/  \
-  -c https://repo.anaconda.com/pkgs/snowflake \
+  -c https://conda.anaconda.org/conda-forge/ \
   --override-channels \
-  snowflake-ml-python==<version_specifier>
+  snowflake-ml-python
 ```
 
-Note that until a `snowflake-ml-python` package version is available in the official Snowflake conda channel, there may
-be compatibility issues. Server-side functionality that `snowflake-ml-python` depends on may not yet be released.
-
 ### Verifying the package
 
 1. Install cosign.
@@ -410,7 +406,56 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.14.0
+## 1.16.0
+
+### Bug Fixes
+
+* Registry: Remove redundant pip dependency warnings when `artifact_repository_map` is provided for warehouse model deployments.
+
+### Behavior Changes
+
+### New Features
+
+* Support scikit-learn < 1.8.
+* ML Job: Added support for configuring the runtime image via `runtime_environment`
+  (image tag or full image URL) at submission time.
+  Examples:
+  * @remote(compute_pool, stage_name = 'payload_stage', runtime_environment = '1.8.0')
+  * submit_file('/path/to/repo/test.py', compute_pool, stage_name = 'payload_stage', runtime_environment = '/mydb/myschema/myrepo/myimage:latest')
+* Registry: Ability to mark model methods as `Volatility.VOLATILE` or `Volatility.IMMUTABLE`.
+
+```python
+from snowflake.ml.model.volatility import Volatility
+
+options = {
+    "embed_local_ml_library": True,
+    "relax_version": True,
+    "save_location": "/path/to/my/directory",
+    "function_type": "TABLE_FUNCTION",
+    "volatility": Volatility.IMMUTABLE,
+    "method_options": {
+        "predict": {
+            "case_sensitive": False,
+            "max_batch_size": 100,
+            "function_type": "TABLE_FUNCTION",
+            "volatility": Volatility.VOLATILE,
+        },
+}
+
+````
+
+## 1.15.0 (09-29-2025)
+
+### Bug Fixes
+
+### Behavior Changes
+
+* Registry: Dropping support for deprecated `conversational` task type for Huggingface models.
+  To read more <https://github.com/huggingface/transformers/pull/31165>
+
+### New Features
+
+## 1.14.0 (09-18-2025)
 
 ### Bug Fixes
 

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.16.0}/README.md

@@ -1,18 +1,18 @@
-# Snowpark ML
+# Snowflake ML Python
 
-Snowpark ML is a set of tools including SDKs and underlying infrastructure to build and deploy machine learning models.
-With Snowpark ML, you can pre-process data, train, manage and deploy ML models all within Snowflake, using a single SDK,
+Snowflake ML Python is a set of tools including SDKs and underlying infrastructure to build and deploy machine learning models.
+With Snowflake ML Python, you can pre-process data, train, manage and deploy ML models all within Snowflake,
 and benefit from Snowflake’s proven performance, scalability, stability and governance at every stage of the Machine
 Learning workflow.
 
-## Key Components of Snowpark ML
+## Key Components of Snowflake ML Python
 
-The Snowpark ML Python SDK provides a number of APIs to support each stage of an end-to-end Machine Learning development
-and deployment process, and includes two key components.
+The Snowflake ML Python SDK provides a number of APIs to support each stage of an end-to-end Machine Learning development
+and deployment process.
 
-### Snowpark ML Development
+### Snowflake ML Model Development
 
-[Snowpark ML Development](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#ml-modeling)
+[Snowflake ML Model Development](https://docs.snowflake.com/developer-guide/snowflake-ml/overview#ml-modeling)
 provides a collection of python APIs enabling efficient ML model development directly in Snowflake:
 
 1. Modeling API (`snowflake.ml.modeling`) for data preprocessing, feature engineering and model training in Snowflake.
@@ -23,19 +23,16 @@ model development classes based on sklearn, xgboost, and lightgbm.
 1. Framework Connectors: Optimized, secure and performant data provisioning for Pytorch and Tensorflow frameworks in
 their native data loader formats.
 
-1. FileSet API: FileSet provides a Python fsspec-compliant API for materializing data into a Snowflake internal stage
-from a query or Snowpark Dataframe along with a number of convenience APIs.
+### Snowflake ML Ops
 
-### Snowflake MLOps
+Snowflake ML Python contains a suite of MLOps tools. It complements
+the Snowflake Modeling API, and provides end to end development to deployment within Snowflake.
+The Snowflake ML Ops suite consists of:
 
-Snowflake MLOps contains suit of tools and objects to make ML development cycle. It complements
-the Snowpark ML Development API, and provides end to end development to deployment within Snowflake.
-Currently, the API consists of:
-
-1. [Registry](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowflake-model-registry): A python API
+1. [Registry](https://docs.snowflake.com/developer-guide/snowflake-ml/overview#snowflake-model-registry): A python API
   allows secure deployment and management of models in Snowflake, supporting models trained both inside and outside of
   Snowflake.
-2. [Feature Store](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowflake-feature-store): A fully
+2. [Feature Store](https://docs.snowflake.com/developer-guide/snowflake-ml/overview#snowflake-feature-store): A fully
   integrated solution for defining, managing, storing and discovering ML features derived from your data. The
   Snowflake Feature Store supports automated, incremental refresh from batch and streaming data sources, so that
   feature pipelines need be defined only once to be continuously updated with new data.
@@ -44,12 +41,19 @@ Currently, the API consists of:
 
 ## Getting started
 
+Learn about all Snowflake ML feature offerings in the [Developer Guide](https://docs.snowflake.com/developer-guide/snowflake-ml/overview).
+
 ### Have your Snowflake account ready
 
 If you don't have a Snowflake account yet, you can [sign up for a 30-day free trial account](https://signup.snowflake.com/).
 
 ### Installation
 
+Snowflake ML Python is pre-installed in Container Runtime notebook environments.
+[Learn more](https://docs.snowflake.com/en/developer-guide/snowflake-ml/notebooks-on-spcs).
+
+In Snowflake Warehouse notebook environments, snowflake-ml-python can be installed using the "Packages" drop-down menu.
+
 Follow the [installation instructions](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#installing-snowpark-ml)
 in the Snowflake documentation.
 
@@ -59,8 +63,8 @@ or [virtualenv](https://docs.python.org/3/tutorial/venv.html) to create a virtua
 
 ### Conda channels
 
-The [Snowflake Conda Channel](https://repo.anaconda.com/pkgs/snowflake/) contains the official snowpark ML package releases.
-The recommended approach is to install `snowflake-ml-python` this conda channel:
+The [Snowflake Anaconda Channel](https://repo.anaconda.com/pkgs/snowflake/) contains the official snowflake-ml-python package
+releases. To install `snowflake-ml-python` from this conda channel:
 
 ```sh
 conda install \
@@ -69,25 +73,18 @@ conda install \
   snowflake-ml-python
 ```
 
-See [the developer guide](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index) for installation instructions.
-
-The latest version of the `snowpark-ml-python` package is also published in a conda channel in this repository. Package versions
-in this channel may not yet be present in the official Snowflake conda channel.
+See [the developer guide](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index) for detailed installation instructions.
 
-Install `snowflake-ml-python` from this channel with the following (being sure to replace `<version_specifier>` with the
-desired version, e.g. `1.0.10`):
+The snowflake-ml-python package is also published in [conda-forge](https://anaconda.org/conda-forge/snowflake-ml-python).
+To install `snowflake-ml-python` from conda forge:
 
-```bash
+```sh
 conda install \
-  -c https://raw.githubusercontent.com/snowflakedb/snowflake-ml-python/conda/releases/  \
-  -c https://repo.anaconda.com/pkgs/snowflake \
+  -c https://conda.anaconda.org/conda-forge/ \
   --override-channels \
-  snowflake-ml-python==<version_specifier>
+  snowflake-ml-python
 ```
 
-Note that until a `snowflake-ml-python` package version is available in the official Snowflake conda channel, there may
-be compatibility issues. Server-side functionality that `snowflake-ml-python` depends on may not yet be released.
-
 ### Verifying the package
 
 1. Install cosign.

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.16.0}/pyproject.toml

@@ -8,7 +8,7 @@ description = "The machine learning client library that is used for interacting
 classifiers = [ "Development Status :: 5 - Production/Stable", "Environment :: Console", "Environment :: Other Environment", "Intended Audience :: Developers", "Intended Audience :: Education", "Intended Audience :: Information Technology", "Intended Audience :: System Administrators", "License :: OSI Approved :: Apache Software License", "Operating System :: OS Independent", "Programming Language :: Python :: 3.9", "Programming Language :: Python :: 3.10", "Programming Language :: Python :: 3.11", "Programming Language :: Python :: 3.12", "Topic :: Database", "Topic :: Software Development", "Topic :: Software Development :: Libraries", "Topic :: Software Development :: Libraries :: Application Frameworks", "Topic :: Software Development :: Libraries :: Python Modules", "Topic :: Scientific/Engineering :: Information Analysis",]
 requires-python = ">=3.9, <3.13"
 dynamic = [ "version", "readme",]
-dependencies = [ "absl-py>=0.15,<2", "anyio>=3.5.0,<5", "cachetools>=3.1.1,<6", "cloudpickle>=2.0.0", "cryptography", "fsspec[http]>=2024.6.1,<2026", "importlib_resources>=6.1.1, <7", "numpy>=1.23,<3", "packaging>=20.9,<25", "pandas>=2.1.4,<3", "platformdirs<5", "pyarrow<19.0.0", "pydantic>=2.8.2, <3", "pyjwt>=2.0.0, <3", "pytimeparse>=1.1.8,<2", "pyyaml>=6.0,<7", "retrying>=1.3.3,<2", "s3fs>=2024.6.1,<2026", "scikit-learn<1.7", "scipy>=1.9,<2", "shap>=0.46.0,<1", "snowflake-connector-python[pandas]>=3.16.0,<4", "snowflake-snowpark-python>=1.17.0,<2,!=1.26.0", "snowflake.core>=1.0.2,<2", "sqlparse>=0.4,<1", "tqdm<5", "typing-extensions>=4.1.0,<5", "xgboost>=1.7.3,<3",]
+dependencies = [ "anyio>=3.5.0,<5", "cachetools>=3.1.1,<6", "cloudpickle>=2.0.0", "cryptography", "fsspec[http]>=2024.6.1,<2026", "importlib_resources>=6.1.1, <7", "numpy>=1.23,<3", "packaging>=20.9,<25", "pandas>=2.1.4,<3", "platformdirs<5", "pyarrow<19.0.0", "pydantic>=2.8.2, <3", "pyjwt>=2.0.0, <3", "pytimeparse>=1.1.8,<2", "pyyaml>=6.0,<7", "retrying>=1.3.3,<2", "s3fs>=2024.6.1,<2026", "scikit-learn<1.8", "scipy>=1.9,<2", "shap>=0.46.0,<1", "snowflake-connector-python[pandas]>=3.16.0,<4", "snowflake-snowpark-python>=1.17.0,<2,!=1.26.0", "snowflake.core>=1.0.2,<2", "sqlparse>=0.4,<1", "tqdm<5", "typing-extensions>=4.1.0,<5", "xgboost>=1.7.3,<3",]
 [[project.authors]]
 name = "Snowflake, Inc"
 email = "support@snowflake.com"

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.16.0}/snowflake/ml/_internal/platform_capabilities.py

@@ -1,8 +1,8 @@
 import json
+import logging
 from contextlib import contextmanager
 from typing import Any, Optional
 
-from absl import logging
 from packaging import version
 
 from snowflake.ml import version as snowml_version
@@ -13,8 +13,11 @@ from snowflake.snowpark import (
     session as snowpark_session,
 )
 
+logger = logging.getLogger(__name__)
+
 LIVE_COMMIT_PARAMETER = "ENABLE_LIVE_VERSION_IN_SDK"
 INLINE_DEPLOYMENT_SPEC_PARAMETER = "ENABLE_INLINE_DEPLOYMENT_SPEC_FROM_CLIENT_VERSION"
+SET_MODULE_FUNCTIONS_VOLATILITY_FROM_MANIFEST = "SET_MODULE_FUNCTIONS_VOLATILITY_FROM_MANIFEST"
 
 
 class PlatformCapabilities:
@@ -60,17 +63,20 @@ class PlatformCapabilities:
     @classmethod  # type: ignore[arg-type]
     @contextmanager
     def mock_features(cls, features: dict[str, Any] = _dummy_features) -> None:  # type: ignore[misc]
-        logging.debug(f"Setting mock features: {features}")
+        logger.debug(f"Setting mock features: {features}")
         cls.set_mock_features(features)
         try:
             yield
         finally:
-            logging.debug(f"Clearing mock features: {features}")
+            logger.debug(f"Clearing mock features: {features}")
             cls.clear_mock_features()
 
     def is_inlined_deployment_spec_enabled(self) -> bool:
         return self._is_version_feature_enabled(INLINE_DEPLOYMENT_SPEC_PARAMETER)
 
+    def is_set_module_functions_volatility_from_manifest(self) -> bool:
+        return self._get_bool_feature(SET_MODULE_FUNCTIONS_VOLATILITY_FROM_MANIFEST, False)
+
     def is_live_commit_enabled(self) -> bool:
         return self._get_bool_feature(LIVE_COMMIT_PARAMETER, False)
 
@@ -98,7 +104,7 @@ class PlatformCapabilities:
                         error_code=error_codes.INTERNAL_SNOWML_ERROR, original_exception=RuntimeError(message)
                     )
         except snowpark_exceptions.SnowparkSQLException as e:
-            logging.debug(f"Failed to retrieve platform capabilities: {e}")
+            logger.debug(f"Failed to retrieve platform capabilities: {e}")
             # This can happen is server side is older than 9.2. That is fine.
         return {}
 
@@ -144,7 +150,7 @@ class PlatformCapabilities:
 
         value = self.features.get(feature_name)
         if value is None:
-            logging.debug(f"Feature {feature_name} not found, returning large version number")
+            logger.debug(f"Feature {feature_name} not found, returning large version number")
             return large_version
 
         try:
@@ -152,7 +158,7 @@ class PlatformCapabilities:
             version_str = str(value)
             return version.Version(version_str)
         except (version.InvalidVersion, ValueError, TypeError) as e:
-            logging.debug(
+            logger.debug(
                 f"Failed to parse version from feature {feature_name} with value '{value}': {e}. "
                 f"Returning large version number"
             )
@@ -171,7 +177,7 @@ class PlatformCapabilities:
         feature_version = self._get_version_feature(feature_name)
 
         result = current_version >= feature_version
-        logging.debug(
+        logger.debug(
             f"Version comparison for feature {feature_name}: "
             f"current={current_version}, feature={feature_version}, enabled={result}"
         )

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.16.0}/snowflake/ml/_internal/utils/connection_params.py

@@ -1,11 +1,13 @@
 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
 
+logger = logging.getLogger(__name__)
+
 _DEFAULT_CONNECTION_FILE = "~/.snowsql/config"
 
 
@@ -106,7 +108,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="#")
@@ -122,7 +124,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_python-1.14.0 → snowflake_ml_python-1.16.0}/snowflake/ml/_internal/utils/jwt_generator.py

@@ -110,15 +110,16 @@ class JWTGenerator:
         }
 
         # Regenerate the actual token
-        token = jwt.encode(payload, key=self.private_key, algorithm=JWTGenerator.ALGORITHM)
+        token = jwt.encode(payload, key=self.private_key, algorithm=JWTGenerator.ALGORITHM)  # type: ignore[arg-type]
         # If you are using a version of PyJWT prior to 2.0, jwt.encode returns a byte string instead of a string.
         # If the token is a byte string, convert it to a string.
         if isinstance(token, bytes):
             token = token.decode("utf-8")
         self.token = token
+        public_key = self.private_key.public_key()
         logger.info(
             "Generated a JWT with the following payload: %s",
-            jwt.decode(self.token, key=self.private_key.public_key(), algorithms=[JWTGenerator.ALGORITHM]),
+            jwt.decode(self.token, key=public_key, algorithms=[JWTGenerator.ALGORITHM]),  # type: ignore[arg-type]
         )
 
         return token

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.16.0}/snowflake/ml/_internal/utils/mixins.py

@@ -1,3 +1,4 @@
+from dataclasses import dataclass
 from typing import Any, Optional
 
 from snowflake.ml._internal.utils import identifier
@@ -16,6 +17,14 @@ def _identifiers_match(saved: Optional[str], current: Optional[str]) -> bool:
     return saved_resolved == current_resolved
 
 
+@dataclass(frozen=True)
+class _SessionState:
+    account: Optional[str]
+    role: Optional[str]
+    database: Optional[str]
+    schema: Optional[str]
+
+
 class SerializableSessionMixin:
     """Mixin that provides pickling capabilities for objects with Snowpark sessions."""
 
@@ -40,17 +49,23 @@ class SerializableSessionMixin:
 
     def __setstate__(self, state: dict[str, Any]) -> None:
         """Restore session from context during unpickling."""
-        saved_account = state.pop(_SESSION_ACCOUNT_KEY, None)
-        saved_role = state.pop(_SESSION_ROLE_KEY, None)
-        saved_database = state.pop(_SESSION_DATABASE_KEY, None)
-        saved_schema = state.pop(_SESSION_SCHEMA_KEY, None)
+        session_state = _SessionState(
+            account=state.pop(_SESSION_ACCOUNT_KEY, None),
+            role=state.pop(_SESSION_ROLE_KEY, None),
+            database=state.pop(_SESSION_DATABASE_KEY, None),
+            schema=state.pop(_SESSION_SCHEMA_KEY, None),
+        )
 
         if hasattr(super(), "__setstate__"):
             super().__setstate__(state)  # type: ignore[misc]
         else:
             self.__dict__.update(state)
 
-        if saved_account is not None:
+        self._set_session(session_state)
+
+    def _set_session(self, session_state: _SessionState) -> None:
+
+        if session_state.account is not None:
             active_sessions = snowpark_session._get_active_sessions()
             if len(active_sessions) == 0:
                 raise RuntimeError("No active Snowpark session available. Please create a session.")
@@ -63,10 +78,10 @@ class SerializableSessionMixin:
                     active_sessions,
                     key=lambda s: sum(
                         (
-                            _identifiers_match(saved_account, s.get_current_account()),
-                            _identifiers_match(saved_role, s.get_current_role()),
-                            _identifiers_match(saved_database, s.get_current_database()),
-                            _identifiers_match(saved_schema, s.get_current_schema()),
+                            _identifiers_match(session_state.account, s.get_current_account()),
+                            _identifiers_match(session_state.role, s.get_current_role()),
+                            _identifiers_match(session_state.database, s.get_current_database()),
+                            _identifiers_match(session_state.schema, s.get_current_schema()),
                         )
                     ),
                 ),

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.16.0}/snowflake/ml/_internal/utils/temp_file_utils.py

@@ -1,10 +1,9 @@
+import logging
 import os
 import shutil
 import tempfile
 from typing import Iterable, Union
 
-from absl.logging import logging
-
 logger = logging.getLogger(__name__)
 
 

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.16.0}/snowflake/ml/experiment/_client/experiment_tracking_sql_client.py

@@ -76,17 +76,30 @@ class ExperimentTrackingSQLClient(_base._BaseSQLClient):
             self._session, f"ALTER EXPERIMENT {experiment_fqn} DROP RUN {run_name}"
         ).has_dimensions(expected_rows=1, expected_cols=1).validate()
 
-    def modify_run(
+    def modify_run_add_metrics(
         self,
         *,
         experiment_name: sql_identifier.SqlIdentifier,
         run_name: sql_identifier.SqlIdentifier,
-        run_metadata: str,
+        metrics: str,
     ) -> None:
         experiment_fqn = self.fully_qualified_object_name(self._database_name, self._schema_name, experiment_name)
         query_result_checker.SqlResultValidator(
             self._session,
-            f"ALTER EXPERIMENT {experiment_fqn} MODIFY RUN {run_name} SET METADATA=$${run_metadata}$$",
+            f"ALTER EXPERIMENT {experiment_fqn} MODIFY RUN {run_name} ADD METRICS=$${metrics}$$",
+        ).has_dimensions(expected_rows=1, expected_cols=1).validate()
+
+    def modify_run_add_params(
+        self,
+        *,
+        experiment_name: sql_identifier.SqlIdentifier,
+        run_name: sql_identifier.SqlIdentifier,
+        params: str,
+    ) -> None:
+        experiment_fqn = self.fully_qualified_object_name(self._database_name, self._schema_name, experiment_name)
+        query_result_checker.SqlResultValidator(
+            self._session,
+            f"ALTER EXPERIMENT {experiment_fqn} MODIFY RUN {run_name} ADD PARAMETERS=$${params}$$",
         ).has_dimensions(expected_rows=1, expected_cols=1).validate()
 
     def put_artifact(

snowflake_ml_python-1.16.0/snowflake/ml/experiment/_entities/__init__.py

@@ -0,0 +1,5 @@
+from snowflake.ml.experiment._entities.experiment import Experiment
+from snowflake.ml.experiment._entities.run import Run
+from snowflake.ml.experiment._entities.run_metadata import Metric, Param
+
+__all__ = ["Experiment", "Run", "Metric", "Param"]