snowflake-ml-python 1.42.0__tar.gz → 1.44.0__tar.gz

{snowflake_ml_python-1.42.0 → snowflake_ml_python-1.44.0}/CHANGELOG.md

@@ -1,6 +1,70 @@
 # Release History
 
-## 1.42.0
+## 1.44.0
+
+### New Features
+
+### Bug Fixes
+
+### Behavior Changes
+
+### Deprecations
+
+## 1.43.0 (2026-06-16)
+
+### New Features
+
+* Registry: `inference_engine_options["engine"]` now accepts case-insensitive strings in addition to
+  `InferenceEngine` enum members. Supported values are `"vllm"` and `"python_generic"` (for example,
+  `"vLLM"`, `"VLLM"`, and `"PYTHON_GENERIC"` are all accepted). This applies to `ModelVersion.create_service()`
+  and `ModelVersion.run_batch()`.
+
+```python
+from snowflake.ml.registry import Registry
+
+registry = Registry(session)
+mv = registry.get_model("my_model").version("v1")
+
+mv.create_service(
+    service_name="my_vllm_service",
+    service_compute_pool="GPU_COMPUTE_POOL",
+    gpu_requests="1",
+    inference_engine_options={
+        "engine": "vLLM",
+        "engine_args_override": ["--max-model-len=4096"],
+    },
+)
+
+job = mv.run_batch(
+    compute_pool="GPU_COMPUTE_POOL",
+    X=input_df,
+    output_spec=output_spec,
+    inference_engine_options={
+        "engine": "python_generic",
+    },
+)
+```
+
+### Bug Fixes
+
+* Registry: Fixed `explain()` failing with type mismatch errors when passing a Snowpark DataFrame.
+  Snowpark DataFrame columns are now explicitly cast to match model signature types before SQL
+  generation. Table functions (used by `explain`) enforce stricter type coercion than scalar functions
+  (used by `predict`), so columns that were implicitly coerced for `predict` would be rejected by
+  `explain`.
+
+### Behavior Changes
+
+* Registry: `SentenceTransformers` models no longer hard-code a default inference batch size of 32.
+  When `batch_size` is not specified at `log_model` time and is not overridden at inference, the
+  `sentence-transformers` library default is used. To pin a specific size, pass `batch_size` via
+  `log_model(..., options={"batch_size": N})` or as an inference-time parameter. All
+  `SentenceTransformers` models logged with this release require a client/serving runtime on version
+  1.43.0 or later (regardless of whether `batch_size` was specified).
+
+### Deprecations
+
+## 1.42.0 (2026-06-11)
 
 ### New Features
 

