snowflake-ml-python 1.8.6__py3-none-any.whl → 1.9.0__py3-none-any.whl

snowflake/ml/registry/registry.py

@@ -1,3 +1,5 @@
+import logging
+import os
 import warnings
 from types import ModuleType
 from typing import Any, Optional, Union, overload
@@ -11,6 +13,7 @@ from snowflake.ml.model import (
     Model,
     ModelVersion,
     model_signature,
+    task,
     type_hints as model_types,
 )
 from snowflake.ml.model._client.model import model_version_impl
@@ -29,6 +32,52 @@ _MODEL_MONITORING_DISABLED_ERROR = (
 )
 
 
+class _NullStatusContext:
+    """A fallback context manager that logs status updates."""
+
+    def __init__(self, label: str) -> None:
+        self._label = label
+
+    def __enter__(self) -> "_NullStatusContext":
+        logging.info(f"Starting: {self._label}")
+        return self
+
+    def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
+        pass
+
+    def update(self, label: str, *, state: str = "running", expanded: bool = True) -> None:
+        """Update the status by logging the message."""
+        logging.info(f"Status update: {label} (state: {state})")
+
+
+class RegistryEventHandler:
+    def __init__(self) -> None:
+        try:
+            import streamlit as st
+
+            if not st.runtime.exists():
+                self._streamlit = None
+            else:
+                self._streamlit = st
+            USE_STREAMLIT_WIDGETS = os.getenv("USE_STREAMLIT_WIDGETS", "1") == "1"
+            if not USE_STREAMLIT_WIDGETS:
+                self._streamlit = None
+        except ImportError:
+            self._streamlit = None
+
+    def update(self, message: str) -> None:
+        """Write a message using streamlit if available, otherwise do nothing."""
+        if self._streamlit is not None:
+            self._streamlit.write(message)
+
+    def status(self, label: str, *, state: str = "running", expanded: bool = True) -> Any:
+        """Context manager that provides status updates with optional enhanced display capabilities."""
+        if self._streamlit is None:
+            return _NullStatusContext(label)
+        else:
+            return self._streamlit.status(label, state=state, expanded=expanded)
+
+
 class Registry:
     @telemetry.send_api_usage_telemetry(project=_TELEMETRY_PROJECT, subproject=_MODEL_TELEMETRY_SUBPROJECT)
     def __init__(
@@ -136,7 +185,7 @@ class Registry:
         user_files: Optional[dict[str, list[str]]] = None,
         code_paths: Optional[list[str]] = None,
         ext_modules: Optional[list[ModuleType]] = None,
-        task: model_types.Task = model_types.Task.UNKNOWN,
+        task: model_types.Task = task.Task.UNKNOWN,
         options: Optional[model_types.ModelSaveOption] = None,
     ) -> ModelVersion:
         """
@@ -159,12 +208,12 @@ class Registry:
                 to specify a dependency. It is a recommended way to specify your dependencies using conda. When channel
                 is not specified, Snowflake Anaconda Channel will be used. Defaults to None.
             pip_requirements: List of Pip package specifications. Defaults to None.
-                Models with pip requirements are currently only runnable in Snowpark Container Services.
-                See https://docs.snowflake.com/en/developer-guide/snowflake-ml/model-registry/container for more.
-                Models with pip requirements specified will not be executable in Snowflake Warehouse where all
-                dependencies must be retrieved from Snowflake Anaconda Channel.
+                Models running in a Snowflake Warehouse must also specify a pip artifact repository (see
+                `artifact_repository_map`). Otherwise, models with pip requirements are runnable only in Snowpark
+                Container Services. See
+                https://docs.snowflake.com/en/developer-guide/snowflake-ml/model-registry/container for more.
             artifact_repository_map: Specifies a mapping of package channels or platforms to custom artifact
-                repositories. Defaults to None. Currently, the mapping applies only to warehouse execution.
+                repositories. Defaults to None. Currently, the mapping applies only to Warehouse execution.
                 Note : This feature is currently in Public Preview.
                 Format: {channel_name: artifact_repository_name}, where:
                    - channel_name: Currently must be 'pip'.
@@ -172,10 +221,13 @@ class Registry:
                      `snowflake.snowpark.pypi_shared_repository`.
             resource_constraint: Mapping of resource constraint keys and values, e.g. {"architecture": "x86"}.
             target_platforms: List of target platforms to run the model. The only acceptable inputs are a combination of
-                "WAREHOUSE" and "SNOWPARK_CONTAINER_SERVICES":
-                - ["WAREHOUSE"]  (Warehouse only)
-                - ["SNOWPARK_CONTAINER_SERVICES"]  (Snowpark Container Services only)
-                - ["WAREHOUSE", "SNOWPARK_CONTAINER_SERVICES"] (Both)
+                "WAREHOUSE" and "SNOWPARK_CONTAINER_SERVICES", or a target platform constant:
+                - ["WAREHOUSE"] or snowflake.ml.model.target_platform.WAREHOUSE_ONLY (Warehouse only)
+                - ["SNOWPARK_CONTAINER_SERVICES"] or
+                  snowflake.ml.model.target_platform.SNOWPARK_CONTAINER_SERVICES_ONLY
+                  (Snowpark Container Services only)
+                - ["WAREHOUSE", "SNOWPARK_CONTAINER_SERVICES"] or
+                  snowflake.ml.model.target_platform.BOTH_WAREHOUSE_AND_SNOWPARK_CONTAINER_SERVICES (Both)
                 Defaults to None. When None, the target platforms will be both.
             python_version: Python version in which the model is run. Defaults to None.
             signatures: Model data signatures for inputs and outputs for various target methods. If it is None,
@@ -280,7 +332,7 @@ class Registry:
         user_files: Optional[dict[str, list[str]]] = None,
         code_paths: Optional[list[str]] = None,
         ext_modules: Optional[list[ModuleType]] = None,
-        task: model_types.Task = model_types.Task.UNKNOWN,
+        task: model_types.Task = task.Task.UNKNOWN,
         options: Optional[model_types.ModelSaveOption] = None,
     ) -> ModelVersion:
         """
