snowflake-ml-python 1.15.0__py3-none-any.whl → 1.16.0__py3-none-any.whl

snowflake/ml/registry/_manager/model_parameter_reconciler.py

@@ -11,6 +11,7 @@ from snowflake.ml._internal.exceptions import error_codes, exceptions
 from snowflake.ml._internal.utils import sql_identifier
 from snowflake.ml.model import target_platform, type_hints as model_types
 from snowflake.ml.model._model_composer.model_manifest import model_manifest_schema
+from snowflake.ml.model.volatility import DEFAULT_VOLATILITY_BY_MODEL_TYPE, Volatility
 from snowflake.snowpark import Session
 from snowflake.snowpark._internal import utils as snowpark_utils
 
@@ -34,6 +35,7 @@ class ModelParameterReconciler:
 
     def __init__(
         self,
+        model: model_types.SupportedModelType,
         session: Session,
         database_name: sql_identifier.SqlIdentifier,
         schema_name: sql_identifier.SqlIdentifier,
@@ -45,6 +47,7 @@ class ModelParameterReconciler:
         python_version: Optional[str] = None,
         statement_params: Optional[dict[str, str]] = None,
     ) -> None:
+        self._model = model
         self._session = session
         self._database_name = database_name
         self._schema_name = schema_name
@@ -67,6 +70,7 @@ class ModelParameterReconciler:
         reconciled_target_platforms = self._reconcile_target_platforms()
         reconciled_options = self._reconcile_explainability_options(reconciled_target_platforms)
         reconciled_options = self._reconcile_relax_version(reconciled_options, reconciled_target_platforms)
+        reconciled_options = self._reconcile_volatility_defaults(reconciled_options)  # ADD THIS LINE
 
         return ReconciledParameters(
             conda_dependencies=self._conda_dependencies,
@@ -292,3 +296,26 @@ class ModelParameterReconciler:
             )
 
         return options
+
+    def _get_default_volatility_for_model(self, model: model_types.SupportedModelType) -> Volatility:
+        """Get default volatility for a model based on its type."""
+        from snowflake.ml.model._packager import model_handler
+
+        handler = model_handler.find_handler(model)
+        # default to IMMUTABLE if no handler found or handler type not in defaults
+        if not handler or handler.HANDLER_TYPE not in DEFAULT_VOLATILITY_BY_MODEL_TYPE:
+            return Volatility.IMMUTABLE
+        return DEFAULT_VOLATILITY_BY_MODEL_TYPE[handler.HANDLER_TYPE]
+
+    def _reconcile_volatility_defaults(self, options: model_types.ModelSaveOption) -> model_types.ModelSaveOption:
+        """Set global default volatility based on model type."""
+
+        # Skip if default_volatility is already explicitly set
+        if "volatility" in options:
+            return options
+
+        # Get default volatility for this model type
+        default_volatility = self._get_default_volatility_for_model(self._model)
+        options["volatility"] = default_volatility
+
+        return options

snowflake/ml/registry/registry.py

@@ -235,6 +235,10 @@ class Registry:
                     - max_batch_size: Maximum batch size that the method could accept in the Snowflake Warehouse.
                         Defaults to None, determined automatically by Snowflake.
                     - function_type: One of supported model method function types (FUNCTION or TABLE_FUNCTION).
