snowflake-ml-python 1.41.0__tar.gz → 1.43.0__tar.gz

{snowflake_ml_python-1.41.0 → snowflake_ml_python-1.43.0}/CHANGELOG.md

@@ -1,6 +1,84 @@
 # Release History
 
-## 1.41.0
+## 1.43.0
+
+### 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
+
+* Registry: Support `truncate_dim` for `SentenceTransformers` models. Truncation set at
+  construction (`SentenceTransformer(..., truncate_dim=N)`) is captured when the model is logged
+  and restored on reload. On clients with `sentence-transformers` 5.0.0 or later, `truncate_dim`
+  is also exposed as a runtime parameter in the model signature.
+
+* Registry: `log_model()` now captures a representative row from `sample_input_data` and stores it alongside the model
+  so inference code snippets shown in the model registry UI can be pre-filled with realistic values. Pass
+  `options={"capture_sample_input_data": False}` to opt out (e.g., for sensitive data); generic placeholder values
+  will be used in the snippets instead.
+
+### Bug Fixes
+
+* Registry: Pin `transformers<5` for additional HuggingFace pipeline tasks removed in
+  `transformers 5.x` (`summarization`, `text2text-generation`, `question-answering`, and
+  `translation_*`), preventing deployment failures from missing pipeline classes.
+
+### Behavior Changes
+
+### Deprecations
+
+## 1.41.0 (2026-05-27)
 
 ### New Features
 
@@ -26,6 +104,14 @@
 
 ### Behavior Changes
 
+* Registry: `log_model(..., options=...)` now rejects unknown top-level keys and unknown keys inside each
+  `method_options` entry (for example misspellings or options that belong to a different model framework).
+  Previously those keys were ignored. Validation is based on the public save-option TypedDicts and runs
+  before dependency reconciliation.
+
+* Registry: `use_gpu` is declared on the CatBoost, XGBoost, PyTorch, and TorchScript save-option
+  TypedDicts to match the save paths that already read this flag.
+
 ### Deprecations
 
 ## 1.39.0

{snowflake_ml_python-1.41.0 → snowflake_ml_python-1.43.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.41.0
+Version: 1.43.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: 
@@ -241,10 +241,11 @@ 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: orjson<4,>=3.11.0
 Requires-Dist: packaging<26,>=20.9
 Requires-Dist: pandas<3,>=2.1.4
 Requires-Dist: platformdirs<5
-Requires-Dist: pyarrow<19.0.0
+Requires-Dist: pyarrow
 Requires-Dist: pydantic<3,>=2.8.2
 Requires-Dist: pyjwt<3,>=2.0.0
 Requires-Dist: pytimeparse<2,>=1.1.8
@@ -416,7 +417,85 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.41.0
+## 1.43.0
+
+### 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
+
+* Registry: Support `truncate_dim` for `SentenceTransformers` models. Truncation set at
+  construction (`SentenceTransformer(..., truncate_dim=N)`) is captured when the model is logged
+  and restored on reload. On clients with `sentence-transformers` 5.0.0 or later, `truncate_dim`
+  is also exposed as a runtime parameter in the model signature.
+
+* Registry: `log_model()` now captures a representative row from `sample_input_data` and stores it alongside the model
+  so inference code snippets shown in the model registry UI can be pre-filled with realistic values. Pass
+  `options={"capture_sample_input_data": False}` to opt out (e.g., for sensitive data); generic placeholder values
+  will be used in the snippets instead.
+
+### Bug Fixes
+
+* Registry: Pin `transformers<5` for additional HuggingFace pipeline tasks removed in
+  `transformers 5.x` (`summarization`, `text2text-generation`, `question-answering`, and
+  `translation_*`), preventing deployment failures from missing pipeline classes.
+
+### Behavior Changes
+
+### Deprecations
+
+## 1.41.0 (2026-05-27)
 
 ### New Features
 
@@ -442,6 +521,14 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 ### Behavior Changes
 
+* Registry: `log_model(..., options=...)` now rejects unknown top-level keys and unknown keys inside each
+  `method_options` entry (for example misspellings or options that belong to a different model framework).
+  Previously those keys were ignored. Validation is based on the public save-option TypedDicts and runs
+  before dependency reconciliation.
+
+* Registry: `use_gpu` is declared on the CatBoost, XGBoost, PyTorch, and TorchScript save-option
+  TypedDicts to match the save paths that already read this flag.
+
 ### Deprecations
 
 ## 1.39.0

{snowflake_ml_python-1.41.0 → snowflake_ml_python-1.43.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", "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", "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", "orjson>=3.11.0,<4", "packaging>=20.9,<26", "pandas>=2.1.4,<3", "platformdirs<5", "pyarrow", "pydantic>=2.8.2, <3", "pyjwt>=2.0.0, <3", "pytimeparse>=1.1.8,<2", "pyyaml>=6.0,<7", "retrying>=1.3.3,<2", "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.41.0 → snowflake_ml_python-1.43.0}/snowflake/ml/_internal/env_utils.py

@@ -57,20 +57,22 @@ def get_execution_context() -> str:
 
 
 def is_local_conda_environment() -> bool:
-    """Return True if the active Python interpreter appears to run inside a conda env
-    or under conda session context (e.g. a venv created inside an activated conda env).
+    """Return True if the active Python interpreter appears to run inside a conda env.
 
-    Uses the presence of a ``conda-meta`` directory under :py:data:`sys.prefix`, which is
-    standard for conda-created environments. A nested venv typically has ``sys.prefix``
-    pointing at the venv without ``conda-meta``; conda still sets ``CONDA_PREFIX`` when
-    an environment is activated, so a non-empty ``CONDA_PREFIX`` is treated as conda context.
+    Uses ``conda-meta`` under :py:data:`sys.prefix` for conda-created environments.
+    Standard-library virtual environments (``pyvenv.cfg`` present) are **not** treated as
+    conda even when ``CONDA_PREFIX`` is still set in the shell (e.g. conda base
+    auto-activation while a separate venv kernel is selected).
 
     Returns:
-        ``True`` if ``sys.prefix`` contains ``conda-meta`` or ``CONDA_PREFIX`` is set and
-        non-empty, else ``False``.
+        ``True`` if ``sys.prefix`` contains ``conda-meta``, or ``CONDA_PREFIX`` is set and
+        there is no ``pyvenv.cfg`` under ``sys.prefix``; else ``False``.
     """
-    if os.path.isdir(os.path.join(sys.prefix, "conda-meta")):
+    prefix = sys.prefix
+    if os.path.isdir(os.path.join(prefix, "conda-meta")):
         return True
+    if os.path.isfile(os.path.join(prefix, "pyvenv.cfg")):
+        return False
     return bool(os.environ.get("CONDA_PREFIX", "").strip())
 
 

{snowflake_ml_python-1.41.0 → snowflake_ml_python-1.43.0}/snowflake/ml/feature_store/_compute_fn_validation.py

@@ -31,7 +31,7 @@ import textwrap
 from typing import Any, Callable, Optional
 
 # Top-level packages permitted in import statements inside compute_fn.
-ALLOWED_MODULES: frozenset[str] = frozenset({"numpy", "pandas", "re", "copy"})
+ALLOWED_MODULES: frozenset[str] = frozenset({"numpy", "pandas", "re", "copy", "dataclasses"})
 
 # Names that may appear as free globals inside compute_fn. Mirrors the
 # runtime namespace populated by RTFV reconstruction so the same source

{snowflake_ml_python-1.41.0 → snowflake_ml_python-1.43.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"