@@ -303,12 +355,12 @@ class Registry:
                 to specify a dependency. It is a recommended way to specify your dependencies using conda. When channel
                 is not specified, Snowflake Anaconda Channel will be used. Defaults to None.
             pip_requirements: List of Pip package specifications. Defaults to None.
-                Models with pip requirements are currently only runnable in Snowpark Container Services.
-                See https://docs.snowflake.com/en/developer-guide/snowflake-ml/model-registry/container for more.
-                Models with pip requirements specified will not be executable in Snowflake Warehouse where all
-                dependencies must be retrieved from Snowflake Anaconda Channel.
+                Models running in a Snowflake Warehouse must also specify a pip artifact repository (see
+                `artifact_repository_map`). Otherwise, models with pip requirements are runnable only in Snowpark
+                Container Services. See
+                https://docs.snowflake.com/en/developer-guide/snowflake-ml/model-registry/container for more.
             artifact_repository_map: Specifies a mapping of package channels or platforms to custom artifact
-                repositories. Defaults to None. Currently, the mapping applies only to warehouse execution.
+                repositories. Defaults to None. Currently, the mapping applies only to Warehouse execution.
                 Note : This feature is currently in Public Preview.
                 Format: {channel_name: artifact_repository_name}, where:
                    - channel_name: Currently must be 'pip'.
@@ -316,10 +368,13 @@ class Registry:
                      `snowflake.snowpark.pypi_shared_repository`.
             resource_constraint: Mapping of resource constraint keys and values, e.g. {"architecture": "x86"}.
             target_platforms: List of target platforms to run the model. The only acceptable inputs are a combination of
-                "WAREHOUSE" and "SNOWPARK_CONTAINER_SERVICES":
-                - ["WAREHOUSE"]  (Warehouse only)
-                - ["SNOWPARK_CONTAINER_SERVICES"]  (Snowpark Container Services only)
-                - ["WAREHOUSE", "SNOWPARK_CONTAINER_SERVICES"] (Both)
+                "WAREHOUSE" and "SNOWPARK_CONTAINER_SERVICES", or a target platform constant:
+                - ["WAREHOUSE"] or snowflake.ml.model.target_platform.WAREHOUSE_ONLY (Warehouse only)
+                - ["SNOWPARK_CONTAINER_SERVICES"] or
+                  snowflake.ml.model.target_platform.SNOWPARK_CONTAINER_SERVICES_ONLY
+                  (Snowpark Container Services only)
+                - ["WAREHOUSE", "SNOWPARK_CONTAINER_SERVICES"] or
+                  snowflake.ml.model.target_platform.BOTH_WAREHOUSE_AND_SNOWPARK_CONTAINER_SERVICES (Both)
                 Defaults to None. When None, the target platforms will be both.
             python_version: Python version in which the model is run. Defaults to None.
             signatures: Model data signatures for inputs and outputs for various target methods. If it is None,
@@ -366,6 +421,7 @@ class Registry:
 
         Raises:
             ValueError: If extra arguments are specified ModelVersion is provided.
+            Exception: If the model logging fails.
 
         Returns:
             ModelVersion: ModelVersion object corresponding to the model just logged.
@@ -421,7 +477,7 @@ class Registry:
             if task is not model_types.Task.UNKNOWN:
                 raise ValueError("`task` cannot be specified when calling log_model with a ModelVersion.")
 
-        if pip_requirements and not artifact_repository_map:
+        if pip_requirements and not artifact_repository_map and self._targets_warehouse(target_platforms):
             warnings.warn(
                 "Models logged specifying `pip_requirements` cannot be executed in a Snowflake Warehouse "
                 "without specifying `artifact_repository_map`. This model can be run in Snowpark Container "
@@ -429,27 +485,39 @@ class Registry:
                 category=UserWarning,
                 stacklevel=1,
             )
