snowflake-ml-python 1.35.0__tar.gz → 1.37.0__tar.gz

{snowflake_ml_python-1.35.0 → snowflake_ml_python-1.37.0}/CHANGELOG.md

@@ -1,6 +1,48 @@
 # Release History
 
-## 1.35.0
+## 1.37.0
+
+### New Features
+
+* Experiment Tracking: `end_run` now accepts an optional `status` argument (`"FINISHED"` or `"FAILED"`) to explicitly
+  set the final run status. When a run's context manager exits with an exception, the status is automatically set to
+  `"FAILED"`.
+
+### Bug Fixes
+
+* Registry: Fixed `log_model()` failing for some Snowpark ML `Pipeline` models with explainability when the
+  full pipeline could not be converted to a native object; task inference now uses the final estimator instead.
+
+* Registry: `run_batch()` now raises a clear `ValueError` when a partitioned model's output signature
+  includes the partition column. Previously this produced duplicate columns in the output, causing
+  cryptic Ray/Arrow errors (`AttributeError` or `KeyError`) deep in the batch inference pipeline.
+
+### Behavior Changes
+
+### Deprecations
+
+## 1.36.0 (2026-04-22)
+
+### New Features
+
+* Feature Store: for latency-sensitive online feature view reads, set use_session_warehouse=True to
+  re-use the warehouse from the current session and achieve the best latency.
+
+### Bug Fixes
+
+* Registry: Pin `transformers<5` when saving HuggingFace pipeline models that use tasks removed in
+  `transformers 5.x` (`image-to-text`, `visual-question-answering`, `conversational`), preventing
+  deployment failures from missing pipeline classes.
+
+### Behavior Changes
+
+* Registry: Logging a model with `pip_requirements` and no `artifact_repository_map` now raises a `ValueError`
+  when `target_platforms` explicitly includes `"WAREHOUSE"`. Previously this combination would silently proceed,
+  resulting in the warehouse deployment being silently dropped.
+
+### Deprecations
+
+## 1.35.0 (2026-04-17)
 
 ### New Features
 

{snowflake_ml_python-1.35.0/snowflake_ml_python.egg-info → snowflake_ml_python-1.37.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.35.0
+Version: 1.37.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: 
@@ -238,6 +238,7 @@ Requires-Dist: cachetools<6,>=3.1.1
 Requires-Dist: cloudpickle>=2.0.0
 Requires-Dist: cryptography
 Requires-Dist: fsspec[http]<2026,>=2024.6.1
+Requires-Dist: h2<5,>=4.3.0
 Requires-Dist: importlib_resources<7,>=6.1.1
 Requires-Dist: numpy<3,>=1.23
 Requires-Dist: packaging<26,>=20.9
@@ -416,7 +417,49 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.35.0
+## 1.37.0
+
+### New Features
+
+* Experiment Tracking: `end_run` now accepts an optional `status` argument (`"FINISHED"` or `"FAILED"`) to explicitly
+  set the final run status. When a run's context manager exits with an exception, the status is automatically set to
+  `"FAILED"`.
+
+### Bug Fixes
+
+* Registry: Fixed `log_model()` failing for some Snowpark ML `Pipeline` models with explainability when the
+  full pipeline could not be converted to a native object; task inference now uses the final estimator instead.
+
+* Registry: `run_batch()` now raises a clear `ValueError` when a partitioned model's output signature
+  includes the partition column. Previously this produced duplicate columns in the output, causing
+  cryptic Ray/Arrow errors (`AttributeError` or `KeyError`) deep in the batch inference pipeline.
+
+### Behavior Changes
+
+### Deprecations
+
+## 1.36.0 (2026-04-22)
+
+### New Features
+
+* Feature Store: for latency-sensitive online feature view reads, set use_session_warehouse=True to
+  re-use the warehouse from the current session and achieve the best latency.
+
+### Bug Fixes
+
+* Registry: Pin `transformers<5` when saving HuggingFace pipeline models that use tasks removed in
+  `transformers 5.x` (`image-to-text`, `visual-question-answering`, `conversational`), preventing
+  deployment failures from missing pipeline classes.
+
+### Behavior Changes
+
+* Registry: Logging a model with `pip_requirements` and no `artifact_repository_map` now raises a `ValueError`
+  when `target_platforms` explicitly includes `"WAREHOUSE"`. Previously this combination would silently proceed,
+  resulting in the warehouse deployment being silently dropped.
+
+### Deprecations
+
+## 1.35.0 (2026-04-17)
 
 ### New Features
 

{snowflake_ml_python-1.35.0 → snowflake_ml_python-1.37.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.14"
 dynamic = [ "version", "readme",]
-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,<26", "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.17.3,<5", "snowflake-snowpark-python>=1.17.0,<2,!=1.26.0", "sqlparse>=0.4,<1", "tqdm<5", "typing-extensions>=4.1.0,<5", "xgboost<4",]
+dependencies = [ "anyio>=3.5.0,<5", "cachetools>=3.1.1,<6", "cloudpickle>=2.0.0", "cryptography", "fsspec[http]>=2024.6.1,<2026", "h2>=4.3.0,<5", "importlib_resources>=6.1.1, <7", "numpy>=1.23,<3", "packaging>=20.9,<26", "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.17.3,<5", "snowflake-snowpark-python>=1.17.0,<2,!=1.26.0", "sqlparse>=0.4,<1", "tqdm<5", "typing-extensions>=4.1.0,<5", "xgboost<4",]
 [[project.authors]]
 name = "Snowflake, Inc"
 email = "support@snowflake.com"

{snowflake_ml_python-1.35.0 → snowflake_ml_python-1.37.0}/snowflake/ml/_internal/utils/connection_params.py

@@ -111,7 +111,8 @@ def _load_from_snowsql_config_file(connection_name: str, login_file: str = "") -
         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="#")
+    # strict=False: ~/.snowsql/config may repeat keys (e.g. warehouse added twice); last value wins.
+    config = configparser.ConfigParser(inline_comment_prefixes="#", strict=False)
 
     snowflake_connection_name = os.getenv("SNOWFLAKE_CONNECTION_NAME")
     if snowflake_connection_name is not None:

{snowflake_ml_python-1.35.0 → snowflake_ml_python-1.37.0}/snowflake/ml/experiment/_client/experiment_tracking_sql_client.py

@@ -3,6 +3,7 @@ from typing import Any, Optional
 from snowflake.ml._internal import telemetry
 from snowflake.ml._internal.utils import query_result_checker, sql_identifier
 from snowflake.ml.experiment._client import artifact
+from snowflake.ml.experiment._entities import run_metadata
 from snowflake.ml.model._client.sql import _base
 from snowflake.ml.utils import sql_client
 from snowflake.snowpark import dataframe, file_operation, row, session, types
@@ -129,11 +130,15 @@ class ExperimentTrackingSQLClient(_base._BaseSQLClient):
         *,
         experiment_name: sql_identifier.SqlIdentifier,
         run_name: sql_identifier.SqlIdentifier,
+        status: Optional[run_metadata.RunStatus] = None,
     ) -> 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} COMMIT RUN {run_name}"
