snowflake-ml-python 2.5.0__tar.gz → 2.6.0__tar.gz

{snowflake_ml_python-2.5.0 → snowflake_ml_python-2.6.0}/CHANGELOG.md

@@ -1,9 +1,32 @@
 # Release History
 
-## 2.5.0
+## 2.6.0
 
 ### New Features
 
+* ML Jobs: Added `register_code_bundle` and `register_job_from_code_bundle` (Public Preview) to
+  create a Snowflake code bundle and run ML Jobs from it. `delete_code_bundle` drops a registered bundle.
+* Feature Store: The Postgres-backed Online Feature Store is now generally available, including tiled aggregations,
+  streaming and realtime feature views and feature groups.
+
+### Bug Fixes
+
+* ML Jobs: `submit_from_stage` accepts stage paths with double-quoted segments, such as a Git repository branch whose
+  name contains `/` (`@repo/branches/"feature/x"/src`). These paths used to fail with "does not exist".
+* ML Jobs: `MLJob.result()` no longer returns another job's result when results are fetched concurrently on one
+  machine.
+
+### Behavior Changes
+
+### Deprecations
+
+## 2.5.0 (2026-10-14)
+
+### New Features
+
+* Registry: `huggingface.TransformersPipeline` can omit `task`; when omitted, the task is inferred
+  automatically from the model's Hugging Face Hub `pipeline_tag` (requires `huggingface_hub`).
+
 ### Bug Fixes
 
 * Modeling: The `snowflake.ml.modeling` estimators and preprocessing transformers are not deprecated.
@@ -16,6 +39,17 @@
 
 ### Behavior Changes
 
+* Registry: `huggingface.TransformersPipeline` now takes required `model` as the first argument;
+  `task` is optional and inferred when not provided. Prefer
+  `TransformersPipeline(model="org/model")` over passing `task` explicitly.
+* Registry: Removed deprecated `huggingface_pipeline.HuggingFacePipelineModel` and its
+  `log_model_and_create_service` API. Use `huggingface.TransformersPipeline` with
+  `registry.log_model` / `ModelVersion.create_service` instead. A pickle-compat alias remains
+  for loading models logged with the old class.
+* Feature Store: Batch feature views with a Postgres online store now default `OnlineConfig.target_lag` to
+  `"1 minute"` instead of `"10 seconds"`. Setting a Postgres `target_lag` below 1 minute emits a `UserWarning`.
+  Hybrid table online stores are unchanged.
+
 ### Deprecations
 
 ## 2.4.0 (2026-10-06)

