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

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

@@ -1,6 +1,17 @@
 # Release History
 
-## 1.14.0
+## 1.15.0
+
+### 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.15.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.14.0
+Version: 1.15.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
@@ -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.
-
-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.
@@ -410,7 +406,18 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.14.0
+## 1.15.0
+
+### 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.15.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.15.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.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",]
 [[project.authors]]
 name = "Snowflake, Inc"
 email = "support@snowflake.com"

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.15.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,6 +13,8 @@ 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"
 
@@ -60,12 +62,12 @@ 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:
@@ -98,7 +100,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 +146,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 +154,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 +173,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.15.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.15.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.15.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.15.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.15.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"]

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.15.0}/snowflake/ml/experiment/_entities/run.py

@@ -1,11 +1,8 @@
-import json
 import types
 from typing import TYPE_CHECKING, Optional
 
 from snowflake.ml._internal.utils import sql_identifier
 from snowflake.ml.experiment import _experiment_info as experiment_info
-from snowflake.ml.experiment._client import experiment_tracking_sql_client
-from snowflake.ml.experiment._entities import run_metadata
 
 if TYPE_CHECKING:
     from snowflake.ml.experiment import experiment_tracking
@@ -41,18 +38,6 @@ class Run:
         if self._experiment_tracking._run is self:
             self._experiment_tracking.end_run()
 
-    def _get_metadata(
-        self,
-    ) -> run_metadata.RunMetadata:
-        runs = self._experiment_tracking._sql_client.show_runs_in_experiment(
-            experiment_name=self.experiment_name, like=str(self.name)
-        )
-        if not runs:
-            raise RuntimeError(f"Run {self.name} not found in experiment {self.experiment_name}.")
-        return run_metadata.RunMetadata.from_dict(
-            json.loads(runs[0][experiment_tracking_sql_client.ExperimentTrackingSQLClient.RUN_METADATA_COL_NAME])
-        )
-
     def _get_experiment_info(self) -> experiment_info.ExperimentInfo:
         return experiment_info.ExperimentInfo(
             fully_qualified_name=self._experiment_tracking._sql_client.fully_qualified_object_name(

snowflake_ml_python-1.15.0/snowflake/ml/experiment/_entities/run_metadata.py

@@ -0,0 +1,20 @@
+import dataclasses
+
+
+@dataclasses.dataclass
+class Metric:
+    name: str
+    value: float
+    step: int
+
+    def to_dict(self) -> dict:  # type: ignore[type-arg]
+        return dataclasses.asdict(self)
+
+
+@dataclasses.dataclass
+class Param:
+    name: str
+    value: str
+
+    def to_dict(self) -> dict:  # type: ignore[type-arg]
+        return dataclasses.asdict(self)

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.15.0}/snowflake/ml/experiment/experiment_tracking.py

@@ -261,13 +261,13 @@ class ExperimentTracking(mixins.SerializableSessionMixin):
             step: The step of the metrics. Defaults to 0.
         """
         run = self._get_or_start_run()
-        metadata = run._get_metadata()
+        metrics_list = []
         for key, value in metrics.items():
-            metadata.set_metric(key, value, step)
-        self._sql_client.modify_run(
+            metrics_list.append(entities.Metric(key, value, step))
+        self._sql_client.modify_run_add_metrics(
             experiment_name=run.experiment_name,
             run_name=run.name,
-            run_metadata=json.dumps(metadata.to_dict()),
+            metrics=json.dumps([metric.to_dict() for metric in metrics_list]),
         )
 
     def log_param(
@@ -296,13 +296,13 @@ class ExperimentTracking(mixins.SerializableSessionMixin):
                 to string.
         """
         run = self._get_or_start_run()
-        metadata = run._get_metadata()
+        params_list = []
         for key, value in params.items():
-            metadata.set_param(key, value)
-        self._sql_client.modify_run(
+            params_list.append(entities.Param(key, str(value)))
+        self._sql_client.modify_run_add_params(
             experiment_name=run.experiment_name,
             run_name=run.name,
-            run_metadata=json.dumps(metadata.to_dict()),
+            params=json.dumps([param.to_dict() for param in params_list]),
         )
 
     def log_artifact(

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.15.0}/snowflake/ml/model/__init__.py

@@ -1,10 +1,6 @@
-from snowflake.ml.model._client.model.batch_inference_specs import (
-    InputSpec,
-    JobSpec,
-    OutputSpec,
-)
+from snowflake.ml.model._client.model.batch_inference_specs import JobSpec, OutputSpec
 from snowflake.ml.model._client.model.model_impl import Model
 from snowflake.ml.model._client.model.model_version_impl import ExportMode, ModelVersion
 from snowflake.ml.model.models.huggingface_pipeline import HuggingFacePipelineModel
 
-__all__ = ["Model", "ModelVersion", "ExportMode", "HuggingFacePipelineModel", "InputSpec", "JobSpec", "OutputSpec"]
+__all__ = ["Model", "ModelVersion", "ExportMode", "HuggingFacePipelineModel", "JobSpec", "OutputSpec"]

{snowflake_ml_python-1.14.0 → snowflake_ml_python-1.15.0}/snowflake/ml/model/_client/model/batch_inference_specs.py

@@ -3,10 +3,6 @@ from typing import Optional, Union
 from pydantic import BaseModel
 
 
-class InputSpec(BaseModel):
-    stage_location: str
-
-
 class OutputSpec(BaseModel):
     stage_location: str
 

snowflake_ml_python-1.15.0/snowflake/ml/model/_client/model/inference_engine_utils.py

@@ -0,0 +1,55 @@
+from typing import Any, Optional, Union
+
+from snowflake.ml.model._client.ops import service_ops
+
+
+def _get_inference_engine_args(
+    experimental_options: Optional[dict[str, Any]],
+) -> Optional[service_ops.InferenceEngineArgs]:
+
+    if not experimental_options:
+        return None
+
+    if "inference_engine" not in experimental_options:
+        raise ValueError("inference_engine is required in experimental_options")
+
+    return service_ops.InferenceEngineArgs(
+        inference_engine=experimental_options["inference_engine"],
+        inference_engine_args_override=experimental_options.get("inference_engine_args_override"),
+    )
+
+
+def _enrich_inference_engine_args(
+    inference_engine_args: service_ops.InferenceEngineArgs,
+    gpu_requests: Optional[Union[str, int]] = None,
+) -> Optional[service_ops.InferenceEngineArgs]:
+    """Enrich inference engine args with model path and tensor parallelism settings.
+
+    Args:
+        inference_engine_args: The original inference engine args
+        gpu_requests: The number of GPUs requested
+
+    Returns:
+        Enriched inference engine args
+
+    Raises:
+        ValueError: Invalid gpu_requests
+    """
+    if inference_engine_args.inference_engine_args_override is None:
+        inference_engine_args.inference_engine_args_override = []
+
+    gpu_count = None
+
+    # Set tensor-parallelism if gpu_requests is specified
+    if gpu_requests is not None:
+        # assert gpu_requests is a string or an integer before casting to int
+        try:
+            gpu_count = int(gpu_requests)
+            if gpu_count > 0:
+                inference_engine_args.inference_engine_args_override.append(f"--tensor-parallel-size={gpu_count}")
+            else:
+                raise ValueError(f"GPU count must be greater than 0, got {gpu_count}")
+        except ValueError:
+            raise ValueError(f"Invalid gpu_requests: {gpu_requests} with type {type(gpu_requests).__name__}")
+
+    return inference_engine_args