{snowflake_ml_python-1.42.0 → snowflake_ml_python-1.44.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.42.0
+Version: 1.44.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: 
@@ -260,14 +260,14 @@ Requires-Dist: tqdm<5
 Requires-Dist: typing-extensions<5,>=4.1.0
 Requires-Dist: xgboost<4
 Provides-Extra: all
-Requires-Dist: altair<6,>=5; extra == "all"
+Requires-Dist: altair<7,>=5; extra == "all"
 Requires-Dist: catboost<2,>=1.2.0; extra == "all"
 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: mlflow<4,>=2.16.0; extra == "all"
 Requires-Dist: prophet<2,>=1.1.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: sentencepiece<0.3,>=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"
@@ -275,7 +275,7 @@ Requires-Dist: torch<3,>=2.0.1; extra == "all"
 Requires-Dist: torchdata<1,>=0.4; extra == "all"
 Requires-Dist: transformers!=4.51.3,<6,>=4.39.3; extra == "all"
 Provides-Extra: altair
-Requires-Dist: altair<6,>=5; extra == "altair"
+Requires-Dist: altair<7,>=5; extra == "altair"
 Provides-Extra: catboost
 Requires-Dist: catboost<2,>=1.2.0; extra == "catboost"
 Provides-Extra: keras
@@ -285,15 +285,15 @@ Requires-Dist: torch<3,>=2.0.1; extra == "keras"
 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: mlflow<4,>=2.16.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: sentencepiece<0.3,>=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,<6,>=4.39.3; extra == "llm"
 Provides-Extra: mlflow
-Requires-Dist: mlflow<3,>=2.16.0; extra == "mlflow"
+Requires-Dist: mlflow<4,>=2.16.0; extra == "mlflow"
 Provides-Extra: prophet
 Requires-Dist: prophet<2,>=1.1.0; extra == "prophet"
 Provides-Extra: streamlit
@@ -305,7 +305,7 @@ Requires-Dist: torch<3,>=2.0.1; extra == "torch"
 Requires-Dist: torchdata<1,>=0.4; extra == "torch"
 Provides-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: sentencepiece<0.3,>=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,<6,>=4.39.3; extra == "transformers"
@@ -416,7 +416,71 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.42.0
+## 1.44.0
+
+### New Features
+
+### Bug Fixes
+
+### Behavior Changes
+
+### Deprecations
+
+## 1.43.0 (2026-06-16)
+
+### New Features
+
+* Registry: `inference_engine_options["engine"]` now accepts case-insensitive strings in addition to
+  `InferenceEngine` enum members. Supported values are `"vllm"` and `"python_generic"` (for example,
+  `"vLLM"`, `"VLLM"`, and `"PYTHON_GENERIC"` are all accepted). This applies to `ModelVersion.create_service()`
+  and `ModelVersion.run_batch()`.
+
+```python
+from snowflake.ml.registry import Registry
+
+registry = Registry(session)
+mv = registry.get_model("my_model").version("v1")
+
+mv.create_service(
+    service_name="my_vllm_service",
+    service_compute_pool="GPU_COMPUTE_POOL",
+    gpu_requests="1",
+    inference_engine_options={
+        "engine": "vLLM",
+        "engine_args_override": ["--max-model-len=4096"],
+    },
+)
+
+job = mv.run_batch(
+    compute_pool="GPU_COMPUTE_POOL",
+    X=input_df,
+    output_spec=output_spec,
+    inference_engine_options={
+        "engine": "python_generic",
+    },
+)
+```
+
+### Bug Fixes
+
+* Registry: Fixed `explain()` failing with type mismatch errors when passing a Snowpark DataFrame.
+  Snowpark DataFrame columns are now explicitly cast to match model signature types before SQL
+  generation. Table functions (used by `explain`) enforce stricter type coercion than scalar functions
+  (used by `predict`), so columns that were implicitly coerced for `predict` would be rejected by
+  `explain`.
+
+### Behavior Changes
+
+* Registry: `SentenceTransformers` models no longer hard-code a default inference batch size of 32.
+  When `batch_size` is not specified at `log_model` time and is not overridden at inference, the
+  `sentence-transformers` library default is used. To pin a specific size, pass `batch_size` via
+  `log_model(..., options={"batch_size": N})` or as an inference-time parameter. All
+  `SentenceTransformers` models logged with this release require a client/serving runtime on version
+  1.43.0 or later (regardless of whether `batch_size` was specified).
+
+### Deprecations
+
+## 1.42.0 (2026-06-11)
 
 ### New Features
 

{snowflake_ml_python-1.42.0 → snowflake_ml_python-1.44.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,<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",]
+all = [ "altair>=5,<7", "catboost>=1.2.0, <2", "keras>=2.0.0,<4", "lightgbm>=4.1.0, <5", "mlflow>=2.16.0, <4", "prophet>=1.1.0, <2", "sentence-transformers>=2.7.0,<6", "sentencepiece>=0.1.95,<0.3", "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,<7",]
 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,<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",]
+llm = [ "mlflow>=2.16.0, <4", "sentence-transformers>=2.7.0,<6", "sentencepiece>=0.1.95,<0.3", "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, <4",]
 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,<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",]
+transformers = [ "sentence-transformers>=2.7.0,<6", "sentencepiece>=0.1.95,<0.3", "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.42.0 → snowflake_ml_python-1.44.0}/snowflake/ml/_internal/exceptions/error_codes.py

@@ -63,6 +63,8 @@ METHOD_NOT_ALLOWED = "2102"
 NOT_IMPLEMENTED = "2103"
 # User needs to opt in to use a feature.
 OPT_IN_REQUIRED = "2104"
+# Caller's current role lacks the required ownership/privilege to perform the operation.
+INSUFFICIENT_PRIVILEGES = "2105"
 
 # Calling an API with unsupported keywords/values.
 INVALID_ARGUMENT = "2110"

{snowflake_ml_python-1.42.0 → snowflake_ml_python-1.44.0}/snowflake/ml/feature_store/feature_group.py

@@ -71,7 +71,7 @@ if TYPE_CHECKING:
 
     from snowflake.ml.feature_store.feature_store import FeatureStore
     from snowflake.ml.feature_store.spec.models import FeatureViewSpec
-    from snowflake.snowpark import DataFrame
+    from snowflake.snowpark import DataFrame, Session
 
 
 _FEATURE_GROUP_NAME_MAX_LENGTH = 255
@@ -256,7 +256,9 @@ class FeatureGroup:
     # Spec construction
     # -----------------------------------------------------------------------
 
-    def _to_spec(self, *, database: str, schema: str, version: str) -> FeatureViewSpec:
+    def _to_spec(
+        self, *, database: str, schema: str, version: str, session: Optional[Session] = None
+    ) -> FeatureViewSpec:
         """Translate this draft FeatureGroup into a validated spec payload.
 
         Builds an internal ``(name, version)``-keyed prefix map and delegates
@@ -269,6 +271,10 @@ class FeatureGroup:
             version: User-facing FeatureGroup version (already validated by
                 :class:`FeatureGroupVersion`). Embedded in the spec metadata
                 and surfaced to the Online Service Query API at read time.
+            session: Optional Snowpark session used to read upstream
+                FeatureView column shapes from the materialized DT/View, so
+                source column shapes match the upstream's stored
+                ``OutputColumn`` for the Online Service exact-shape check.
 
         Returns:
             FeatureViewSpec: validated spec for this FeatureGroup, ready for
@@ -288,6 +294,7 @@ class FeatureGroup:
             schema=schema,
             name=self._name,
             version=version,
+            session=session,
         )
         builder.set_sources(list(self._features))
         builder.set_source_prefixes(prefix_map)
@@ -1109,6 +1116,7 @@ def register_feature_group(fs: FeatureStore, feature_group: FeatureGroup, versio
         database=fs._config.database.resolved(),
         schema=fs._config.schema.resolved(),
         version=validated_version,
+        session=fs._session,
     )
     primary_key = list(spec.spec.ordered_entity_column_names)
     assert primary_key, "FeatureGroup spec must derive at least one entity column"