{snowflake_ml_python-2.5.0 → snowflake_ml_python-2.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 2.5.0
+Version: 2.6.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: 
@@ -426,10 +426,33 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 2.5.0
+## 2.6.0
 
 ### New Features
 
+* ML Jobs: Added `register_code_bundle` and `register_job_from_code_bundle` (Public Preview) to
+  create a Snowflake code bundle and run ML Jobs from it. `delete_code_bundle` drops a registered bundle.
+* Feature Store: The Postgres-backed Online Feature Store is now generally available, including tiled aggregations,
+  streaming and realtime feature views and feature groups.
+
+### Bug Fixes
+
+* ML Jobs: `submit_from_stage` accepts stage paths with double-quoted segments, such as a Git repository branch whose
+  name contains `/` (`@repo/branches/"feature/x"/src`). These paths used to fail with "does not exist".
+* ML Jobs: `MLJob.result()` no longer returns another job's result when results are fetched concurrently on one
+  machine.
+
+### Behavior Changes
+
+### Deprecations
+
+## 2.5.0 (2026-10-14)
+
+### New Features
+
+* Registry: `huggingface.TransformersPipeline` can omit `task`; when omitted, the task is inferred
+  automatically from the model's Hugging Face Hub `pipeline_tag` (requires `huggingface_hub`).
+
 ### Bug Fixes
 
 * Modeling: The `snowflake.ml.modeling` estimators and preprocessing transformers are not deprecated.
@@ -442,6 +465,17 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 ### Behavior Changes
 
+* Registry: `huggingface.TransformersPipeline` now takes required `model` as the first argument;
+  `task` is optional and inferred when not provided. Prefer
+  `TransformersPipeline(model="org/model")` over passing `task` explicitly.
+* Registry: Removed deprecated `huggingface_pipeline.HuggingFacePipelineModel` and its
+  `log_model_and_create_service` API. Use `huggingface.TransformersPipeline` with
+  `registry.log_model` / `ModelVersion.create_service` instead. A pickle-compat alias remains
+  for loading models logged with the old class.
+* Feature Store: Batch feature views with a Postgres online store now default `OnlineConfig.target_lag` to
+  `"1 minute"` instead of `"10 seconds"`. Setting a Postgres `target_lag` below 1 minute emits a `UserWarning`.
+  Hybrid table online stores are unchanged.
+
 ### Deprecations
 
 ## 2.4.0 (2026-10-06)

snowflake_ml_python-2.6.0/snowflake/cortex/_classify_text.py

@@ -0,0 +1,99 @@
+import json
+from typing import Any, cast
+
+from typing_extensions import deprecated
+
+from snowflake import snowpark
+from snowflake.cortex._util import (
+    CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
+    SnowflakeAuthenticationException,
+    warn_cortex_deprecated,
+)
+from snowflake.ml._internal import telemetry
+from snowflake.snowpark import context, functions
+
+_AI_CLASSIFY_FUNCTION_NAME = "AI_CLASSIFY"
+# AI_CLASSIFY requires its config argument to be a constant object literal. A Python dict is rendered by
+# Snowpark as PARSE_JSON('...')::OBJECT, which the server rejects ("... is not a constant"), so emit the
+# config as a SQL object constant via sql_expr -- this mirrors snowflake.snowpark.functions.ai_classify.
+# Legacy CLASSIFY_TEXT was always single-label, so set output_mode explicitly to keep that contract.
+_AI_CLASSIFY_SINGLE_LABEL_CONFIG_SQL = "{'output_mode': 'single'}"
+
+
+def _adapt_ai_classify_result(result: Any) -> str:
+    """Map an AI_CLASSIFY result to the legacy CLASSIFY_TEXT {"label": ...} JSON string."""
+    # Used for non-Column results (e.g. a dict/JSON string): reproduce the legacy single-label object shape.
+    parsed: Any = result
+    if isinstance(result, str):
+        try:
+            parsed = json.loads(result)
+        except json.JSONDecodeError:
+            return result
+    label: Any = None
+    if isinstance(parsed, dict):
+        labels = parsed.get("labels")
+        if isinstance(labels, list) and labels:
+            label = labels[0]
+        elif parsed.get("label") is not None:
+            label = parsed.get("label")
+    return json.dumps({"label": label})
+
+
+def _ai_classify_label_column(
+    str_input: str | snowpark.Column,
+    categories: list[str] | snowpark.Column,
+) -> snowpark.Column:
+    """Call AI_CLASSIFY in single-label mode and re-wrap to the legacy CLASSIFY_TEXT {"label": ...} OBJECT."""
+    result = functions.builtin(_AI_CLASSIFY_FUNCTION_NAME)(
+        str_input, categories, functions.sql_expr(_AI_CLASSIFY_SINGLE_LABEL_CONFIG_SQL)
+    )
+    # AI_CLASSIFY returns {"labels": [...]}; collapse to the legacy single {"label": ...}. Subscript access
+    # (not functions.get with a bare string, which Snowpark treats as a column identifier) uses literal keys.
+    label = functions.coalesce(result["labels"][0], result["label"])
+    return functions.object_construct(functions.lit("label"), label)
+
+
+@telemetry.send_api_usage_telemetry(
+    project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
+)
+def classify_text(
+    str_input: str | snowpark.Column,
+    categories: list[str] | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> str | snowpark.Column:
+    """Use the LLM inference service to classify the INPUT text into one of the target CATEGORIES.
+
+    Args:
+        str_input: A Column of strings to classify.
+        categories: A list of candidate categories to classify the INPUT text into.
+        session: The snowpark session to use. Will be inferred by context if not specified.
+
+    Returns:
+        A column of classification responses (legacy CLASSIFY_TEXT {"label": ...} shape).
+
+    Raises:
+        SnowflakeAuthenticationException: if no session is provided and none is active.
+    """
+    warn_cortex_deprecated("classify_text", "ai_classify")
+    column = _ai_classify_label_column(str_input, categories)
+    if isinstance(str_input, snowpark.Column) or isinstance(categories, snowpark.Column):
+        return column
+
+    # Immediate (literal) path: evaluate and return the collected OBJECT as a JSON string, exactly as the
+    # legacy CLASSIFY_TEXT immediate path did.
+    session = session or context.get_active_session()
+    if session is None:
+        raise SnowflakeAuthenticationException(
+            """Session required. Provide the session through a session=... argument or ensure an active session is
+            available in your environment."""
+        )
+    return cast(str, session.create_dataframe([snowpark.Row()]).select(column).collect()[0][0])
+
+
+ClassifyText = deprecated(
+    "ClassifyText() is deprecated and will be removed in a future release. Please use classify_text() instead."
+)(
+    telemetry.send_api_usage_telemetry(
+        project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
+    )(classify_text)
+)

{snowflake_ml_python-2.5.0 → snowflake_ml_python-2.6.0}/snowflake/ml/_internal/platform_capabilities.py

@@ -93,7 +93,7 @@ class PlatformCapabilities:
         return self._get_bool_feature(ENABLE_MODEL_METHOD_SIGNATURE_PARAMETERS, False)
 
     def is_lora_adapters_enabled(self) -> bool:
-        return False
+        return self._get_bool_feature(ENABLE_LORA_ADAPTERS, False)
 
     def is_single_output_native_type_enabled(self) -> bool:
         """Whether a single, non-OBJECT model FUNCTION output is registered with its native SQL type.

{snowflake_ml_python-2.5.0 → snowflake_ml_python-2.6.0}/snowflake/ml/dataset/dataset.py

@@ -341,7 +341,7 @@ class Dataset(lineage_node.LineageNode):
         metadata = dataset_metadata.DatasetMetadata(
             source_query=source_query,
             owner=self._session.sql("SELECT CURRENT_USER()").collect(statement_params=_TELEMETRY_STATEMENT_PARAMS)[0][
-                "CURRENT_USER()"
+                0
             ],
             exclude_cols=exclude_cols,
             label_cols=label_cols,

snowflake_ml_python-2.6.0/snowflake/ml/feature_store/decl/__init__.py

@@ -0,0 +1,168 @@
+"""Public API for the declarative feature store library.
+
+Import the key functions and types from this package:
+
+    from snowflake.ml.feature_store.decl import (
+        load_specs,
+        validate_specs,
+        generate_plan,
+        fetch_applied_state,
+        SpecBatch,
+        AppliedState,
+        Plan,
+        PlanOptions,
+        ValidationResult,
+        PlanOp,
+        AppliedObject,
+        SpecLoadError,
+        ValidationError,
+        DependencyError,
+        StateDriftError,
+        FSBaseType,
+        FeatureViewKind,
+        SourceType,
+        FeatureAggregationMethod,
+        OpKind,
+        TYPE_ALIASES,
+        normalize_type,
+        FSColumn,
+        SpecBase,
+        Entity,
+        StreamingSource,
+        BatchSource,
+        SourceRef,
+        UDF,
+        Feature,
+        FeatureView,
+        FeatureViewRef,
+        FeatureGroup,
+    )
+"""
+
+try:
+    from snowflake.ml.feature_store.decl.api import (
+        discover_project,
+        fetch_applied_state,
+        generate_plan,
+        load_manifest,
+        load_project,
+        load_specs,
+        resolve_datasource_columns,
+        resolve_target,
+        validate_specs,
+    )
+except ModuleNotFoundError:  # lower-layer BUILD targets omit api.py
+    pass
+from snowflake.ml.feature_store.decl.enums import (
+    TYPE_ALIASES,
+    FeatureAggregationMethod,
+    FeatureViewKind,
+    FSBaseType,
+    OpKind,
+    SourceType,
+    normalize_type,
+)
+from snowflake.ml.feature_store.decl.errors import (
+    DependencyError,
+    SpecLoadError,
+    StateDriftError,
+    ValidationError,
+)
+from snowflake.ml.feature_store.decl.manifest import (
+    FSManifest,
+    FSTarget,
+    FSTemplating,
+    InvalidManifestError,
+    ManifestConfigurationError,
+    ManifestNotFoundError,
+    TargetContext,
+)
+from snowflake.ml.feature_store.decl.project_paths import FSProjectPaths
+from snowflake.ml.feature_store.decl.spec_models import (
+    UDF,
+    Backfill,
+    BatchFeatureView,
+    BatchSource,
+    Entity,
+    Feature,
+    FeatureGroup,
+    FeatureView,
+    FeatureViewRef,
+    FSColumn,
+    RealtimeFeatureView,
+    SourceRef,
+    SpecBase,
+    StorageConfig,
+    StreamingFeatureView,
+    StreamingSource,
+)
+from snowflake.ml.feature_store.decl.types import (
+    AppliedObject,
+    AppliedState,
+    Plan,
+    PlanOp,
+    PlanOptions,
+    SpecBatch,
+    ValidationResult,
+)
+
+__all__ = [
+    # api.py
+    "load_specs",
+    "validate_specs",
+    "generate_plan",
+    "fetch_applied_state",
+    "resolve_datasource_columns",
+    "load_manifest",
+    "discover_project",
+    "resolve_target",
+    "load_project",
+    # enums.py
+    "FSBaseType",
+    "FeatureViewKind",
+    "SourceType",
+    "FeatureAggregationMethod",
+    "OpKind",
+    "TYPE_ALIASES",
+    "normalize_type",
+    # errors.py
+    "SpecLoadError",
+    "ValidationError",
+    "DependencyError",
+    "StateDriftError",
+    # manifest.py
+    "FSManifest",
+    "FSTarget",
+    "FSTemplating",
+    "TargetContext",
+    "ManifestNotFoundError",
+    "InvalidManifestError",
+    "ManifestConfigurationError",
+    # project_paths.py
+    "FSProjectPaths",
+    # spec_models.py
+    "FSColumn",
+    "SpecBase",
+    "Entity",
+    "StreamingSource",
+    "BatchSource",
+    "SourceRef",
+    "UDF",
+    "Feature",
+    "FeatureView",
+    "StreamingFeatureView",
+    "BatchFeatureView",
+    "RealtimeFeatureView",
+    "FeatureViewRef",
+    "FeatureGroup",
+    "Backfill",
+    "StorageConfig",
+    # types.py
+    "AppliedObject",
+    "AppliedState",
+    "PlanOp",
+    "Plan",
+    "ValidationResult",
+    "PlanOptions",
+    "SpecBatch",
+]