snowflake-ml-python 1.32.0__tar.gz → 1.34.0__tar.gz

{snowflake_ml_python-1.32.0 → snowflake_ml_python-1.34.0}/CHANGELOG.md

@@ -1,6 +1,50 @@
 # Release History
 
-## 1.32.0
+## 1.34.0
+
+### New Features
+
+* Experiment Tracking: Added `list_params` and `list_metrics` methods to retrieve parameters and metrics
+  for runs within an experiment. Both methods return a Dataframe and accept an optional `run_name` argument
+  to filter to a specific run.
+
+### Bug Fixes
+
+* Feature Store: Fixed `generate_dataset()`/`generate_training_set()` SQL generation for Unicode and case-sensitive
+  identifiers (for example Japanese column names), ensuring columns are quoted exactly once and preventing SQL
+  compilation errors such as `unexpected '<column_name>'`.
+
+### Behavior Changes
+
+* Registry: `enable_explainability=True` is now allowed for SPCS-only and non-warehouse-runnable models. Previously
+  this raised a `ValueError`.
+
+### Deprecations
+
+## 1.33.0 (2026-03-31)
+
+### New Features
+
+* Registry: Extended `ParamSpec` support to PyTorch models. Parameters declared in the model signature
+  can now be passed at inference time, matching the existing support for custom models.
+* Registry (PrPr): Added `model_init_once` model save option to `log_model`. When set to `True`, the
+  model is loaded once per worker process at startup, eliminating model-loading overhead. Defaults to `False`.
+* Experiment Tracking live logging is out of private preview
+  and will become generally available over the next few weeks.
+* Registry: Bumped `transformers` upper bound to `<6`, adding compatibility with Hugging Face Transformers v5.
+
+### Bug Fixes
+
+* Experiment Tracking live logging: Fixed a bug where the tracebacks for uncaught exceptions were not logged when
+  the code was running in a `with exp.start_run()` block.
+* Registry: Fixed a thread leak in `create_service(block=True)` where the log-streaming thread could outlive the
+  error handler when service deployment fails.
+
+### Behavior Changes
+
+### Deprecations
+
+## 1.32.0 (2026-03-26)
 
 ### New Features
 
@@ -101,6 +145,9 @@
 
 ### Behavior Changes
 
+* Dependencies: Removed `snowflake.core` as a runtime dependency. If you use `snowflake.core` (e.g.,
+  `TaskDag`, `Root`, `Task`), add `snowflake.core` explicitly to your dependencies.
+
 ### Deprecations
 
 ## 1.28.0 (2026-02-17)