-        ).has_dimensions(expected_rows=1, expected_cols=1).validate()
+        query = f"ALTER EXPERIMENT {experiment_fqn} COMMIT RUN {run_name}"
+        if status is not None:
+            query += f" WITH STATUS='{status.value}'"
+        query_result_checker.SqlResultValidator(self._session, query).has_dimensions(
+            expected_rows=1, expected_cols=1
+        ).validate()
 
     def get_run_id(
         self,

{snowflake_ml_python-1.35.0 → snowflake_ml_python-1.37.0}/snowflake/ml/experiment/_entities/run.py

@@ -4,6 +4,7 @@ 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._entities import run_metadata
 
 if TYPE_CHECKING:
     from snowflake.ml.experiment import experiment_tracking
@@ -44,7 +45,9 @@ class Run:
         if self._experiment_tracking._run is self:
             if exc_type is not None:
                 self._experiment_tracking._try_log_exception(exc_type, exc_value, tb)
-            self._experiment_tracking.end_run()
+                self._experiment_tracking.end_run(status=run_metadata.RunStatus.FAILED.value)
+            else:
+                self._experiment_tracking.end_run()
 
     def _get_experiment_info(self) -> experiment_info.ExperimentInfo:
         return experiment_info.ExperimentInfo(

{snowflake_ml_python-1.35.0 → snowflake_ml_python-1.37.0}/snowflake/ml/experiment/_entities/run_metadata.py

@@ -1,4 +1,10 @@
 import dataclasses
+import enum
+
+
+class RunStatus(enum.Enum):
+    FAILED = "FAILED"
+    FINISHED = "FINISHED"
 
 
 @dataclasses.dataclass

{snowflake_ml_python-1.35.0 → snowflake_ml_python-1.37.0}/snowflake/ml/experiment/experiment_tracking.py

@@ -23,6 +23,7 @@ from snowflake.ml.experiment._client import (
     artifact,
     experiment_tracking_sql_client as sql_client,
 )
+from snowflake.ml.experiment._entities import run_metadata
 from snowflake.ml.model import type_hints
 from snowflake.ml.utils import sql_client as sql_client_utils
 
@@ -311,15 +312,17 @@ class ExperimentTracking:
         assert self._run is not None  # for mypy
         return self._run
 
-    def end_run(self, run_name: Optional[str] = None) -> None:
+    def end_run(self, run_name: Optional[str] = None, status: Optional[str] = None) -> None:
         """
         End the current run if no run name is provided. Otherwise, the specified run is ended.
 
         Args:
             run_name: The name of the run to be ended. If None, the current run is ended.
+            status: The final status for the run, e.g. "FINISHED" or "FAILED".
 
         Raises:
             RuntimeError: If no run is active.
+            ValueError: If the status value is not supported.
         """
         if not self._experiment:
             raise RuntimeError("No experiment set. Please set an experiment before ending a run.")
@@ -332,9 +335,18 @@ class ExperimentTracking:
         else:
             raise RuntimeError("No run is active. Please start a run before ending it.")
 
+        try:
+            run_status = run_metadata.RunStatus(status) if status is not None else None
+        except ValueError:
+            raise ValueError(
+                f"Unsupported status value: '{status}'. Supported values are:"
+                f" {[s.value for s in run_metadata.RunStatus]}."
+            )
+
         self._sql_client.commit_run(
             experiment_name=experiment_name,
             run_name=run_name,
+            status=run_status,
         )
         if self._run and run_name == self._run.name:
             self._unset_run()

{snowflake_ml_python-1.35.0 → snowflake_ml_python-1.37.0}/snowflake/ml/feature_store/__init__.py

@@ -5,6 +5,7 @@ from snowflake.ml._internal import init_utils
 from .access_manager import setup_feature_store
 from .feature import Feature
 from .feature_view import RollupConfig
+from .spec.enums import FeatureAggregationMethod
 from .stream_config import StreamConfig
 from .stream_source import StreamSource
 
@@ -17,6 +18,7 @@ for k, v in exportable_classes.items():
 __all__ = list(exportable_classes.keys()) + [
     "setup_feature_store",
     "Feature",
+    "FeatureAggregationMethod",
     "RollupConfig",
     "StreamConfig",
     "StreamSource",