-        return self._model_manager.log_model(
-            model=model,
-            model_name=model_name,
-            version_name=version_name,
-            comment=comment,
-            metrics=metrics,
-            conda_dependencies=conda_dependencies,
-            pip_requirements=pip_requirements,
-            artifact_repository_map=artifact_repository_map,
-            resource_constraint=resource_constraint,
-            target_platforms=target_platforms,
-            python_version=python_version,
-            signatures=signatures,
-            sample_input_data=sample_input_data,
-            user_files=user_files,
-            code_paths=code_paths,
-            ext_modules=ext_modules,
-            task=task,
-            options=options,
-            statement_params=statement_params,
-        )
+
+        event_handler = RegistryEventHandler()
+        with event_handler.status("Logging model to registry...") as status:
+            # Perform the actual model logging
+            try:
+                result = self._model_manager.log_model(
+                    model=model,
+                    model_name=model_name,
+                    version_name=version_name,
+                    comment=comment,
+                    metrics=metrics,
+                    conda_dependencies=conda_dependencies,
+                    pip_requirements=pip_requirements,
+                    artifact_repository_map=artifact_repository_map,
+                    resource_constraint=resource_constraint,
+                    target_platforms=target_platforms,
+                    python_version=python_version,
+                    signatures=signatures,
+                    sample_input_data=sample_input_data,
+                    user_files=user_files,
+                    code_paths=code_paths,
+                    ext_modules=ext_modules,
+                    task=task,
+                    options=options,
+                    statement_params=statement_params,
+                    event_handler=event_handler,
+                )
+                status.update(label="Model logged successfully!", state="complete", expanded=False)
+                return result
+            except Exception as e:
+                event_handler.update("❌ Model logging failed!")
+                status.update(label="Model logging failed!", state="error", expanded=False)
+                raise e
 
     @telemetry.send_api_usage_telemetry(
         project=_TELEMETRY_PROJECT,
@@ -626,3 +694,12 @@ class Registry:
         if not self.enable_monitoring:
             raise ValueError(_MODEL_MONITORING_DISABLED_ERROR)
         self._model_monitor_manager.delete_monitor(name)
+
+    @staticmethod
+    def _targets_warehouse(target_platforms: Optional[list[model_types.SupportedTargetPlatformType]]) -> bool:
+        """Returns True if warehouse is a target platform (None defaults to True)."""
+        return (
+            target_platforms is None
+            or model_types.TargetPlatform.WAREHOUSE in target_platforms
+            or "WAREHOUSE" in target_platforms
+        )

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.8.6"
+VERSION = "1.9.0"

{snowflake_ml_python-1.8.6.dist-info → snowflake_ml_python-1.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.8.6
+Version: 1.9.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: 
@@ -408,13 +408,48 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
+## 1.9.0
+
+### Bug Fixes
+
+- Registry: Fixed bug causing snowpark to pandas dataframe conversion to fail when `QUOTED_IDENTIFIERS_IGNORE_CASE`
+  parameter is enabled
+- Registry: Fixed duplicate UserWarning logs during model packaging
+
+### Behavior Changes
+
+- ML Job: The `list_jobs()` API has been modified. The `scope` parameter has been removed,
+  optional `database` and `schema` parameters have been added, the return type has changed
+  from `snowpark.DataFrame` to `pandas.DataFrame`, and the returned columns have been updated
+  to `name`, `status`, `message`, `database_name`, `schema_name`, `owner`, `compute_pool`,
+  `target_instances`, `created_time`, and `completed_time`.
+- Registry: Set `relax_version` to false when pip_requirements are specified while logging model
+- Registry: UserWarning will now be raised based on specified target_platforms (addresses spurious warnings)
+
+### New Features
+
+- Registry: `target_platforms` supports `TargetPlatformMode`: `WAREHOUSE_ONLY`, `SNOWPARK_CONTAINER_SERVICES_ONLY`,
+  or `BOTH_WAREHOUSE_AND_SNOWPARK_CONTAINER_SERVICES`.
+- Registry: Introduce `snowflake.ml.model.target_platform.TargetPlatform`, target platform constants, and
+  `snowflake.ml.model.task.Task`.
+- ML Job: Single-node ML Jobs are now in GA. Multi-node support is now in PuPr
+  - Moved less frequently used job submission parameters to `**kwargs`
+  - Platform metrics are now enabled by default
+  - `list_jobs()` behavior changed, see [Behavior Changes](#behavior-changes) for more info
+
 ## 1.8.6
 
 ### Bug Fixes
 
+- Fixed fatal errors from internal telemetry wrappers.
+
 ### New Features
 
 - Registry: Add service container info to logs.
+- ML Job (PuPr): Add new `submit_from_stage()` API for submitting a payload from an existing stage path.
+- ML Job (PuPr): Add support for `snowpark.Session` objects in the argument list of
+  `@remote` decorated functions. `Session` object will be injected from context in
+  the job execution environment.
 
 ## 1.8.5
 
@@ -425,17 +460,17 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 - Explainability: bump minimum streamlit version down to 1.30
 - Modeling: Make XGBoost a required dependency (xgboost is not a required dependency in snowflake-ml-python 1.8.4).
 
-### Breaking change
+### Behavior Changes
 
-- ML Job: Rename argument `num_instances` to `target_instances` in job submission APIs and
+- ML Job (Multi-node PrPr): Rename argument `num_instances` to `target_instances` in job submission APIs and
   change type from `Optional[int]` to `int`
 
 ### New Features
 
 - Registry: No longer checks if the snowflake-ml-python version is available in the Snowflake Conda channel when logging
   an SPCS-only model.
-- ML Job: Add `min_instances` argument to the job decorator to allow waiting for workers to be ready.
-- ML Job: Adjust polling behavior to reduce number of SQL calls.
+- ML Job (PuPr): Add `min_instances` argument to the job decorator to allow waiting for workers to be ready.
+- ML Job (PuPr): Adjust polling behavior to reduce number of SQL calls.
 
 ### Deprecations
 
@@ -450,18 +485,19 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 - Registry: Fixed a bug when logging pytroch and tensorflow models that caused
   `UnboundLocalError: local variable 'multiple_inputs' referenced before assignment`.
 
-### Breaking change
+### Behavior Changes
 
-- ML Job: Updated property `id` to be fully qualified name; Introduced new property `name` to represent the ML Job name
-- ML Job: Modified `list_jobs()` to return ML Job `name` instead of `id`
+- ML Job (PuPr) Updated property `id` to be fully qualified name; Introduced new property `name`
+  to represent the ML Job name
+- ML Job (PuPr) Modified `list_jobs()` to return ML Job `name` instead of `id`
 - Registry: Error in `log_model` if `enable_explainability` is True and model is only deployed to
    Snowpark Container Services, instead of just user warning.
 
 ### New Features
 
-- ML Job: Extend `@remote` function decorator, `submit_file()` and `submit_directory()` to accept `database` and
+- ML Job (PuPr): Extend `@remote` function decorator, `submit_file()` and `submit_directory()` to accept `database` and
   `schema` parameters
-- ML Job: Support querying by fully qualified name in `get_job()`
+- ML Job (PuPr): Support querying by fully qualified name in `get_job()`
 - Explainability: Added visualization functions to `snowflake.ml.monitoring` to plot explanations in notebooks.
 - Explainability: Support explain for categorical transforms for sklearn pipeline
 - Support categorical type for `xgboost.DMatrix` inputs.
@@ -471,7 +507,7 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 ### New Features
 
 - Registry: Default to the runtime cuda version if available when logging a GPU model in Container Runtime.
-- ML Job: Added `as_list` argument to `MLJob.get_logs()` to enable retrieving logs
+- ML Job (PuPr): Added `as_list` argument to `MLJob.get_logs()` to enable retrieving logs
   as a list of strings
 - Registry: Support `ModelVersion.run_job` to run inference with a single-node Snowpark Container Services job.
 - DataConnector: Removed PrPr decorators
@@ -482,11 +518,11 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 ### New Features
 
 - ML Job now available as a PuPr feature
-- ML Job: Add ability to retrieve results for `@remote` decorated functions using
-  new `MLJobWithResult.result()` API, which will return the unpickled result
-  or raise an exception if the job execution failed.
-- ML Job: Pre-created Snowpark Session is now available inside job payloads using
-  `snowflake.snowpark.context.get_active_session()`
+  - Add ability to retrieve results for `@remote` decorated functions using
+    new `MLJobWithResult.result()` API, which will return the unpickled result
+    or raise an exception if the job execution failed.
+  - Pre-created Snowpark Session is now available inside job payloads using
+    `snowflake.snowpark.context.get_active_session()`
 - Registry: Introducing `save_location` to `log_model` using the `options` argument.
   User's can provide the path to write the model version's files that get stored in Snowflake's stage.
 

{snowflake_ml_python-1.8.6.dist-info → snowflake_ml_python-1.9.0.dist-info}/RECORD

@@ -10,7 +10,7 @@ snowflake/cortex/_sse_client.py,sha256=sLYgqAfTOPADCnaWH2RWAJi8KbU_7gSRsTUDcDD5T
 snowflake/cortex/_summarize.py,sha256=7GH8zqfIdOiHA5w4b6EvJEKEWhaTrL4YA6iDGbn7BNM,1307
 snowflake/cortex/_translate.py,sha256=9ZGjvAnJFisbzJ_bXnt4pyug5UzhHJRXW8AhGQEersM,1652
 snowflake/cortex/_util.py,sha256=krNTpbkFLXwdFqy1bd0xi7ZmOzOHRnIfHdQCPiLZJxk,3288
-snowflake/ml/version.py,sha256=RmOLx4RFta98Fs3ULvfN2-mFLyI7DdFGWRoNnWjn_IQ,98
+snowflake/ml/version.py,sha256=UTrGrSYSFfbGixnYeTcCTHPZkoH52lAVaVnxGidBygY,98
 snowflake/ml/_internal/env.py,sha256=EY_2KVe8oR3LgKWdaeRb5rRU-NDNXJppPDsFJmMZUUY,265
 snowflake/ml/_internal/env_utils.py,sha256=tzz8BziiwJEnZwkzDEYCMO20Sb-mnXwDtSakGfgG--M,29364
 snowflake/ml/_internal/file_utils.py,sha256=7sA6loOeSfmGP4yx16P4usT9ZtRqG3ycnXu7_Tk7dOs,14206
@@ -37,9 +37,10 @@ snowflake/ml/_internal/lineage/lineage_utils.py,sha256=-_PKuznsL_w38rVj3wXgbPdm6
 snowflake/ml/_internal/utils/connection_params.py,sha256=ejtI-_vYt7tpxCZKjOBzuGyrOxh251xc-ekahQP9XZ4,8196
 snowflake/ml/_internal/utils/db_utils.py,sha256=HlxdMrgV8UpnxvfKDM-ZR5N566eWZLC-mE291ByrPEQ,1662
 snowflake/ml/_internal/utils/formatting.py,sha256=PswZ6Xas7sx3Ok1MBLoH2o7nfXOxaJqpUPg_UqXrQb8,3676
-snowflake/ml/_internal/utils/identifier.py,sha256=0Tn07XNxyUFYdFIYNvZ0iA7k9jiyOFAqrVx5ZLvoDwQ,12582
+snowflake/ml/_internal/utils/identifier.py,sha256=HrcCBOyn93fRjMj4K1YJG37ONtw7e3EZnt29LzhEgLA,12586
 snowflake/ml/_internal/utils/import_utils.py,sha256=msvUDaCcJpAcNCS-5Ynz4F1CvUhXjRsuZyOv1rN6Yhk,3213
 snowflake/ml/_internal/utils/jwt_generator.py,sha256=bj7Ltnw68WjRcxtV9t5xrTRvV5ETnvovB-o3Y8QWNBg,5357
+snowflake/ml/_internal/utils/mixins.py,sha256=ZE76Oc7EEbPtlwtm1oALozKdjQATE4n3WzhkNQeiUZg,2847
 snowflake/ml/_internal/utils/parallelize.py,sha256=l8Zjo-hp8zqoLgHxBlpz9Zmn2Z-MRQ0fS_NnrR4jWR8,4522
 snowflake/ml/_internal/utils/pkg_version_utils.py,sha256=EaY_3IsVOZ9BCH28F5VLjp-0AiEqDlL7L715vkPsgrY,5149
 snowflake/ml/_internal/utils/query_result_checker.py,sha256=1PR41Xn9BUIXvp-UmJ9FgEbj8WfgU7RUhz3PqvvVQ5E,10656
@@ -95,14 +96,15 @@ snowflake/ml/fileset/sfcfs.py,sha256=FJFc9-gc0KXaNyc10ZovN_87aUCShb0WztVwa02t0io
 snowflake/ml/fileset/snowfs.py,sha256=uF5QluYtiJ-HezGIhF55dONi3t0E6N7ByaVAIAlM3nk,5133
 snowflake/ml/fileset/stage_fs.py,sha256=V4pysouSKKDPLzuW3u_extxfvjkQa5OlwIRES9Srpzo,20151
 snowflake/ml/jobs/__init__.py,sha256=v-v9-SA1Vy-M98B31-NlqJgpI6uEg9jEEghJLub1RUY,468
-snowflake/ml/jobs/decorators.py,sha256=LlrRaAa7xJJu70QSqVjNP63-2uBg06RKfwwPi0_cbZo,3618
-snowflake/ml/jobs/job.py,sha256=47x8Sij6FeBuGYil0Jqy59BhkA0nP1ofoVl8XMKXchc,16416
-snowflake/ml/jobs/manager.py,sha256=NGVu8Ysd5MmZnykb41mqz7EPs8xEgjvdDjSMrPUHqPo,19627
-snowflake/ml/jobs/_utils/constants.py,sha256=3mKan6fj0kxoG8sHvWRiOMn5lvrfJYeHGFolVPmkMvE,3642
+snowflake/ml/jobs/decorators.py,sha256=mQgdWvvCwD7q79cSFKZHKegXGh2j1u8WM64UD3lCKr4,3428
+snowflake/ml/jobs/job.py,sha256=0DXIl8CM7Ld9_GQOo0r5cAChf48j8uiGX1ye9MdvYMw,21945
+snowflake/ml/jobs/manager.py,sha256=XjKGQPVO1i1nzyAx3KPiQY37Mstz-O5Qr-UFZaSlqqc,21384
+snowflake/ml/jobs/_utils/constants.py,sha256=tFZzFcWNA1okEjhjrfsDwa9ugb2qVdhk3csVOwk95E8,3642
 snowflake/ml/jobs/_utils/function_payload_utils.py,sha256=4LBaStMdhRxcqwRkwFje-WwiEKRWnBfkaOYouF3N3Kg,1308
-snowflake/ml/jobs/_utils/interop_utils.py,sha256=8_HzGyz1bl-We6_tBp1NKxlYZ2VqWT4svJzKTEh7Wx4,18844
-snowflake/ml/jobs/_utils/payload_utils.py,sha256=AkfWb7az0-W-478c_9fUQQim_MtpDKrjERiTAPzsE18,24313
-snowflake/ml/jobs/_utils/spec_utils.py,sha256=iubk_b_hA0T4mfE1WgqLp4CDpuQvvETeHFrlE1BJTIM,12991
+snowflake/ml/jobs/_utils/interop_utils.py,sha256=7mODMTjKCLXkJloACG6_9b2wvmRgjXF0Jx3wpWYyJeA,21413
+snowflake/ml/jobs/_utils/payload_utils.py,sha256=Fjf5mZpgsleZsPA9lFJELnb80mEB3Lkwb0sYiU2xmnM,24362
+snowflake/ml/jobs/_utils/query_helper.py,sha256=vCRA3TqSo77L0xJcDjGhojxAD6SHDPLDVh4HGqOJx1Q,386
+snowflake/ml/jobs/_utils/spec_utils.py,sha256=bwAvRiV4VUffWC1iFQO8kIv9q__11WFeXjnSPr9bB1Y,13254
 snowflake/ml/jobs/_utils/stage_utils.py,sha256=RrJPKHFBmEqOoRe3Lr9ypq0A-tuf8uqmfzxAy-yo9o4,4053
 snowflake/ml/jobs/_utils/types.py,sha256=l4onybhHhW1hhsbtXVhJ_RmzptClAPHY-fZRTIIcSLY,1087
 snowflake/ml/jobs/_utils/scripts/constants.py,sha256=PtqQp-KFUjsBBoQIs5TyphmauYJzd5R1m4L31FOWBr0,912
@@ -115,14 +117,16 @@ snowflake/ml/lineage/lineage_node.py,sha256=jCxCwQRvUkH-5nyF1PvdKAyRombOjWDYs5ZJ
 snowflake/ml/model/__init__.py,sha256=EvPtblqPN6_T6dyVfaYUxCfo_M7D2CQ1OR5giIH4TsQ,314
 snowflake/ml/model/custom_model.py,sha256=fDhMObqlyzD_qQG1Bq6HHkBN1w3Qzg9e81JWPiqRfc4,12249
 snowflake/ml/model/model_signature.py,sha256=bVRdMx4JEj31gLe2dr10y7aVy9fPDfPlcKYlE1NBOeQ,32265
-snowflake/ml/model/type_hints.py,sha256=oCyzLllloC_GZVddHSBQMg_fvWQfhLLXwJPxPKpwvtE,9574
+snowflake/ml/model/target_platform.py,sha256=H5d-wtuKQyVlq9x33vPtYZAlR5ka0ytcKRYgwlKl0bQ,390
+snowflake/ml/model/task.py,sha256=Zp5JaLB-YfX5p_HSaw81P3J7UnycQq5EMa87A35VOaQ,286
+snowflake/ml/model/type_hints.py,sha256=F0EW6lbSpZyv5prXc7HBZkPga6LeeCdBpV59CfLdUI4,9309
 snowflake/ml/model/_client/model/model_impl.py,sha256=Yabrbir5vPMOnsVmQJ23YN7vqhi756Jcm6pfO8Aq92o,17469
 snowflake/ml/model/_client/model/model_version_impl.py,sha256=TGBSIr4JrdxSfFZyd9G0jW4CKW0aP-ReY4ZNb05CJyY,47033
 snowflake/ml/model/_client/ops/metadata_ops.py,sha256=qpK6PL3OyfuhyOmpvLCpHLy6vCxbZbp1HlEvakFGwv4,4884
-snowflake/ml/model/_client/ops/model_ops.py,sha256=-nhyXCt2wBwgKTO5yDaKArA5vwtb1n7SbXAjS4k4mbA,48121
-snowflake/ml/model/_client/ops/service_ops.py,sha256=GMFT_ArQmrMT49D717D3siBgnDyrge7hogncp99JuqM,29301
+snowflake/ml/model/_client/ops/model_ops.py,sha256=z3T71w9ZNIU5eEA5G59Ous59WzEBs3YBcPO1_zeMI8M,48586
+snowflake/ml/model/_client/ops/service_ops.py,sha256=LqwuIYwlT0RDinM9dUzPBF3v5ZoZZ5KcaJymEDboObQ,29401
 snowflake/ml/model/_client/service/model_deployment_spec.py,sha256=V1j1WJ-gYrXN9SBsbg-908MbsJejl86rmaXHg4-tZiw,17508
-snowflake/ml/model/_client/service/model_deployment_spec_schema.py,sha256=cr1yNVlbLzpHIDeyIIHb6m06-w3LfJc12DLQAqEHQqQ,1895
+snowflake/ml/model/_client/service/model_deployment_spec_schema.py,sha256=sQ13qPYSe3Po-gZDT0ZCOa3-n6zdK1ggln5AcrCY1tw,1948
 snowflake/ml/model/_client/sql/_base.py,sha256=Qrm8M92g3MHb-QnSLUlbd8iVKCRxLhG_zr5M2qmXwJ8,1473
 snowflake/ml/model/_client/sql/model.py,sha256=nstZ8zR7MkXVEfhqLt7PWMik6dZr06nzq7VsF5NVNow,5840
 snowflake/ml/model/_client/sql/model_version.py,sha256=QwzFlDH5laTqK2qF7SJQSbt28DgspWj3R11l-yD1Da0,23496
@@ -130,7 +134,7 @@ snowflake/ml/model/_client/sql/service.py,sha256=i8BDpFs7AJQ8D8UXcE4rW_iNJbXGUYt
 snowflake/ml/model/_client/sql/stage.py,sha256=2gxYNtmEXricwxeACVUr63OUDCy_iQvCi-kRT4qQtBA,887
 snowflake/ml/model/_client/sql/tag.py,sha256=9sI0VoldKmsfToWSjMQddozPPGCxYUI6n0gPBiqd6x8,4333
 snowflake/ml/model/_model_composer/model_composer.py,sha256=SJyaw8Pcp-n_VYLEraIxrispRYMkIU90DuEisZj4z-U,11631
-snowflake/ml/model/_model_composer/model_manifest/model_manifest.py,sha256=0z0TKJ-qI1cGJ9gQOfmxAoWzo0-tBmMkl80bO-P0TKg,9157
+snowflake/ml/model/_model_composer/model_manifest/model_manifest.py,sha256=Ux7YBWhnv_jsyzV0Z2bAab7kWRjt1Dz_fa2Nh1Vwh-8,10628
 snowflake/ml/model/_model_composer/model_manifest/model_manifest_schema.py,sha256=eqv-4-tvA9Lgrp7kQAQGS_CJVR9D6uOd8-SxADNOkeM,2887
 snowflake/ml/model/_model_composer/model_method/constants.py,sha256=hoJwIopSdZiYn0fGq15_NiirC0l02d5LEs2D-4J_tPk,35
 snowflake/ml/model/_model_composer/model_method/function_generator.py,sha256=nnUJki3bJVCTF3gZ-usZW3xQ6wwlJ08EfNsPAgsnI3s,2625
@@ -141,7 +145,7 @@ snowflake/ml/model/_model_composer/model_method/model_method.py,sha256=NhTAkjRlf
 snowflake/ml/model/_model_composer/model_user_file/model_user_file.py,sha256=dYNgg8P9p6nRH47-OLxZIbt_Ja3t1VPGNQ0qJtpGuAw,1018
 snowflake/ml/model/_packager/model_handler.py,sha256=qZB5FVRWZD5wDdm6vuuoXnDFar7i2nHarbe8iZRCLPo,2630
 snowflake/ml/model/_packager/model_packager.py,sha256=FBuepy_W8ZTd4gsQHLnCj-EhO0H2wvjL556YRKOKsO8,6061
-snowflake/ml/model/_packager/model_env/model_env.py,sha256=D9NBAPSVxPiDl82Dw07OPSLlwtAJqs4fUxm3VSDbYCs,18924
+snowflake/ml/model/_packager/model_env/model_env.py,sha256=tWZVz0KOt5CixAk5P317XzdejNPbN3EG_oWlIg-9EC0,19571
 snowflake/ml/model/_packager/model_handlers/_base.py,sha256=OZhGv7nyej3PqaoBz021uGa40T06d9rv-kDcKUY3VnM,7152
 snowflake/ml/model/_packager/model_handlers/_utils.py,sha256=8y-LfiBfoj2txQD4Yh_GM0eEEOrm1S0R1149J5z31O0,12572
 snowflake/ml/model/_packager/model_handlers/catboost.py,sha256=dbI2QizGZS04l6ehgXb3oy5YSXrlwRHz8YENVefEbms,10676
@@ -150,7 +154,7 @@ snowflake/ml/model/_packager/model_handlers/huggingface_pipeline.py,sha256=dBxSq
 snowflake/ml/model/_packager/model_handlers/keras.py,sha256=JKBCiJEjc41zaoEhsen7rnlyPo2RBuEqG9Vq6JR_Cq0,8696
 snowflake/ml/model/_packager/model_handlers/lightgbm.py,sha256=DAFMiqpXEUmKqeq5rgn5j6rtuwScNnuiMUBwS4OyC7Q,11074
 snowflake/ml/model/_packager/model_handlers/mlflow.py,sha256=xSpoXO0UOfBUpzx2W1O8P2WF0Xi1vrZ_J-DdgzQG0o8,9177
-snowflake/ml/model/_packager/model_handlers/pytorch.py,sha256=jHYRjPUlCpSU2yvrJwuKAYLbG6CethxQx4brQ5ZmiVM,9784
+snowflake/ml/model/_packager/model_handlers/pytorch.py,sha256=mF-pzH1kqL7egpYA3kP1NVwOLNPYdOViEkywdzRXYJc,9867
 snowflake/ml/model/_packager/model_handlers/sentence_transformers.py,sha256=sKp-bt-fAnruDMZJ5cN6F_m9dJRY0G2FjJ4-KjNLgcg,11380
 snowflake/ml/model/_packager/model_handlers/sklearn.py,sha256=dH6S7FhJBqVOWPPXyEhN9Kj9tzDdDrD0phaGacoXQ14,18094
 snowflake/ml/model/_packager/model_handlers/snowmlmodel.py,sha256=uvz-hosuNbtcQFprnS8GzjnM8fWULBDMRbXq8immW9Q,18352
@@ -163,7 +167,7 @@ snowflake/ml/model/_packager/model_handlers_migrator/tensorflow_migrator_2023_12
 snowflake/ml/model/_packager/model_handlers_migrator/tensorflow_migrator_2025_01_01.py,sha256=0DxwZtXFgXpxb5LQEAfTUfEFV7zgbG4j3F-oNHLkTgE,769
 snowflake/ml/model/_packager/model_handlers_migrator/torchscript_migrator_2023_12_01.py,sha256=MDOAGV6kML9sJh_hnYjnrPH4GtECP5DDCjaRT7NmYpU,768
 snowflake/ml/model/_packager/model_meta/model_blob_meta.py,sha256=CzY_MhiSshKi9dWzXc4lrC9PysU0FCdHG2oRlz1vCb8,1943
-snowflake/ml/model/_packager/model_meta/model_meta.py,sha256=hX-gZzNxPyWDPBfGPWjhvnEWaPdKyNZcUcywONMndHg,19912
+snowflake/ml/model/_packager/model_meta/model_meta.py,sha256=CctjNVwdC7ghVIPqbhb62t43SOFsmk2j2FdoZMZ8KXs,20063
 snowflake/ml/model/_packager/model_meta/model_meta_schema.py,sha256=e4TUbWl998xQOZUzEWvb9CrUyHwGHBGb0TNbtezAeQ0,3707
 snowflake/ml/model/_packager/model_meta_migrator/base_migrator.py,sha256=8zTgq3n6TBXv7Vcwmf7b9wjK3m-9HHMsY0Qy1Rs-sZ4,1305
 snowflake/ml/model/_packager/model_meta_migrator/migrator_plans.py,sha256=5butM-lyaDRhCAO2BaCOIQufpAxAfSAinsNuGqbbjMU,1029
@@ -178,7 +182,7 @@ snowflake/ml/model/_signatures/dmatrix_handler.py,sha256=ldcWqadJ9fJp9cOaZ3Mn-hT
 snowflake/ml/model/_signatures/numpy_handler.py,sha256=xy7mBEAs9U5eM8F51NLabLbWXRmyQUffhVweO6jmLBA,5461
 snowflake/ml/model/_signatures/pandas_handler.py,sha256=rYgSaqdh8d-w22e_ZDt4kCFCkPWEhs-KwL9wyoLUacI,10704
 snowflake/ml/model/_signatures/pytorch_handler.py,sha256=Xy-ITCCX_EgHcyIIqeYSDUIvE2kiqECa8swy1hmohyc,5036
-snowflake/ml/model/_signatures/snowpark_handler.py,sha256=0SYtWnmJ_Nji52GZG1CDvfIGiLydTkPwpc0YUMs9aPQ,5396
+snowflake/ml/model/_signatures/snowpark_handler.py,sha256=YOBC_Wx-H8bQ967A47nYgqcqLjEA15FbZK69TyAEgvU,7590
 snowflake/ml/model/_signatures/tensorflow_handler.py,sha256=_yrvMg-w_jJoYuyrGXKPX4Dv7Vt8z1e6xIKiWGuZcc4,5660
 snowflake/ml/model/_signatures/utils.py,sha256=WLaHpb-4BPB7IBFg2sOkH2N7AojXt2PQR7M8hmtNkXA,17164
 snowflake/ml/model/models/huggingface_pipeline.py,sha256=7tYyhcqLATtzidWBAnip0qSsUstqtLBaiCUO78qgMvY,10311
@@ -329,7 +333,7 @@ snowflake/ml/modeling/metrics/__init__.py,sha256=1lc1DCVNeo7D-gvvCjmpI5tFIIrOsEd
 snowflake/ml/modeling/metrics/classification.py,sha256=E-Dx3xSmZQrF_MXf2BHAjrDstbCXVyU5g6x6CeizosQ,66411
 snowflake/ml/modeling/metrics/correlation.py,sha256=Roi17Sx5F81VlJaLQTeBAe5qZ7sZYc31UkIuC6z4qkQ,4803
 snowflake/ml/modeling/metrics/covariance.py,sha256=HxJK1mwyt6lMSg8yonHFQ8IxAEa62MHeb1M3eHEtqlk,4672
-snowflake/ml/modeling/metrics/metrics_utils.py,sha256=MLqTN59F3NLkldhsUrJFAZsHmfo5CArmLeLGnui1RpI,13189
+snowflake/ml/modeling/metrics/metrics_utils.py,sha256=XuAjYfL437LCeBY8RMElunk8jgVzemAgln573JzS3Qk,13315
 snowflake/ml/modeling/metrics/ranking.py,sha256=znjIIRkGqnGzid7BAGhBowGHbau7mTV5gc-RY_HVfoQ,17760
 snowflake/ml/modeling/metrics/regression.py,sha256=TcqnADqfL9_1XY47HQeul09t3DMPBkPSVtHP5Z9SyV4,26043
 snowflake/ml/modeling/mixture/__init__.py,sha256=rY5qSOkHj59bHiTV6LhBiEhUA0StoCb0ACNR2vkV4v0,297
@@ -408,15 +412,15 @@ snowflake/ml/monitoring/_client/queries/rmse.ssql,sha256=OEJiSStRz9-qKoZaFvmubtY
 snowflake/ml/monitoring/_manager/model_monitor_manager.py,sha256=0jpT1-aRU2tsxSM87I-C2kfJeLevCgM-a-OwU_-VUdI,10302
 snowflake/ml/monitoring/entities/model_monitor_config.py,sha256=1W6TFTPicC6YAbjD7A0w8WMhWireyUxyuEy0RQXmqyY,1787
 snowflake/ml/registry/__init__.py,sha256=XdPQK9ejYkSJVrSQ7HD3jKQO0hKq2mC4bPCB6qrtH3U,76
-snowflake/ml/registry/registry.py,sha256=JJ3mONTPxbslphvSExJzT7uqnQUBmWIbvzmTulITTpg,31519
-snowflake/ml/registry/_manager/model_manager.py,sha256=5HMLGSEJK8uYD4OVlxJqa83g9OPvdj-K7j_UaW-dde8,18271
+snowflake/ml/registry/registry.py,sha256=D87X05zQG4HRkWX3SF0Zf_2Pn2PwDixx-9i4btuVkBU,34932
+snowflake/ml/registry/_manager/model_manager.py,sha256=rqtpWL6epksv7k1syOoQUTFKApmSuzArGfsEbEx6tfE,18811
 snowflake/ml/utils/authentication.py,sha256=E1at4TIAQRDZDsMXSbrKvSJaT6_kSYJBkkr37vU9P2s,2606
 snowflake/ml/utils/connection_params.py,sha256=JuadbzKlgDZLZ5vJ9cnyAiSitvZT9jGSfSSNjIY9P1Q,8282
 snowflake/ml/utils/html_utils.py,sha256=L4pzpvFd20SIk4rie2kTAtcQjbxBHfjKmxonMAT2OoA,7665
 snowflake/ml/utils/sparse.py,sha256=zLBNh-ynhGpKH5TFtopk0YLkHGvv0yq1q-sV59YQKgg,3819
 snowflake/ml/utils/sql_client.py,sha256=pSe2od6Pkh-8NwG3D-xqN76_uNf-ohOtVbT55HeQg1Y,668
-snowflake_ml_python-1.8.6.dist-info/licenses/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
-snowflake_ml_python-1.8.6.dist-info/METADATA,sha256=nQja0TiCef4FgucRD7wpZqgo5NIOngq8rKgPIRknhHg,85172
-snowflake_ml_python-1.8.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-snowflake_ml_python-1.8.6.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
-snowflake_ml_python-1.8.6.dist-info/RECORD,,
+snowflake_ml_python-1.9.0.dist-info/licenses/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
+snowflake_ml_python-1.9.0.dist-info/METADATA,sha256=H98YP36JCenEz7IdiYeStZWdsP0ourCqqBFD2RZ1_sg,87105
+snowflake_ml_python-1.9.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+snowflake_ml_python-1.9.0.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
+snowflake_ml_python-1.9.0.dist-info/RECORD,,