{snowflake_ml_python-1.32.0/snowflake_ml_python.egg-info → snowflake_ml_python-1.34.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.32.0
+Version: 1.34.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: 
@@ -266,14 +266,14 @@ Requires-Dist: keras<4,>=2.0.0; extra == "all"
 Requires-Dist: lightgbm<5,>=4.1.0; extra == "all"
 Requires-Dist: mlflow<3,>=2.16.0; extra == "all"
 Requires-Dist: prophet<2,>=1.1.0; extra == "all"
-Requires-Dist: sentence-transformers<4,>=2.7.0; extra == "all"
+Requires-Dist: sentence-transformers<6,>=2.7.0; extra == "all"
 Requires-Dist: sentencepiece<0.2.0,>=0.1.95; extra == "all"
 Requires-Dist: streamlit<2,>=1.30.0; extra == "all"
 Requires-Dist: tensorflow<3,>=2.17.0; extra == "all"
 Requires-Dist: tokenizers<1,>=0.15.1; extra == "all"
 Requires-Dist: torch<3,>=2.0.1; extra == "all"
 Requires-Dist: torchdata<1,>=0.4; extra == "all"
-Requires-Dist: transformers!=4.51.3,<5,>=4.39.3; extra == "all"
+Requires-Dist: transformers!=4.51.3,<6,>=4.39.3; extra == "all"
 Provides-Extra: altair
 Requires-Dist: altair<6,>=5; extra == "altair"
 Provides-Extra: catboost
@@ -286,12 +286,12 @@ Provides-Extra: lightgbm
 Requires-Dist: lightgbm<5,>=4.1.0; extra == "lightgbm"
 Provides-Extra: llm
 Requires-Dist: mlflow<3,>=2.16.0; extra == "llm"
-Requires-Dist: sentence-transformers<4,>=2.7.0; extra == "llm"
+Requires-Dist: sentence-transformers<6,>=2.7.0; extra == "llm"
 Requires-Dist: sentencepiece<0.2.0,>=0.1.95; extra == "llm"
 Requires-Dist: tokenizers<1,>=0.15.1; extra == "llm"
 Requires-Dist: torch<3,>=2.0.1; extra == "llm"
 Requires-Dist: torchdata<1,>=0.4; extra == "llm"
-Requires-Dist: transformers!=4.51.3,<5,>=4.39.3; extra == "llm"
+Requires-Dist: transformers!=4.51.3,<6,>=4.39.3; extra == "llm"
 Provides-Extra: mlflow
 Requires-Dist: mlflow<3,>=2.16.0; extra == "mlflow"
 Provides-Extra: prophet
@@ -304,11 +304,11 @@ Provides-Extra: torch
 Requires-Dist: torch<3,>=2.0.1; extra == "torch"
 Requires-Dist: torchdata<1,>=0.4; extra == "torch"
 Provides-Extra: transformers
-Requires-Dist: sentence-transformers<4,>=2.7.0; extra == "transformers"
+Requires-Dist: sentence-transformers<6,>=2.7.0; extra == "transformers"
 Requires-Dist: sentencepiece<0.2.0,>=0.1.95; extra == "transformers"
 Requires-Dist: tokenizers<1,>=0.15.1; extra == "transformers"
 Requires-Dist: torch<3,>=2.0.1; extra == "transformers"
-Requires-Dist: transformers!=4.51.3,<5,>=4.39.3; extra == "transformers"
+Requires-Dist: transformers!=4.51.3,<6,>=4.39.3; extra == "transformers"
 Dynamic: license-file
 
 # Snowflake ML Python
@@ -416,7 +416,51 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.32.0
+## 1.34.0
+
+### New Features
+
+* Experiment Tracking: Added `list_params` and `list_metrics` methods to retrieve parameters and metrics
+  for runs within an experiment. Both methods return a Dataframe and accept an optional `run_name` argument
+  to filter to a specific run.
+
+### Bug Fixes
+
+* Feature Store: Fixed `generate_dataset()`/`generate_training_set()` SQL generation for Unicode and case-sensitive
+  identifiers (for example Japanese column names), ensuring columns are quoted exactly once and preventing SQL
+  compilation errors such as `unexpected '<column_name>'`.
+
+### Behavior Changes
+
+* Registry: `enable_explainability=True` is now allowed for SPCS-only and non-warehouse-runnable models. Previously
+  this raised a `ValueError`.
+
+### Deprecations
+
+## 1.33.0 (2026-03-31)
+
+### New Features
+
+* Registry: Extended `ParamSpec` support to PyTorch models. Parameters declared in the model signature
+  can now be passed at inference time, matching the existing support for custom models.
+* Registry (PrPr): Added `model_init_once` model save option to `log_model`. When set to `True`, the
+  model is loaded once per worker process at startup, eliminating model-loading overhead. Defaults to `False`.
+* Experiment Tracking live logging is out of private preview
+  and will become generally available over the next few weeks.
+* Registry: Bumped `transformers` upper bound to `<6`, adding compatibility with Hugging Face Transformers v5.
+
+### Bug Fixes
+
+* Experiment Tracking live logging: Fixed a bug where the tracebacks for uncaught exceptions were not logged when
+  the code was running in a `with exp.start_run()` block.
+* Registry: Fixed a thread leak in `create_service(block=True)` where the log-streaming thread could outlive the
+  error handler when service deployment fails.
+
+### Behavior Changes
+
+### Deprecations
+
+## 1.32.0 (2026-03-26)
 
 ### New Features
 
@@ -517,6 +561,9 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 ### Behavior Changes
 
+* Dependencies: Removed `snowflake.core` as a runtime dependency. If you use `snowflake.core` (e.g.,
+  `TaskDag`, `Root`, `Task`), add `snowflake.core` explicitly to your dependencies.
+
 ### Deprecations
 
 ## 1.28.0 (2026-02-17)

{snowflake_ml_python-1.32.0 → snowflake_ml_python-1.34.0}/pyproject.toml

@@ -24,18 +24,18 @@ Issues = "https://github.com/snowflakedb/snowflake-ml-python/issues"
 Changelog = "https://github.com/snowflakedb/snowflake-ml-python/blob/master/CHANGELOG.md"
 
 [project.optional-dependencies]
-all = [ "altair>=5,<6", "catboost>=1.2.0, <2", "keras>=2.0.0,<4", "lightgbm>=4.1.0, <5", "mlflow>=2.16.0, <3", "prophet>=1.1.0, <2", "sentence-transformers>=2.7.0,<4", "sentencepiece>=0.1.95,<0.2.0", "streamlit>=1.30.0,<2", "tensorflow>=2.17.0,<3", "tokenizers>=0.15.1,<1", "torch>=2.0.1,<3", "torchdata>=0.4,<1", "transformers>=4.39.3,<5,!=4.51.3",]
+all = [ "altair>=5,<6", "catboost>=1.2.0, <2", "keras>=2.0.0,<4", "lightgbm>=4.1.0, <5", "mlflow>=2.16.0, <3", "prophet>=1.1.0, <2", "sentence-transformers>=2.7.0,<6", "sentencepiece>=0.1.95,<0.2.0", "streamlit>=1.30.0,<2", "tensorflow>=2.17.0,<3", "tokenizers>=0.15.1,<1", "torch>=2.0.1,<3", "torchdata>=0.4,<1", "transformers>=4.39.3,!=4.51.3,<6",]
 altair = [ "altair>=5,<6",]
 catboost = [ "catboost>=1.2.0, <2",]
 keras = [ "keras>=2.0.0,<4", "tensorflow>=2.17.0,<3", "torch>=2.0.1,<3",]
 lightgbm = [ "lightgbm>=4.1.0, <5",]
-llm = [ "mlflow>=2.16.0, <3", "sentence-transformers>=2.7.0,<4", "sentencepiece>=0.1.95,<0.2.0", "tokenizers>=0.15.1,<1", "torch>=2.0.1,<3", "torchdata>=0.4,<1", "transformers>=4.39.3,<5,!=4.51.3",]
+llm = [ "mlflow>=2.16.0, <3", "sentence-transformers>=2.7.0,<6", "sentencepiece>=0.1.95,<0.2.0", "tokenizers>=0.15.1,<1", "torch>=2.0.1,<3", "torchdata>=0.4,<1", "transformers>=4.39.3,!=4.51.3,<6",]
 mlflow = [ "mlflow>=2.16.0, <3",]
 prophet = [ "prophet>=1.1.0, <2",]
 streamlit = [ "streamlit>=1.30.0,<2",]
 tensorflow = [ "tensorflow>=2.17.0,<3",]
 torch = [ "torch>=2.0.1,<3", "torchdata>=0.4,<1",]
-transformers = [ "sentence-transformers>=2.7.0,<4", "sentencepiece>=0.1.95,<0.2.0", "tokenizers>=0.15.1,<1", "torch>=2.0.1,<3", "transformers>=4.39.3,<5,!=4.51.3",]
+transformers = [ "sentence-transformers>=2.7.0,<6", "sentencepiece>=0.1.95,<0.2.0", "tokenizers>=0.15.1,<1", "torch>=2.0.1,<3", "transformers>=4.39.3,!=4.51.3,<6",]
 
 [tool.setuptools.package-data]
 "*" = [ "*",]

{snowflake_ml_python-1.32.0 → snowflake_ml_python-1.34.0}/snowflake/ml/_internal/env_utils.py

@@ -591,18 +591,104 @@ def save_conda_env_file(
         yaml.safe_dump(env, stream=f, default_flow_style=False)
 
 
-def save_requirements_file(path: pathlib.Path, pip_deps: list[requirements.Requirement]) -> None:
+def _get_pytorch_cuda_suffix(cuda_version: str) -> str:
+    """Get PyTorch CUDA wheel suffix (e.g., '124') for the specified CUDA version.
+
+    Args:
+        cuda_version: CUDA version string.
+
+    Returns:
+        The CUDA suffix for PyTorch wheels.
+    """
+    parsed = version.parse(cuda_version)
+    return f"{parsed.major}{parsed.minor}"
+
+
+def _get_pytorch_cuda_index_url(cuda_version: str) -> str:
+    """Get PyTorch CUDA index URL for the specified CUDA version.
+
+    Args:
+        cuda_version: CUDA version string.
+
+    Returns:
+        The --extra-index-url argument for pip/uv.
+    """
+    cuda_suffix = _get_pytorch_cuda_suffix(cuda_version)
+    return f"--extra-index-url https://download.pytorch.org/whl/cu{cuda_suffix}"
+
+
+def _pin_torch_to_cuda_variant(
+    pip_deps: list[requirements.Requirement], cuda_version: str
+) -> list[requirements.Requirement]:
+    """Pin torch requirement to the CUDA variant (e.g., torch==2.5.0+cu124).
+
+    PyTorch CUDA wheels use local version identifiers like +cu124 to distinguish
+    CUDA-enabled builds from CPU-only builds.
+
+    Args:
+        pip_deps: Validated pip requirements.
+        cuda_version: CUDA version string (e.g., "12.4", "12.4.1").
+
+    Returns:
+        A new list of requirements with ``torch==X.Y.Z`` rewritten to ``torch==X.Y.Z+cu...``
+        when the requirement is exactly pinned and has no ``+cu`` suffix yet.
+    """
+    cuda_suffix = _get_pytorch_cuda_suffix(cuda_version)
+    result = []
+
+    for req in pip_deps:
+        if req.name != "torch":
+            result.append(req)
+            continue
+
+        # Extract pinned version (e.g., ==2.5.0)
+        pinned_version = next((s.version for s in req.specifier if s.operator == "=="), None)
+
+        # Skip if not pinned or already has +cu suffix
+        if not pinned_version or "+cu" in pinned_version:
+            result.append(req)
+            continue
+
+        # Create new requirement with CUDA local version
+        new_req = requirements.Requirement(f"torch=={pinned_version}+cu{cuda_suffix}")
+        new_req.extras = req.extras
+        new_req.marker = req.marker
+        result.append(new_req)
+
+    return result
+
+
+def _has_torch_dependency(pip_deps: list[requirements.Requirement]) -> bool:
+    """Check if torch is in the pip dependencies."""
+    return any(dep.name == "torch" for dep in pip_deps)
+
+
+def save_requirements_file(
+    path: pathlib.Path,
+    pip_deps: list[requirements.Requirement],
+    cuda_version: Optional[str] = None,
+) -> None:
     """Generate Python requirements.txt file in the given directory path.
 
     Args:
         path: Path to the requirements.txt file.
-        pip_deps: List of dependencies string after validated.
+        pip_deps: List of dependencies after validation.
+        cuda_version: Optional CUDA version for GPU models. If provided and torch is in
+            dependencies, adds PyTorch CUDA index URL and pins torch to CUDA variant.
     """
     assert path.suffix in [".txt"], "PIP requirement file should have extension of txt."
     path.parent.mkdir(parents=True, exist_ok=True)
-    requirements = "\n".join(map(str, pip_deps))
+
+    lines: list[str] = []
+
+    if cuda_version and _has_torch_dependency(pip_deps):
+        lines.append(_get_pytorch_cuda_index_url(cuda_version))
+        pip_deps = _pin_torch_to_cuda_variant(pip_deps, cuda_version)
+
+    lines.extend(map(str, pip_deps))
+
     with open(path, "w", encoding="utf-8") as out:
-        out.write(requirements)
+        out.write("\n".join(lines))
 
 
 def load_conda_env_file(
@@ -702,6 +788,9 @@ def load_requirements_file(path: pathlib.Path) -> list[requirements.Requirement]
     with open(path, encoding="utf-8") as f:
         reqs = f.readlines()
 
+    # Filter out pip options
+    reqs = [line for line in reqs if line.strip() and not line.strip().startswith(("-", "#"))]
+
     return validate_pip_requirement_string_list(reqs)
 
 

{snowflake_ml_python-1.32.0 → snowflake_ml_python-1.34.0}/snowflake/ml/_internal/platform_capabilities.py

@@ -1,7 +1,7 @@
 import json
 import logging
 from contextlib import contextmanager
-from typing import Any, Optional
+from typing import Any, Iterator, Optional
 
 from packaging import version
 
@@ -57,13 +57,11 @@ class PlatformCapabilities:
     def clear_mock_features(cls) -> None:
         cls._mock_features = None
 
-    # For contextmanager, we need to have return type Iterator[Never]. However, Never type is introduced only in
-    # Python 3.11. So, we are ignoring the type for this method.
     _dummy_features: dict[str, Any] = {"dummy": "dummy"}
 
-    @classmethod  # type: ignore[arg-type]
+    @classmethod
     @contextmanager
-    def mock_features(cls, features: dict[str, Any] = _dummy_features) -> None:  # type: ignore[misc]
+    def mock_features(cls, features: dict[str, Any] = _dummy_features) -> Iterator[None]:
         logger.debug(f"Setting mock features: {features}")
         cls.set_mock_features(features)
         try:

{snowflake_ml_python-1.32.0 → snowflake_ml_python-1.34.0}/snowflake/ml/experiment/_client/experiment_tracking_sql_client.py

@@ -1,16 +1,62 @@
-from typing import Optional
+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.model._client.sql import _base
 from snowflake.ml.utils import sql_client
-from snowflake.snowpark import file_operation, row, session
+from snowflake.snowpark import dataframe, file_operation, row, session, types
 
 RUN_NAME_COL_NAME = "name"
 RUN_METADATA_COL_NAME = "metadata"
 
 
+def pivot_run_attributes(
+    sp_session: session.Session,
+    rows: list[row.Row],
+    run_names: list[str],
+    cast_value: Optional[type] = None,
+) -> dataframe.DataFrame:
+    """Pivot SHOW RUN METRICS/PARAMETERS rows into one row per run.
+
+    Each row is expected to have "run_name", "name", and "value" keys.
+    Returns a Snowpark DataFrame with a "run_name" column and one column per distinct
+    attribute name (sorted alphabetically). Runs present in ``run_names`` but absent
+    from ``rows`` appear with all-NULL attribute columns.
+
+    Args:
+        sp_session: Snowpark session used to create the result DataFrame.
+        rows: Collected rows from a SHOW RUN METRICS/PARAMETERS command.
+        run_names: Exhaustive list of run names (from SHOW RUNS) to include in the output.
+        cast_value: If provided, cast each value with this callable (e.g. ``float`` for metrics).
+                    Values that fail to cast are set to ``None``.
+
+    Returns:
+        A Snowpark DataFrame with a "run_name" column and one column per distinct attribute name.
+    """
+    pivoted: dict[str, dict[str, Any]] = {name: {} for name in run_names}
+    attr_names: set[str] = set()
+    for r in rows:
+        d = r.as_dict()
+        value = d["value"]
+        if cast_value is not None:
+            try:
+                value = cast_value(value)
+            except (ValueError, TypeError):
+                value = None
+        pivoted.setdefault(d["run_name"], {})[d["name"]] = value
+        attr_names.add(d["name"])
+
+    col_order = sorted(attr_names)
+    data = [[run_name] + [attrs.get(c) for c in col_order] for run_name, attrs in pivoted.items()]
+    value_type = types.DoubleType() if cast_value is float else types.StringType()
+    schema = types.StructType(
+        [types.StructField('"run_name"', types.StringType())]
+        + [types.StructField(f'"{c}"', value_type) for c in col_order]
+    )
+    return sp_session.create_dataframe(data, schema=schema)
+
+
 class ExperimentTrackingSQLClient(_base._BaseSQLClient):
     def __init__(
         self,
@@ -203,6 +249,36 @@ class ExperimentTrackingSQLClient(_base._BaseSQLClient):
             self._session, f"SHOW RUNS {like_clause} IN EXPERIMENT {experiment_fqn}"
         ).validate()
 
+    @telemetry.send_api_usage_telemetry(project=telemetry.TelemetryProject.EXPERIMENT_TRACKING.value)
+    def show_run_metrics_in_experiment(
+        self,
+        *,
+        experiment_name: sql_identifier.SqlIdentifier,
+        run_name: Optional[sql_identifier.SqlIdentifier] = None,
+        like: Optional[str] = None,
+    ) -> list[row.Row]:
+        experiment_fqn = self.fully_qualified_object_name(self._database_name, self._schema_name, experiment_name)
+        run_name_clause = f"RUN {run_name}" if run_name else ""
+        like_clause = f"LIKE '{like}'" if like else ""
+        return query_result_checker.SqlResultValidator(
+            self._session, f"SHOW RUN METRICS {like_clause} IN EXPERIMENT {experiment_fqn} {run_name_clause}"
+        ).validate()
+
+    @telemetry.send_api_usage_telemetry(project=telemetry.TelemetryProject.EXPERIMENT_TRACKING.value)
+    def show_run_parameters_in_experiment(
+        self,
+        *,
+        experiment_name: sql_identifier.SqlIdentifier,
+        run_name: Optional[sql_identifier.SqlIdentifier] = None,
+        like: Optional[str] = None,
+    ) -> list[row.Row]:
+        experiment_fqn = self.fully_qualified_object_name(self._database_name, self._schema_name, experiment_name)
+        run_name_clause = f"RUN {run_name}" if run_name else ""
+        like_clause = f"LIKE '{like}'" if like else ""
+        return query_result_checker.SqlResultValidator(
+            self._session, f"SHOW RUN PARAMETERS {like_clause} IN EXPERIMENT {experiment_fqn} {run_name_clause}"
+        ).validate()
+
     def _build_snow_uri(
         self, experiment_name: sql_identifier.SqlIdentifier, run_name: sql_identifier.SqlIdentifier, artifact_path: str
     ) -> str:

{snowflake_ml_python-1.32.0 → snowflake_ml_python-1.34.0}/snowflake/ml/experiment/_entities/run.py

@@ -38,10 +38,12 @@ class Run:
         self,
         exc_type: Optional[type[BaseException]],
         exc_value: Optional[BaseException],
-        traceback: Optional[types.TracebackType],
+        tb: Optional[types.TracebackType],
     ) -> None:
-        self._patcher.__exit__(exc_type, exc_value, traceback)
+        self._patcher.__exit__(exc_type, exc_value, tb)
         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()
 
     def _get_experiment_info(self) -> experiment_info.ExperimentInfo:

{snowflake_ml_python-1.32.0 → snowflake_ml_python-1.34.0}/snowflake/ml/experiment/experiment_tracking.py

@@ -2,12 +2,13 @@ import contextlib
 import functools
 import json
 import sys
+import traceback
 import warnings
 from collections.abc import Callable
+from types import TracebackType
 from typing import Any, Optional, Union
 from urllib.parse import quote
 
-import snowflake.snowpark._internal.utils as snowpark_utils
 from snowflake import snowpark
 from snowflake.ml import model as ml_model, registry
 from snowflake.ml._internal import env_utils
@@ -477,6 +478,89 @@ class ExperimentTracking:
                 file_path=file_path,
             )
 
+    def list_metrics(self, run_name: Optional[str] = None) -> snowpark.DataFrame:
+        """
+        List metrics for runs within the current experiment.
+
+        When a metric is logged at multiple steps, the returned value corresponds to the highest step.
+
+        The returned DataFrame has a ``run_name`` column (str) and one float column per distinct metric name.
+        Runs that did not log a given metric have ``NULL`` for that column.
+
+        Example output::
+
+              run_name  accuracy  loss    f1
+            0 RUN_1     0.95      0.05    0.90
+            1 RUN_2     0.90      0.10    NULL
+
+        Args:
+            run_name: Name of the run to list metrics from. If None, lists metrics for all runs in the current
+                experiment.
+
+        Returns:
+            A Snowpark DataFrame (one row per run). Use ``.to_pandas()`` for a pandas DataFrame or
+            ``.collect()`` for a list of Row objects.
+
+        Raises:
+            RuntimeError: If no experiment is currently set.
+        """
+        if not self._experiment:
+            raise RuntimeError("No experiment set. Please use the `set_experiment` method before listing metrics.")
+
+        if run_name:
+            run_names = [run_name]
+        else:
+            runs = self._sql_client.show_runs_in_experiment(experiment_name=self._experiment.name)
+            run_names = [r[sql_client.RUN_NAME_COL_NAME] for r in runs]
+
+        rows = self._sql_client.show_run_metrics_in_experiment(
+            experiment_name=self._experiment.name,
+            run_name=sql_identifier.SqlIdentifier(run_name) if run_name else None,
+        )
+
+        return sql_client.pivot_run_attributes(self._session, rows, run_names=run_names, cast_value=float)
+
+    def list_params(self, run_name: Optional[str] = None) -> snowpark.DataFrame:
+        """
+        List parameters for runs within the current experiment.
+
+        The returned DataFrame has a ``run_name`` column (str) and one string column per distinct parameter name.
+        Use ``.cast()`` for numeric comparisons on parameter values. Runs that did not log a given parameter
+        have ``NULL`` for that column.
+
+        Example output::
+
+              run_name  learning_rate  batch_size  optimizer
+            0 RUN_1     0.01           32          adam
+            1 RUN_2     0.02           64          NULL
+
+        Args:
+            run_name: Name of the run to list parameters from. If None, lists parameters for all runs in the current
+                experiment.
+
+        Returns:
+            A Snowpark DataFrame (one row per run). Use ``.to_pandas()`` for a pandas DataFrame or
+            ``.collect()`` for a list of Row objects.
+
+        Raises:
+            RuntimeError: If no experiment is currently set.
+        """
+        if not self._experiment:
+            raise RuntimeError("No experiment set. Please use the `set_experiment` method before listing parameters.")
+
+        if run_name:
+            run_names = [run_name]
+        else:
+            runs = self._sql_client.show_runs_in_experiment(experiment_name=self._experiment.name)
+            run_names = [r[sql_client.RUN_NAME_COL_NAME] for r in runs]
+
+        rows = self._sql_client.show_run_parameters_in_experiment(
+            experiment_name=self._experiment.name,
+            run_name=sql_identifier.SqlIdentifier(run_name) if run_name else None,
+        )
+
+        return sql_client.pivot_run_attributes(self._session, rows, run_names=run_names)
+
     def list_artifacts(
         self,
         run_name: str,
@@ -536,7 +620,6 @@ class ExperimentTracking:
                 target_path=local_dir,
             )
 
-    @snowpark_utils.private_preview(version="1.30.0")
     def set_live_logging_status(self, enabled: bool) -> None:
         """
         Enable or disable live logging. When enabled, stdout and stderr are captured and logged to persistent storage.
@@ -599,7 +682,6 @@ class ExperimentTracking:
         scheme: str = "https",
         host: str = "app.snowflake.com",
     ) -> None:
-
         experiment_url = (
             f"{scheme}://{host}/_deeplink/#/experiments"
             f"/databases/{quote(str(self._database_name))}"
@@ -635,7 +717,7 @@ class ExperimentTracking:
                 ) -> None:
                     try:
                         text = ip.InteractiveTB.stb2text(stb)
-                        stderr_logger.write(text + "\n")
+                        stderr_logger.write(text)
                         stderr_logger.flush()
                     except Exception:
                         pass
@@ -651,6 +733,39 @@ class ExperimentTracking:
 
         return None, None
 
+    def _try_log_exception(
+        self,
+        exc_type: type[BaseException],
+        exc_value: Optional[BaseException],
+        exc_tb: Optional[TracebackType],
+    ) -> None:
+        """Append a traceback to the stderr logger before unpatching.
+
+        On exception, `Run.__exit__` ends the run and restores `sys.stderr` before the
+        interpreter prints the traceback for the exception, so the default traceback path
+        never reaches the stderr logger.
+
+        Args:
+            exc_type: The type of the exception.
+            exc_value: The value of the exception.
+            exc_tb: The traceback of the exception.
+        """
+        if self._logging_context is None:
+            return
+        ip = self._logging_context.ipython_instance
+        if ip:
+            try:
+                stb = ip.InteractiveTB.structured_traceback(exc_type, exc_value, exc_tb)
+                text = ip.InteractiveTB.stb2text(stb)
+                self._logging_context.stderr_logger.write(text)
+                self._logging_context.stderr_logger.flush()
+                return
+            except Exception:
+                pass
+        err_msg = "".join(traceback.format_exception(exc_type, exc_value, exc_tb))
+        self._logging_context.stderr_logger.write(err_msg)
+        self._logging_context.stderr_logger.flush()
+
     def _patch_stdout_and_stderr(self) -> None:
         if not self._live_logging_enabled:
             return

{snowflake_ml_python-1.32.0 → snowflake_ml_python-1.34.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 .stream_config import StreamConfig
 from .stream_source import StreamSource
 
 pkg_dir = os.path.dirname(__file__)
@@ -17,5 +18,6 @@ __all__ = list(exportable_classes.keys()) + [
     "setup_feature_store",
     "Feature",
     "RollupConfig",
+    "StreamConfig",
     "StreamSource",
 ]