snowflake-ml-python 1.51.0__tar.gz → 1.52.0__tar.gz

{snowflake_ml_python-1.51.0 → snowflake_ml_python-1.52.0}/CHANGELOG.md

@@ -1,6 +1,16 @@
 # Release History
 
-## 1.51.0
+## 1.52.0
+
+### New Features
+
+### Bug Fixes
+
+### Behavior Changes
+
+### Deprecations
+
+## 1.51.0 (2026-08-12)
 
 ### New Features
 
@@ -16,6 +26,9 @@
 
 ### Behavior Changes
 
+* Feature Store: `stream_ingest` now reuses the Online Service ingest endpoint cached on the
+  `StreamSource` by `get_stream_source`, avoiding a per-call server status round-trip.
+
 ### Deprecations
 
 ## 1.50.0 (2026-08-03)

{snowflake_ml_python-1.51.0 → snowflake_ml_python-1.52.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.51.0
+Version: 1.52.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: 
@@ -416,7 +416,17 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.51.0
+## 1.52.0
+
+### New Features
+
+### Bug Fixes
+
+### Behavior Changes
+
+### Deprecations
+
+## 1.51.0 (2026-08-12)
 
 ### New Features
 
@@ -432,6 +442,9 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 ### Behavior Changes
 
+* Feature Store: `stream_ingest` now reuses the Online Service ingest endpoint cached on the
+  `StreamSource` by `get_stream_source`, avoiding a per-call server status round-trip.
+
 ### Deprecations
 
 ## 1.50.0 (2026-08-03)

{snowflake_ml_python-1.51.0 → snowflake_ml_python-1.52.0}/snowflake/ml/_internal/telemetry.py

@@ -300,7 +300,7 @@ def add_statement_params_custom_tags(
     """
     if not statement_params:
         return {}
-    existing_custom_tags: dict[str, Any] = statement_params.pop(TelemetryField.KEY_CUSTOM_TAGS.value, {})
+    existing_custom_tags: dict[str, Any] = dict(statement_params.get(TelemetryField.KEY_CUSTOM_TAGS.value, {}))
     existing_custom_tags.update(custom_tags)
     # NOTE: This can be done with | operator after upgrade from py3.8
     return {

{snowflake_ml_python-1.51.0 → snowflake_ml_python-1.52.0}/snowflake/ml/feature_store/feature_store.py

@@ -3218,7 +3218,9 @@ class FeatureStore:
                 original_exception=ValueError(f"Cannot find StreamSource with name: {name_id}."),
             )
 
-        return StreamSource._from_dict(metadata)
+        stream_source = StreamSource._from_dict(metadata)
+        self._hydrate_stream_source_online_service(stream_source)
+        return stream_source
 
     @dispatch_decorator()
     def stream_ingest(
@@ -3259,20 +3261,26 @@ class FeatureStore:
             src = self.get_stream_source(str(stream_source))
             stream_name_wire = src.name.resolved()
 
-        stmp = self._telemetry_stmp if statement_params is None else statement_params
-        st = online_service.assert_online_service_running_with_ingest_endpoint(
-            self._session,
-            self._config.database,
-            self._config.schema,
-            statement_params=stmp,
-        )
-        ingest_url = online_service.endpoint_url(st, "ingest", access=self._online_service_access)
-        if not ingest_url:
-            raise snowml_exceptions.SnowflakeMLException(
-                error_code=error_codes.INVALID_ARGUMENT,
-                original_exception=ValueError("Online Service returned no ingest endpoint."),
+        # Reuse the ingest endpoint cached on the StreamSource by get_stream_source to
+        # skip the per-call GS status lookup. Fall back to a live status fetch when the
+        # URL is absent (manually constructed StreamSource, or hydrated before the
+        # Online Service was RUNNING).
+        ingest_base = getattr(src, "_postgres_online_ingest_url", None)
+        if not ingest_base:
+            stmp = self._telemetry_stmp if statement_params is None else statement_params
+            st = online_service.assert_online_service_running_with_ingest_endpoint(
+                self._session,
+                self._config.database,
+                self._config.schema,
+                statement_params=stmp,
             )
-        ingest_base = ingest_url
+            ingest_url = online_service.endpoint_url(st, "ingest", access=self._online_service_access)
+            if not ingest_url:
+                raise snowml_exceptions.SnowflakeMLException(
+                    error_code=error_codes.INVALID_ARGUMENT,
+                    original_exception=ValueError("Online Service returned no ingest endpoint."),
+                )
+            ingest_base = ingest_url
 
         expected_cols = _stream_source_schema_field_names(src.schema)
         rows = _normalize_stream_ingest_records(records)
@@ -8320,21 +8328,28 @@ FROM {batch_cte_names[0]}{''.join(merge_join_parts)}
 
             return fv
 
-    def _resolve_postgres_online_query_url(self, *, log_label: str) -> Optional[str]:
-        """Fetch the Postgres OFT query endpoint URL or warn and return ``None``.
+    def _resolve_postgres_online_endpoint_url(
+        self, endpoint_name: str, *, log_label: str, warn: bool = True
+    ) -> Optional[str]:
+        """Fetch a Postgres OFT Online Service endpoint URL or return ``None``.
 
         Failures (status not RUNNING, missing endpoint URL, transient
-        SYSTEM$ errors) are downgraded to ``UserWarning`` so callers can
-        still construct the target object and see the "Online Service not
-        RUNNING" error on first read.
+        SYSTEM$ errors) are downgraded to a return value of ``None`` so callers
+        can still construct the target object and surface the "Online Service
+        not RUNNING" error at use time. When ``warn`` is ``True`` these
+        failures also emit a ``UserWarning``.
 
         Args:
+            endpoint_name: Endpoint to resolve (``"query"`` or ``"ingest"``).
             log_label: Caller name embedded in the warning log
-                (e.g. ``"get_feature_view"``, ``"get_feature_group"``).
+                (e.g. ``"get_feature_view"``, ``"get_stream_source"``).
+            warn: When ``True``, emit a ``UserWarning`` on failure. Callers that
+                have their own use-time fallback (e.g. ``stream_ingest``) pass
+                ``False`` to avoid spurious warnings during metadata reads.
 
         Returns:
-            The query endpoint URL when the Online Service is RUNNING or
-            UPDATING and advertises a ``query`` endpoint; ``None`` otherwise.
+            The endpoint URL when the Online Service is RUNNING or UPDATING and
+            advertises the named endpoint; ``None`` otherwise.
         """
         try:
             st = online_service.fetch_online_service_status(
@@ -8345,30 +8360,62 @@ FROM {batch_cte_names[0]}{''.join(merge_join_parts)}
             )
         except Exception as ex:
             logger.warning("Could not fetch Online Service status during %s: %s", log_label, ex)
+            if warn:
+                warnings.warn(
+                    online_service.online_service_not_ready_message(),
+                    category=UserWarning,
+                    stacklevel=4,
+                )
+            return None
+        endpoint = online_service.endpoint_url(st, endpoint_name, access=self._online_service_access)
+        # UPDATING (redeploy in progress) still serves endpoints, so accept it too.
+        if st.status in ("RUNNING", "UPDATING") and endpoint:
+            # Temporary: SYSTEM$ may return host-only URLs until server sends full https URLs.
+            return endpoint
+        if warn:
             warnings.warn(
                 online_service.online_service_not_ready_message(),
                 category=UserWarning,
                 stacklevel=4,
             )
-            return None
-        query_url = online_service.endpoint_url(st, "query", access=self._online_service_access)
-        # UPDATING (redeploy in progress) still serves the query endpoint, so accept it too.
-        if st.status in ("RUNNING", "UPDATING") and query_url:
-            # Temporary: SYSTEM$ may return host-only query URLs until server sends full https URLs.
-            return query_url
-        warnings.warn(
-            online_service.online_service_not_ready_message(),
-            category=UserWarning,
-            stacklevel=4,
-        )
         return None
 
+    def _resolve_postgres_online_query_url(self, *, log_label: str) -> Optional[str]:
+        """Resolve the Postgres OFT ``query`` endpoint URL (thin wrapper).
+
+        Args:
+            log_label: Caller name embedded in the warning log
+                (e.g. ``"get_feature_view"``, ``"get_feature_group"``).
+
+        Returns:
+            The query endpoint URL when the Online Service is RUNNING or
+            UPDATING and advertises a ``query`` endpoint; ``None`` otherwise.
+        """
+        return self._resolve_postgres_online_endpoint_url("query", log_label=log_label)
+
     def _hydrate_postgres_online_service(self, fv: FeatureView) -> None:
         """Set ``_postgres_online_query_url`` and warn when Postgres online reads are not ready."""
         if not fv.online or fv.online_config is None or fv.online_config.store_type != OnlineStoreType.POSTGRES:
             return
         fv._postgres_online_query_url = self._resolve_postgres_online_query_url(log_label="get_feature_view")
 
+    def _hydrate_stream_source_online_service(self, stream_source: StreamSource) -> None:
+        """Cache the Online Service ingest endpoint URL on a StreamSource for reuse.
+
+        Best-effort: resolves the ``ingest`` endpoint without emitting a
+        ``UserWarning`` (``warn=False``) because ``get_stream_source`` is a
+        metadata read that also backs ``register_stream_source``; a not-yet-RUNNING
+        service should not warn here. ``stream_ingest`` surfaces a clear,
+        actionable error at ingest time via its live fallback when the URL is
+        unavailable.
+
+        Args:
+            stream_source: The StreamSource whose ingest URL should be populated.
+        """
+        stream_source._postgres_online_ingest_url = self._resolve_postgres_online_endpoint_url(
+            "ingest", log_label="get_stream_source", warn=False
+        )
+
     def _hydrate_fg_postgres_online_service(self, fg: FeatureGroup) -> None:
         """Set ``FeatureGroup._postgres_online_query_url`` for read-time use.
 

{snowflake_ml_python-1.51.0 → snowflake_ml_python-1.52.0}/snowflake/ml/feature_store/stream_source.py

@@ -179,6 +179,9 @@ class StreamSource:
         self.schema: StructType = schema
         self.owner: Optional[str] = None
         self.desc: str = desc
+        # Runtime-only cache of the Online Service ingest endpoint URL, populated by
+        # FeatureStore.get_stream_source. Not persisted to metadata.
+        self._postgres_online_ingest_url: Optional[str] = None
 
     def _validate(self, name: str, schema: StructType) -> None:
         if len(name) > _STREAM_SOURCE_NAME_LENGTH_LIMIT:

{snowflake_ml_python-1.51.0 → snowflake_ml_python-1.52.0}/snowflake/ml/model/_client/model/model_version_impl.py

@@ -1229,7 +1229,8 @@ class ModelVersion(lineage_node.LineageNode):
             raise exceptions.SnowflakeMLException(
                 error_code=error_codes.INSUFFICIENT_PRIVILEGES,
                 original_exception=RuntimeError(
-                    f"model registry: cannot {operation} this model — only the model's owner role is permitted."
+                    f"model registry: cannot {operation} this model — only the model's owner role is permitted. "
+                    "You may use export() instead."
                 ),
             )
 

{snowflake_ml_python-1.51.0 → snowflake_ml_python-1.52.0}/snowflake/ml/registry/_manager/model_manager.py

@@ -35,6 +35,12 @@ if TYPE_CHECKING:
 
 logger = logging.getLogger(__name__)
 
+# Query-level custom_tags so telemetry can distinguish log_model SQL paths.
+MODEL_LOG_PATH_TAG = "model_log_path"
+MODEL_LOG_PATH_LIVE_COMMIT = "live_commit"
+MODEL_LOG_PATH_FROM_STAGE = "from_stage"
+MODEL_LOG_PATH_LIVE_COMMIT_FALLBACK = "live_commit_fallback"
+
 
 def _validate_user_model_save_options(
     model: type_hints.SupportedModelType,
@@ -248,8 +254,13 @@ class ModelManager:
         checkout_model_name_id = model_name_id
         checkout_version_name_id: Optional[sql_identifier.SqlIdentifier] = None
         rename_model_to_id: Optional[sql_identifier.SqlIdentifier] = None
+        attempted_hidden_live_commit = False
 
         if use_hidden_live_commit:
+            attempted_hidden_live_commit = True
+            statement_params = telemetry.add_statement_params_custom_tags(
+                statement_params, {MODEL_LOG_PATH_TAG: MODEL_LOG_PATH_LIVE_COMMIT}
+            )
             model_action = model_ops.ModelAction.ALTER if model_exists else model_ops.ModelAction.CREATE
             live_version_name_id = live_commit_naming.generate_live_version_name()
             checkout_version_name_id = live_version_name_id
@@ -282,6 +293,14 @@ class ModelManager:
                 checkout_version_name_id = None
                 rename_model_to_id = None
 
+        if not use_hidden_live_commit:
+            model_log_path = (
+                MODEL_LOG_PATH_LIVE_COMMIT_FALLBACK if attempted_hidden_live_commit else MODEL_LOG_PATH_FROM_STAGE
+            )
+            statement_params = telemetry.add_statement_params_custom_tags(
+                statement_params, {MODEL_LOG_PATH_TAG: model_log_path}
+            )
+
         if use_hidden_live_commit:
             assert checkout_version_name_id is not None
             stage_path = self._model_ops.get_model_version_stage_path(

{snowflake_ml_python-1.51.0 → snowflake_ml_python-1.52.0}/snowflake/ml/utils/sparse.py

@@ -40,11 +40,16 @@ def _pandas_to_sparse_pandas(pandas_df: pd.DataFrame, sparse_cols: list[str]) ->
                 indices[key].append(row_i)
                 vals[key].append(value)
 
+        # numpy 2.x stores a str dtype as a fixed-width array, which casts the SparseArray fill value (nan) into a
+        # truncated string (e.g. "nan"). Use object dtype for non-numeric values so the fill stays nan.
+        sparse_array_dtype = object if dtype is str else dtype
         for col_i in range(array_length or 0):
             # for each index create a SparseArray column. For now use sequential number as suffix for column name
             col_name_index = col_name + "_" + str(col_i)
             pandas_df[col_name_index] = pandas_arrays.SparseArray(
-                vals[col_i], pandas_sparse.make_sparse_index(num_rows, indices[col_i], "integer"), dtype=dtype
+                vals[col_i],
+                pandas_sparse.make_sparse_index(num_rows, indices[col_i], "integer"),
+                dtype=sparse_array_dtype,
             )
 
         del pandas_df[col_name]

{snowflake_ml_python-1.51.0 → snowflake_ml_python-1.52.0}/snowflake/ml/version.py

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

{snowflake_ml_python-1.51.0 → snowflake_ml_python-1.52.0}/snowflake_ml_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.51.0
+Version: 1.52.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: 
@@ -416,7 +416,17 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.51.0
+## 1.52.0
+
+### New Features
+
+### Bug Fixes
+
+### Behavior Changes
+
+### Deprecations
+
+## 1.51.0 (2026-08-12)
 
 ### New Features
 
@@ -432,6 +442,9 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 ### Behavior Changes
 
+* Feature Store: `stream_ingest` now reuses the Online Service ingest endpoint cached on the
+  `StreamSource` by `get_stream_source`, avoiding a per-call server status round-trip.
+
 ### Deprecations
 
 ## 1.50.0 (2026-08-03)