+                    - volatility: Volatility level for the function (use Volatility.VOLATILE or Volatility.IMMUTABLE).
+                        Volatility.VOLATILE functions may return different results for the same arguments, while
+                        Volatility.IMMUTABLE functions always return the same result for the same arguments.
+                        Defaults to None (no volatility specified).
         Returns:
             ModelVersion: ModelVersion object corresponding to the model just logged.
         """
@@ -366,6 +370,12 @@ class Registry:
                     Warehouse. It detects any ==x.y.z in specifiers and replaced with >=x.y, <(x+1). Defaults to True.
                 - function_type: Set the method function type globally. To set method function types individually see
                   function_type in model_options.
+                - volatility: Set the volatility for all model methods globally (use Volatility.VOLATILE or
+                  Volatility.IMMUTABLE). Volatility.VOLATILE functions may return different results for the same
+                  arguments, while Volatility.IMMUTABLE functions always return the same result for the same arguments.
+                  Defaults are set automatically based on model type: supported models (sklearn, xgboost, pytorch,
+                  huggingface_pipeline, mlflow, etc.) default to IMMUTABLE, while custom models default to VOLATILE.
+                  Individual method volatility can be set in method_options and will override this global setting.
                 - target_methods: List of target methods to register when logging the model.
                   This option is not used in MLFlow models. Defaults to None, in which case the model handler's
                   default target methods will be used.
@@ -382,6 +392,11 @@ class Registry:
                     - max_batch_size: Maximum batch size that the method could accept in the Snowflake Warehouse.
                         Defaults to None, determined automatically by Snowflake.
                     - function_type: One of supported model method function types (FUNCTION or TABLE_FUNCTION).
+                    - volatility: Volatility level for the function (use Volatility.VOLATILE or Volatility.IMMUTABLE).
+                        Volatility.VOLATILE functions may return different results for the same arguments, while
+                        Volatility.IMMUTABLE functions always return the same result for the same arguments.
+                        This per-method setting overrides any global volatility setting.
+                        Defaults to None (no volatility specified).
 
         Raises:
             ValueError: If extra arguments are specified ModelVersion is provided.

snowflake/ml/utils/authentication.py

@@ -73,3 +73,19 @@ class SnowflakeJWTTokenAuth(auth.AuthBase):
     def __call__(self, r: requests.PreparedRequest) -> requests.PreparedRequest:
         r.headers["Authorization"] = f'Snowflake Token="{self.snowflake_token}"'
         return r
+
+
+class SnowflakePATAuth(auth.AuthBase):
+    """Authentication using Snowflake Programmatic Access Token (PAT)."""
+
+    def __init__(self, pat_token: str) -> None:
+        """Initialize with a PAT token.
+
+        Args:
+            pat_token: The programmatic access token string
+        """
+        self.pat_token = pat_token
+
+    def __call__(self, r: requests.PreparedRequest) -> requests.PreparedRequest:
+        r.headers["Authorization"] = f'Snowflake Token="{self.pat_token}"'
+        return r

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.15.0"
+VERSION = "1.16.0"

{snowflake_ml_python-1.15.0.dist-info → snowflake_ml_python-1.16.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.15.0
+Version: 1.16.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: 
@@ -250,7 +250,7 @@ Requires-Dist: pytimeparse<2,>=1.1.8
 Requires-Dist: pyyaml<7,>=6.0
 Requires-Dist: retrying<2,>=1.3.3
 Requires-Dist: s3fs<2026,>=2024.6.1
-Requires-Dist: scikit-learn<1.7
+Requires-Dist: scikit-learn<1.8
 Requires-Dist: scipy<2,>=1.9
 Requires-Dist: shap<1,>=0.46.0
 Requires-Dist: snowflake-connector-python[pandas]<4,>=3.16.0
@@ -406,7 +406,45 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.15.0
+## 1.16.0
+
+### Bug Fixes
+
+* Registry: Remove redundant pip dependency warnings when `artifact_repository_map` is provided for warehouse model deployments.
+
+### Behavior Changes
+
+### New Features
+
+* Support scikit-learn < 1.8.
+* ML Job: Added support for configuring the runtime image via `runtime_environment`
+  (image tag or full image URL) at submission time.
+  Examples:
+  * @remote(compute_pool, stage_name = 'payload_stage', runtime_environment = '1.8.0')
+  * submit_file('/path/to/repo/test.py', compute_pool, stage_name = 'payload_stage', runtime_environment = '/mydb/myschema/myrepo/myimage:latest')
+* Registry: Ability to mark model methods as `Volatility.VOLATILE` or `Volatility.IMMUTABLE`.
+
+```python
+from snowflake.ml.model.volatility import Volatility
+
+options = {
+    "embed_local_ml_library": True,
+    "relax_version": True,
+    "save_location": "/path/to/my/directory",
+    "function_type": "TABLE_FUNCTION",
+    "volatility": Volatility.IMMUTABLE,
+    "method_options": {
+        "predict": {
+            "case_sensitive": False,
+            "max_batch_size": 100,
+            "function_type": "TABLE_FUNCTION",
+            "volatility": Volatility.VOLATILE,
+        },
+}
+
+````
+
+## 1.15.0 (09-29-2025)
 
 ### Bug Fixes