snowflake-ml-python 1.8.0__py3-none-any.whl → 1.8.2__py3-none-any.whl

snowflake/ml/model/_packager/model_packager.py

@@ -44,6 +44,7 @@ class ModelPackager:
         conda_dependencies: Optional[List[str]] = None,
         pip_requirements: Optional[List[str]] = None,
         artifact_repository_map: Optional[Dict[str, str]] = None,
+        target_platforms: Optional[List[model_types.TargetPlatform]] = None,
         python_version: Optional[str] = None,
         ext_modules: Optional[List[ModuleType]] = None,
         code_paths: Optional[List[str]] = None,
@@ -77,6 +78,7 @@ class ModelPackager:
             artifact_repository_map=artifact_repository_map,
             python_version=python_version,
             task=task,
+            target_platforms=target_platforms,
             **options,
         ) as meta:
             model_blobs_path = os.path.join(self.local_dir_path, ModelPackager.MODEL_BLOBS_DIR)

snowflake/ml/model/_packager/model_runtime/_snowml_inference_alternative_requirements.py

@@ -1 +1 @@
-REQUIREMENTS = ['absl-py>=0.15,<2', 'aiohttp!=4.0.0a0, !=4.0.0a1', 'anyio>=3.5.0,<5', 'cachetools>=3.1.1,<6', 'cloudpickle>=2.0.0', 'cryptography', 'fsspec>=2024.6.1,<2026', 'importlib_resources>=6.1.1, <7', 'numpy>=1.23,<2', 'packaging>=20.9,<25', 'pandas>=1.0.0,<3', 'pyarrow', 'pyjwt>=2.0.0, <3', 'pytimeparse>=1.1.8,<2', 'pyyaml>=6.0,<7', 'requests', 'retrying>=1.3.3,<2', 's3fs>=2024.6.1,<2026', 'scikit-learn>=1.4,<1.6', 'scipy>=1.9,<2', 'snowflake-connector-python>=3.12.0,<4', 'snowflake-snowpark-python>=1.17.0,<2,!=1.26.0', 'sqlparse>=0.4,<1', 'typing-extensions>=4.1.0,<5', 'xgboost>=1.7.3,<3']
+REQUIREMENTS = ['absl-py>=0.15,<2', 'aiohttp!=4.0.0a0, !=4.0.0a1', 'anyio>=3.5.0,<5', 'cachetools>=3.1.1,<6', 'cloudpickle>=2.0.0,<3', 'cryptography', 'fsspec>=2024.6.1,<2026', 'importlib_resources>=6.1.1, <7', 'numpy>=1.23,<2', 'packaging>=20.9,<25', 'pandas>=1.0.0,<3', 'pyarrow', 'pyjwt>=2.0.0, <3', 'pytimeparse>=1.1.8,<2', 'pyyaml>=6.0,<7', 'requests', 'retrying>=1.3.3,<2', 's3fs>=2024.6.1,<2026', 'scikit-learn>=1.4,<1.6', 'scipy>=1.9,<2', 'snowflake-connector-python>=3.12.0,<4', 'snowflake-snowpark-python>=1.17.0,<2,!=1.26.0', 'snowflake.core>=1.0.2,<2', 'sqlparse>=0.4,<1', 'typing-extensions>=4.1.0,<5', 'xgboost>=1.7.3,<3']

snowflake/ml/model/type_hints.py

@@ -147,6 +147,7 @@ class BaseModelSaveOption(TypedDict):
     embed_local_ml_library: Embedding local SnowML into the code directory of the folder.
     relax_version: Whether or not relax the version constraints of the dependencies if unresolvable in Warehouse.
         It detects any ==x.y.z in specifiers and replaced with >=x.y, <(x+1). Defaults to True.
+    save_location: Local directory path to save the model and metadata.
     """
 
     embed_local_ml_library: NotRequired[bool]
@@ -154,6 +155,7 @@ class BaseModelSaveOption(TypedDict):
     function_type: NotRequired[Literal["FUNCTION", "TABLE_FUNCTION"]]
     method_options: NotRequired[Dict[str, ModelMethodSaveOptions]]
     enable_explainability: NotRequired[bool]
+    save_location: NotRequired[str]
 
 
 class CatBoostModelSaveOptions(BaseModelSaveOption):

snowflake/ml/modeling/_internal/estimator_utils.py

@@ -130,7 +130,11 @@ def is_single_node(session: Session) -> bool:
     warehouse_name = session.get_current_warehouse()
     if warehouse_name:
         warehouse_name = warehouse_name.replace('"', "")
-        df = session.sql(f"SHOW WAREHOUSES like '{warehouse_name}';")['"type"', '"size"'].collect()[0]
+        df_list = session.sql(f"SHOW WAREHOUSES like '{warehouse_name}';")['"type"', '"size"'].collect()
+        # If no warehouse data is found, default to True (single node)
+        if not df_list:
+            return True
+        df = df_list[0]
         # filter out the conditions when it is single node
         single_node: bool = (df[0] == "SNOWPARK-OPTIMIZED" and df[1] == "Medium") or (
             df[0] == "STANDARD" and df[1] == "X-Small"

snowflake/ml/registry/_manager/model_manager.py

@@ -1,3 +1,4 @@
+import os
 from types import ModuleType
 from typing import Any, Dict, List, Optional, Tuple, Union
 
@@ -13,6 +14,7 @@ from snowflake.ml.model._client.model import model_impl, model_version_impl
 from snowflake.ml.model._client.ops import metadata_ops, model_ops, service_ops
 from snowflake.ml.model._model_composer import model_composer
 from snowflake.ml.model._packager.model_meta import model_meta
+from snowflake.ml.modeling._internal import constants
 from snowflake.snowpark import exceptions as snowpark_exceptions, session
 
 logger = logging.getLogger(__name__)
@@ -208,6 +210,14 @@ class ModelManager:
         if target_platforms:
             # Convert any string target platforms to TargetPlatform objects
             platforms = [model_types.TargetPlatform(platform) for platform in target_platforms]
+        else:
+            # Default the target platform to SPCS if not specified when running in ML runtime
+            if os.getenv(constants.IN_ML_RUNTIME_ENV_VAR):
+                logger.info(
+                    "Logging the model on Container Runtime for ML without specifying `target_platforms`. "
+                    'Default to `target_platforms=["SNOWPARK_CONTAINER_SERVICES"]`.'
+                )
+                platforms = [model_types.TargetPlatform.SNOWPARK_CONTAINER_SERVICES]
 
         if artifact_repository_map:
             for channel, artifact_repository_name in artifact_repository_map.items():
@@ -223,8 +233,17 @@ class ModelManager:
 
         logger.info("Start packaging and uploading your model. It might take some time based on the size of the model.")
 
+        # Extract save_location from options if present
+        save_location = None
+        if options and "save_location" in options:
+            save_location = options.get("save_location")
+            logger.info(f"Model will be saved to local directory: {save_location}")
+
         mc = model_composer.ModelComposer(
-            self._model_ops._session, stage_path=stage_path, statement_params=statement_params
+            self._model_ops._session,
+            stage_path=stage_path,
+            statement_params=statement_params,
+            save_location=save_location,
         )
         model_metadata: model_meta.ModelMetadata = mc.save(
             name=model_name_id.resolved(),

snowflake/ml/registry/registry.py

@@ -75,7 +75,9 @@ class Registry:
             )
 
         self._model_manager = model_manager.ModelManager(
-            session, database_name=self._database_name, schema_name=self._schema_name
+            session,
+            database_name=self._database_name,
+            schema_name=self._schema_name,
         )
 
         self.enable_monitoring = options.get("enable_monitoring", True) if options else True
@@ -113,8 +115,10 @@ class Registry:
         python_version: Optional[str] = None,
         signatures: Optional[Dict[str, model_signature.ModelSignature]] = None,
         sample_input_data: Optional[model_types.SupportedDataType] = None,
+        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,
         options: Optional[model_types.ModelSaveOption] = None,
     ) -> ModelVersion:
         """
@@ -156,10 +160,15 @@ class Registry:
                 infer the signature. If not None, sample_input_data should not be specified. Defaults to None.
             sample_input_data: Sample input data to infer model signatures from.
                 It would also be used as background data in explanation and to capture data lineage. Defaults to None.
+            user_files: Dictionary where the keys are subdirectories, and values are lists of local file name
+                strings. The local file name strings can include wildcards (? or *) for matching multiple files.
             code_paths: List of directories containing code to import. Defaults to None.
             ext_modules: List of external modules to pickle with the model object.
                 Only supported when logging the following types of model:
                 Scikit-learn, Snowpark ML, PyTorch, TorchScript and Custom Model. Defaults to None.
+            task: The task of the Model Version. It is an enum class Task with values TABULAR_REGRESSION,
+                TABULAR_BINARY_CLASSIFICATION, TABULAR_MULTI_CLASSIFICATION, TABULAR_RANKING, or UNKNOWN. By default,
+                it is set to Task.UNKNOWN and may be overridden by inferring from the Model Object.
             options (Dict[str, Any], optional): Additional model saving options.
 
                 Model Saving Options include:
@@ -171,6 +180,10 @@ 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.
+                - 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.
+                - save_location: Location to save the model and metadata.
                 - method_options: Per-method saving options. This dictionary has method names as keys and dictionary
                     values with the desired options.
 
@@ -304,6 +317,10 @@ 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.
+                - 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.
+                - save_location: Location to save the model and metadata.
                 - method_options: Per-method saving options. This dictionary has method names as keys and dictionary
                     values with the desired options. See the example below.
 
@@ -317,6 +334,9 @@ class Registry:
                         Defaults to None, determined automatically by Snowflake.
                     - function_type: One of supported model method function types (FUNCTION or TABLE_FUNCTION).
 
+        Raises:
+            ValueError: If extra arguments are specified ModelVersion is provided.
+
         Returns:
             ModelVersion: ModelVersion object corresponding to the model just logged.
 
@@ -339,13 +359,37 @@ class Registry:
             registry.log_model(
               model=model,
               model_name="my_model",
-              method_options=method_options,
+              options={"method_options": method_options},
             )
         """
         statement_params = telemetry.get_statement_params(
             project=_TELEMETRY_PROJECT,
             subproject=_MODEL_TELEMETRY_SUBPROJECT,
         )
+        if isinstance(model, ModelVersion):
+            # check that no arguments are provided other than the ones for copy model.
+            invalid_args = [
+                comment,
+                conda_dependencies,
+                pip_requirements,
+                artifact_repository_map,
+                target_platforms,
+                python_version,
+                signatures,
+                sample_input_data,
+                user_files,
+                code_paths,
+                ext_modules,
+                options,
+            ]
+            for arg in invalid_args:
+                if arg is not None:
+                    raise ValueError(
+                        "When calling log_model with a ModelVersion, only model_name and version_name may be specified."
+                    )
+            if task is not model_types.Task.UNKNOWN:
+                raise ValueError("`task` cannot be specified when calling log_model with a ModelVersion.")
+
         if pip_requirements:
             warnings.warn(
                 "Models logged specifying `pip_requirements` can not be executed "

snowflake/ml/version.py

@@ -1 +1 @@
-VERSION="1.8.0"
+VERSION="1.8.2"

{snowflake_ml_python-1.8.0.dist-info → snowflake_ml_python-1.8.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.8.0
+Version: 1.8.2
 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: 
@@ -236,7 +236,7 @@ License-File: LICENSE.txt
 Requires-Dist: absl-py<2,>=0.15
 Requires-Dist: anyio<5,>=3.5.0
 Requires-Dist: cachetools<6,>=3.1.1
-Requires-Dist: cloudpickle>=2.0.0
+Requires-Dist: cloudpickle<3,>=2.0.0
 Requires-Dist: cryptography
 Requires-Dist: fsspec[http]<2026,>=2024.6.1
 Requires-Dist: importlib_resources<7,>=6.1.1
@@ -253,6 +253,7 @@ Requires-Dist: scikit-learn<1.6,>=1.4
 Requires-Dist: scipy<2,>=1.9
 Requires-Dist: snowflake-connector-python[pandas]<4,>=3.12.0
 Requires-Dist: snowflake-snowpark-python!=1.26.0,<2,>=1.17.0
+Requires-Dist: snowflake.core<2,>=1.0.2
 Requires-Dist: sqlparse<1,>=0.4
 Requires-Dist: typing-extensions<5,>=4.1.0
 Requires-Dist: xgboost<3,>=1.7.3
@@ -402,7 +403,58 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.8.0
+## 1.8.2
+
+### Bug Fixes
+
+### Behavior Change
+
+### 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()`
+- 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.
+- Registry: Include model dependencies in pip requirements by default when logging in Container Runtime.
+
+```python
+reg.log_model(
+    model=...,
+    model_name=...,
+    version_name=...,
+    ...,
+    options={"save_location": "./model_directory"},
+)
+```
+
+- ML Job (PrPr): Add `instance_id` argument to `get_logs` and `show_logs` method to support multi node log retrieval
+- ML Job (PrPr): Add `job.get_instance_status(instance_id=...)` API to support multi node status retrieval
+
+## 1.8.1 (03-26-2025)
+
+### Bug Fixes
+
+- Registry: Fix a bug that caused `unsupported model type` error while logging a sklearn model with `score_samples`
+  inference method.
+- Registry: Fix a bug that model inference service creation fails on an existing and suspended service.
+
+### Behavior Change
+
+### New Features
+
+- ML Job (PrPr): Update Container Runtime image version to `1.0.1`
+- ML Job (PrPr): Add `enable_metrics` argument to job submission APIs to enable publishing service metrics to Event Table.
+  See [Accessing Event Table service metrics](https://docs.snowflake.com/en/developer-guide/snowpark-container-services/monitoring-services#accessing-event-table-service-metrics)
+  for retrieving published metrics
+  and [Costs of telemetry data collection](https://docs.snowflake.com/en/developer-guide/logging-tracing/logging-tracing-billing)
+  for cost implications.
+- Registry: When creating a copy of a `ModelVersion` with `log_model`, raise an exception if unsupported arguments are provided.
+
+## 1.8.0 (03-20-2025)
 
 ### Bug Fixes
 
@@ -703,7 +755,6 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 - Registry: Added support for handling Hugging Face model configurations with auto-mapping functionality.
 - Registry: Added support for `keras` 3.x model with `tensorflow` and `pytorch` backend
-- ML Job (PrPr): Support any serializable (pickleable) argument for `@remote` decorated functions
 
 ## 1.7.4 (01-28-2025)
 

{snowflake_ml_python-1.8.0.dist-info → snowflake_ml_python-1.8.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 snowflake/cortex/__init__.py,sha256=gboUvJBYzJIq11AK_Qa0ipOUbKctHahNXe1p1Z7j8xY,1032
 snowflake/cortex/_classify_text.py,sha256=zlAUJXPgcEwmh9KtkqafAD0NRP3UBboCJGVzMylW4Lk,1640
-snowflake/cortex/_complete.py,sha256=H_g1WWxTX9HlG-97UHip9Yg1Na4WITHJ_DoVEXlVKgc,18546
+snowflake/cortex/_complete.py,sha256=FBdO3ylXcp3TBvUj2dRnL339za4qFBWuFwCkFVZ38FM,19750
 snowflake/cortex/_embed_text_1024.py,sha256=P3oMsrMol_rTK0lZxDmEBx7XSA9HB2y3AMjaD66LrgY,1651
 snowflake/cortex/_embed_text_768.py,sha256=_WMt_FHu3n-CKXj8yiposDSi763Sd-NHV9UeGQ7YHaE,1643
 snowflake/cortex/_extract_answer.py,sha256=7C-23JytRKdZN9ZYY9w10RfAe_GzmvzKAqqUDl3T4aQ,1605
@@ -10,13 +10,13 @@ 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=cwRGgrcUo3E05ZaIDT9436vXLQ7GfuBVAjR0QeQ2bDE,3320
-snowflake/ml/version.py,sha256=9LJrO29muGIhXmJeaLCfizOkgrtiucnrXh8-DuB37oY,16
+snowflake/ml/version.py,sha256=2JIOu22NGQEnRBBjazfkoim0b-FI0TFKadK5Q0ff7EQ,16
 snowflake/ml/_internal/env.py,sha256=kCrJTRnqQ97VGUVI1cWUPD8HuBWeL5vOOtwUR0NB9Mg,161
 snowflake/ml/_internal/env_utils.py,sha256=5ps0v7c655lXsFVfnASxIyEwiVSGxZXke-VjeAWDs0A,27866
 snowflake/ml/_internal/file_utils.py,sha256=R3GRaKzJPLOa1yq9E55rhxUr59DztZlIqAnaqcZ1HfE,14275
 snowflake/ml/_internal/init_utils.py,sha256=U-oPOtyVf22hCwDH_CH2uDr9yuN6Mr3kwQ_yRAs1mcM,2696
 snowflake/ml/_internal/migrator_utils.py,sha256=k3erO8x3YJcX6nkKeyJAUNGg1qjE3RFmD-W6dtLzIH0,161
-snowflake/ml/_internal/platform_capabilities.py,sha256=GTifopHZmJ0TiYaWXX2gGYk7CGAqsIdtpbntSNPUgfA,3717
+snowflake/ml/_internal/platform_capabilities.py,sha256=X6c27VCIhZxxiSVBMb2bqR_zUtF8sper718x5muXWcY,5224
 snowflake/ml/_internal/relax_version_strategy.py,sha256=MYEIZrx1HfKNhl9Na3GN50ipX8c0MKIj9nwxjB0IC0Y,484
 snowflake/ml/_internal/telemetry.py,sha256=D2ZgVdibSYKvPg0CZhf4lW3QQot7cgOPhex7a0CXsao,30996
 snowflake/ml/_internal/type_utils.py,sha256=x0sm7lhpDyjdA1G7KvJb06z4PEGsogWiMwFrskPTWkA,2197
@@ -50,7 +50,7 @@ snowflake/ml/_internal/utils/sql_identifier.py,sha256=A5mfeDuz4z6VuUYG3EBpDyQQQC
 snowflake/ml/_internal/utils/table_manager.py,sha256=pU7v8Cx-jGObf6RtTmfCmALfhbpJD-lL45T1gWX1nSY,4982
 snowflake/ml/_internal/utils/temp_file_utils.py,sha256=0-HTjXdxVt0kE6XcgyvRvY0btflWlmotP2bMXVpFJPA,1553
 snowflake/ml/data/__init__.py,sha256=nm5VhN98Lzxr4kb679kglQfqbDbHhd9zYsnFJiQiThg,351
-snowflake/ml/data/data_connector.py,sha256=havoMBDiWTtc65mvDEAHk2Omy4BOVaMV38_acErjkbw,9696
+snowflake/ml/data/data_connector.py,sha256=wTD6qBMd9aBJKP1tV_wVa-vubzgbQBKkia0BAjRpwic,10496
 snowflake/ml/data/data_ingestor.py,sha256=mH1_DBDS_XwxYsOoWI3EGQpBzr8j8jBgH2t2S1V1eGM,1035
 snowflake/ml/data/data_source.py,sha256=dRemXGi_HHQdn6gaNkxxGJixnQPuUYFDP8NBjmB_ZMk,518
 snowflake/ml/data/ingestor_utils.py,sha256=--nEwJHbYqYHpAzR1APgoeVF9CMgq_fDX81X29HAB4w,2727
@@ -60,7 +60,7 @@ snowflake/ml/dataset/__init__.py,sha256=nESj7YEI2u90Oxyit_hKCQMWb7N1BlEM3Ho2Fm0M
 snowflake/ml/dataset/dataset.py,sha256=eN9hStlwLlWXU3bp7RSruzIgSPEXNunpagCUDRbtz54,21041
 snowflake/ml/dataset/dataset_factory.py,sha256=Fym4ICK-B1j6Om4ENwWxEvryq3ZKoCslBSZDBenmjOo,1615
 snowflake/ml/dataset/dataset_metadata.py,sha256=tWR3fa2WG3Kj2btKMbg51l5jX68qm1rfXRswU0IDYTg,4157
-snowflake/ml/dataset/dataset_reader.py,sha256=e-IRbxbxFfNbsglmqtzhV_wYFsEflBW6-U_krbfXPpw,4371
+snowflake/ml/dataset/dataset_reader.py,sha256=uYOK8e0f_cDctPLCATkWR8dWC5zoco2GpWVEYvVxU1s,4655
 snowflake/ml/feature_store/__init__.py,sha256=VKBVkS050WNF8rcqNqwFfNXa_B9GZjcUpuibOGsUSls,423
 snowflake/ml/feature_store/access_manager.py,sha256=LcsfBKsZzfERQQ_pqZG0W-XbpVGx9jkZOI-7nbfryhg,10666
 snowflake/ml/feature_store/entity.py,sha256=A65FOGlljREUG8IRMSN84v1x2uTeVGCM4NqKXO2Ui8w,4059
@@ -94,34 +94,40 @@ snowflake/ml/fileset/sfcfs.py,sha256=uPn8v5jlC3h_FrNqb4UMRAZjRZLn0I3tzu0sfi5RHik
 snowflake/ml/fileset/snowfs.py,sha256=uF5QluYtiJ-HezGIhF55dONi3t0E6N7ByaVAIAlM3nk,5133
 snowflake/ml/fileset/stage_fs.py,sha256=IEVZ6imH77JiSOIRlRHNWalwafoACRgHFr8RAaICSP8,20170
 snowflake/ml/jobs/__init__.py,sha256=ORX_0blPSpl9u5442R-i4e8cqWYfO_vVjFFtX3as184,420
-snowflake/ml/jobs/decorators.py,sha256=wMLO2hZem_6vPav0ZJPSNbKPZdSDspkKviCTx8zSaV0,2972
-snowflake/ml/jobs/job.py,sha256=dWILWTNaumgdqs6P58xl4PdyoARU9Sk89OMfRU39SQE,4239
-snowflake/ml/jobs/manager.py,sha256=DwxuLRf-PxQOG2w3hg1ewTEZTq3RrTngbIYJhtgURpQ,11311
-snowflake/ml/jobs/_utils/constants.py,sha256=aSnZKjV3TCi79G8ylaAucXxjW7Ca4vVY_HsnkwAZJm0,3242
-snowflake/ml/jobs/_utils/payload_utils.py,sha256=RC9vJt2CKmrkxzdWD7vEw0D7xWhXAC6RnuvS-0Ly_yY,20421
-snowflake/ml/jobs/_utils/spec_utils.py,sha256=D0mxCHdV0laU7HZgNGX2ue8A2Z1gn3qsMd4lf8GDtMo,12256
-snowflake/ml/jobs/_utils/types.py,sha256=sEV-jCA-0Bb8KdKMBZGrIdqJhSWGb6QI8pYRL56dIDA,874
+snowflake/ml/jobs/decorators.py,sha256=MCdVLyz_hv_GOqT89c3NxKORS1xmSCmo5VknMcffzM0,3176
+snowflake/ml/jobs/job.py,sha256=-zgtg_hC8PjFz2KL3c8l9VejvmcZsJQMZtVpcfRDk4k,9448
+snowflake/ml/jobs/manager.py,sha256=kkJDD9EaBNppnF0-QGTUiOnkPmCtChRJ53-wZMpTAtc,13411
+snowflake/ml/jobs/_utils/constants.py,sha256=5P_hKBg2JpV_yh1SH97DHjTLa2nZftKd05EJiXELriU,3169
+snowflake/ml/jobs/_utils/interop_utils.py,sha256=cmZWpvhiqk8nnS63LseO2ZLV7U0HktGtJiFxcSquBmE,18861
+snowflake/ml/jobs/_utils/payload_utils.py,sha256=ZvvuRcsQSdmqd8_WOuWwKEBhfCqchTazbbD9yttV9rI,21631
+snowflake/ml/jobs/_utils/spec_utils.py,sha256=fH_XAs2tLny4abWoghBETxVGdQd06cDe7AQi_eWurQ8,12643
+snowflake/ml/jobs/_utils/types.py,sha256=yW9A7su6gHDRsOSgpdgdjjdElYaz6aHhN2uMtYsM5oU,978
+snowflake/ml/jobs/_utils/scripts/constants.py,sha256=YyIWZqQPYOTtgCY6SfyJjk2A98I5RQVmrOuLtET5Pqg,173
+snowflake/ml/jobs/_utils/scripts/get_instance_ip.py,sha256=bh23hn1mVebiU7CytzlMVTgfYBlpXHrwjyHLSlfEJB8,5259
+snowflake/ml/jobs/_utils/scripts/mljob_launcher.py,sha256=gKps2uicKksFSeKWz_elrzLUMNunjgjtNZ9VxargR2Y,6747
+snowflake/ml/jobs/_utils/scripts/signal_workers.py,sha256=6dr2GWGSJOLxKHCDkcbJn7unWGN4xm7cgB2kmrmSpRs,7450
+snowflake/ml/jobs/_utils/scripts/worker_shutdown_listener.py,sha256=SeJ8v5XDriwHAjIGpcQkwVP-f-lO9QIdVjVD7Fkgafs,7893
 snowflake/ml/lineage/__init__.py,sha256=8p1YGynC-qOxAZ8jZX2z84Reg5bv1NoJMoJmNJCrzI4,65
 snowflake/ml/lineage/lineage_node.py,sha256=e6L4bdYDSVgTv0BEfqgPQWNoDiTiuI7HmfJ6n-WmNLE,5812
 snowflake/ml/model/__init__.py,sha256=EvPtblqPN6_T6dyVfaYUxCfo_M7D2CQ1OR5giIH4TsQ,314
 snowflake/ml/model/custom_model.py,sha256=Dmf9oLf71BQtakSagaGG_8xfr3oXoEuW_Jzpt2RqIis,11780
 snowflake/ml/model/model_signature.py,sha256=FJnLn6XtlaYWOgwMPRFb__QY3BQCeYGHWGHEFcySbT0,32302
-snowflake/ml/model/type_hints.py,sha256=fvHjtvnt_CAsXAn3q4DDQlxCn6wqIEHypHh93Ce337w,9472
+snowflake/ml/model/type_hints.py,sha256=DbCC6JXJuiGob8hohxUxP8LikE-IppHYp2mW-CXbfmQ,9580
 snowflake/ml/model/_client/model/model_impl.py,sha256=pqjK8mSZIQJ_30tRWWFPIo8X35InSVoAunXlQNtSJEM,15369
 snowflake/ml/model/_client/model/model_version_impl.py,sha256=kslv-oOyh5OvoG8BjNBl_t4mqRjMn-kLpla6CdJRBaA,40272
 snowflake/ml/model/_client/ops/metadata_ops.py,sha256=7cGx8zYzye2_cvZnyGxoukPtT6Q-Kexd-s4yeZmpmj8,4890
-snowflake/ml/model/_client/ops/model_ops.py,sha256=hJ1D7ZR6G6T8bbL2KqL6ee27JC9S9u8Q_5D7rq0g_BE,47863
-snowflake/ml/model/_client/ops/service_ops.py,sha256=bT713pRVqS5t5YMaGxDOp3eupKTT7tW0_jo6c6FK8Ck,18800
-snowflake/ml/model/_client/service/model_deployment_spec.py,sha256=K1MkVFgATk_OHCV68QR0jNF1lkY7brDxv7yVhQhK7eY,4599
+snowflake/ml/model/_client/ops/model_ops.py,sha256=N6xuikPmuK7jpag3CluKyz4bu00oJbMC3_266T2nBLQ,47990
+snowflake/ml/model/_client/ops/service_ops.py,sha256=7i4nzXjLlwyYXvT75UNbsmNMpFmQHE4Vppw_9K77p_A,19804
+snowflake/ml/model/_client/service/model_deployment_spec.py,sha256=3CZmXXePQHpWNUdiNtdBzxJCqOEHwop8izK6O0yGKRk,4749
 snowflake/ml/model/_client/service/model_deployment_spec_schema.py,sha256=eaulF6OFNuDfQz3oPYlDjP26Ww2jWWatm81dCbg602E,825
 snowflake/ml/model/_client/sql/_base.py,sha256=Qrm8M92g3MHb-QnSLUlbd8iVKCRxLhG_zr5M2qmXwJ8,1473
 snowflake/ml/model/_client/sql/model.py,sha256=o36oPq4aU9TwahqY2uODYvICxmj1orLztijJ0yMbWnM,5852
 snowflake/ml/model/_client/sql/model_version.py,sha256=R0TnsRta7tSmd7RBphERzBKXpYBxOhec1CefW6VWrVE,23543
-snowflake/ml/model/_client/sql/service.py,sha256=2orADBhG4m4ltz98uKN-cIJOfT1C7SNUZUuSGol8QVg,11224
+snowflake/ml/model/_client/sql/service.py,sha256=EFCTukL0ng8Fcle0eusBxJYw2DaEYNshBi812LxBf2w,11757
 snowflake/ml/model/_client/sql/stage.py,sha256=165vyAtrScSQWJB8wLXKRUO1QvHTWDmPykeWOyxrDRg,826
 snowflake/ml/model/_client/sql/tag.py,sha256=pwwrcyPtSnkUfDzL3M8kqM0KSx7CaTtgty3HDhVC9vg,4345
-snowflake/ml/model/_model_composer/model_composer.py,sha256=g96CQx3sh75VlPdhKQfA6Hzr6CzyZ-s0T3XaiFfr3A4,9325
-snowflake/ml/model/_model_composer/model_manifest/model_manifest.py,sha256=IaUM1VDySf2cOE0_1lVxCodN8rd14GcugrspKBmBkPA,9023
+snowflake/ml/model/_model_composer/model_composer.py,sha256=rXQDGEjhbXIWJm0RJf9PhGLDTayRiuLn6sehUInG_g8,10226
+snowflake/ml/model/_model_composer/model_manifest/model_manifest.py,sha256=d3Ej_OzQkfSugx15m5zrheBktKVod5ZDtjrd3O5bTU8,8984
 snowflake/ml/model/_model_composer/model_manifest/model_manifest_schema.py,sha256=s71r-JGI-9aPpA7dDiic5bF2y-7m18UgHipqNhn9aS4,2836
 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
@@ -131,8 +137,8 @@ snowflake/ml/model/_model_composer/model_method/infer_table_function.py_template
 snowflake/ml/model/_model_composer/model_method/model_method.py,sha256=Yxmr3uLpxFFuYdJ5f9MQLIXP3yfbL-ym2rxl1hUIRRM,7173
 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=wMPGOegXx5GgiSA81gbKpfODosdj2mvD1bFbeN4OmNc,2642
-snowflake/ml/model/_packager/model_packager.py,sha256=6z8t_fwDMQjIq92L7kyPNtMcXKAUaVSvJVyhH3_jCWU,5834
-snowflake/ml/model/_packager/model_env/model_env.py,sha256=tdlvBqHXQvDCxMFY-p0GnryJtLHcF7qwHLMEwcQqUz8,18601
+snowflake/ml/model/_packager/model_packager.py,sha256=Gr2ENRgs9_qQevI-V5vSKDr5Lywx9hyyzuzc3VMinl4,5958
+snowflake/ml/model/_packager/model_env/model_env.py,sha256=9ifFoFyqqRaRlOwhSlnuqAjrY7qsQYNoyiioJgDO1VE,18730
 snowflake/ml/model/_packager/model_handlers/_base.py,sha256=qQS1ZSz1Ikdj0TvyLU9n8K6KAj-PknL4s801qpnWodo,7164
 snowflake/ml/model/_packager/model_handlers/_utils.py,sha256=6c7xsQqDIlI07NgQRnhAGIZqyeuwoR1T4wFYFvpEKpE,10770
 snowflake/ml/model/_packager/model_handlers/catboost.py,sha256=vwlKuXwJGYrna7wkXDgEu8-nUNegNhHMCaNQdhL5p44,10677
@@ -143,7 +149,7 @@ snowflake/ml/model/_packager/model_handlers/lightgbm.py,sha256=qm_Q7FxD19r1fFnSt
 snowflake/ml/model/_packager/model_handlers/mlflow.py,sha256=A3HnCa065jtHsRM40ZxfLv5alk0RYhVmsU4Jt2klRwQ,9189
 snowflake/ml/model/_packager/model_handlers/pytorch.py,sha256=FamqiwFhtIwlPeb6CoQD2Xkout7f5wKVugvWFX98DV0,9790
 snowflake/ml/model/_packager/model_handlers/sentence_transformers.py,sha256=EKgpN6e4c8bi0znnV-pWzAR3cwDvORcsL72x6o-JPqA,11381
-snowflake/ml/model/_packager/model_handlers/sklearn.py,sha256=LiWA58qz3Yd79L78dfPKl507CKIfTCPsi8wT2lPdbPI,15594
+snowflake/ml/model/_packager/model_handlers/sklearn.py,sha256=v2ZSQ_MWrORiKvUpdjY_SwzhpY4vxOyARJQJuQnxKdw,15443
 snowflake/ml/model/_packager/model_handlers/snowmlmodel.py,sha256=Jx6MnlfNGdPcBUcLcSs-E1yaWCB4hM3OroeGZb2kE2I,17185
 snowflake/ml/model/_packager/model_handlers/tensorflow.py,sha256=Fr_iqjJf69_az7uUSagt9qB0ipkq4f1AkjeEGw7PcI4,11205
 snowflake/ml/model/_packager/model_handlers/torchscript.py,sha256=Tmkpj_4RAlz3r8YvEQvtTTcB_q30UpeLymPGkHmQ390,9536
@@ -153,14 +159,14 @@ snowflake/ml/model/_packager/model_handlers_migrator/pytorch_migrator_2023_12_01
 snowflake/ml/model/_packager/model_handlers_migrator/tensorflow_migrator_2023_12_01.py,sha256=dXIisQteU55QMw5OvC_1E_sGqFgE88WRhGCWFqUyauM,2239
 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/_packaging_requirements.py,sha256=WxLbVm8a5Wu_SJDylEdYA6XWiiPFIXkrgboWCEWEIoc,38
+snowflake/ml/model/_packager/model_meta/_packaging_requirements.py,sha256=E8LiAHewa-lsm6_SL6d9AcpO0m23fYdsKPXOevmHjB8,41
 snowflake/ml/model/_packager/model_meta/model_blob_meta.py,sha256=GmiqqI-XVjrOX7cSa5GKerKhfHptlsg74MKqTGwJ5Jk,1949
-snowflake/ml/model/_packager/model_meta/model_meta.py,sha256=kYfCyX8Q7tlpoxrXNsryKJ_XZDrMFHZ8fBbZX3XkMhA,19332
-snowflake/ml/model/_packager/model_meta/model_meta_schema.py,sha256=uVgzGaEltd4TUoXJCEqyBU_6rzin1TJeR7j6VH35RMg,3521
+snowflake/ml/model/_packager/model_meta/model_meta.py,sha256=TPHi9qxfSUmcVQQ2x0LrtF1IYl2aIuykgSCmh0j-XpI,19650
+snowflake/ml/model/_packager/model_meta/model_meta_schema.py,sha256=xWMbdWcDBzC-ZwFgA_hPK60o91AserkR-DpXEEKYK_w,3551
 snowflake/ml/model/_packager/model_meta_migrator/base_migrator.py,sha256=SORlqpPbOeBg6dvJ3DidHeLVi0w9YF0Zv4tC0Kbc20g,1311
 snowflake/ml/model/_packager/model_meta_migrator/migrator_plans.py,sha256=nf6PWDH_gvX_OiS4A-G6BzyCLFEG4dASU0t5JTsijM4,1041
 snowflake/ml/model/_packager/model_meta_migrator/migrator_v1.py,sha256=qEPzdCw_FzExMbPuyFHupeWlYD88yejLdcmkPwjJzDk,2070
-snowflake/ml/model/_packager/model_runtime/_snowml_inference_alternative_requirements.py,sha256=MeKrqT-CMve9LDBWbOvesjnKQOAMaClYB1x-xIiuXhw,606
+snowflake/ml/model/_packager/model_runtime/_snowml_inference_alternative_requirements.py,sha256=OiFGM-wDeGxFtmm5MSeehNp2g4fgNXn0eaUAujto6Po,637
 snowflake/ml/model/_packager/model_runtime/model_runtime.py,sha256=0l8mgrfmpvTn516Id6xgIG4jIqxSy9nN2PFlnqOljiI,5365
 snowflake/ml/model/_packager/model_task/model_task_utils.py,sha256=_nm3Irl5W6Oa8_OnJyp3bLeA9QAbV9ygGCsgHI70GX4,6641
 snowflake/ml/model/_signatures/base_handler.py,sha256=4CTZKKbg4WIz_CmXjyVy8tKZW-5OFcz0J8XVPHm2dfQ,1269
@@ -175,7 +181,7 @@ snowflake/ml/model/_signatures/tensorflow_handler.py,sha256=_yrvMg-w_jJoYuyrGXKP
 snowflake/ml/model/_signatures/utils.py,sha256=gHEU2u8VCil0wvmd9C61ZNB-KiNz4QazpC7-0XQPHd8,17176
 snowflake/ml/model/models/huggingface_pipeline.py,sha256=62GpPZxBheqCnFNxNOggiDE1y9Dhst-v6D4IkGLuDeQ,10221
 snowflake/ml/modeling/_internal/constants.py,sha256=aJGngY599w3KqN8cDZCYrjbWe6UwYIbgv0gx0Ukdtc0,105
-snowflake/ml/modeling/_internal/estimator_utils.py,sha256=mbMm8_5tQde_sQDwI8pS3ljHZ8maCHl2Shb5nQwLYac,11872
+snowflake/ml/modeling/_internal/estimator_utils.py,sha256=jpiq6h6mJfPa1yZbEjrP1tEFdw-1f_XBxDRHgdH3hps,12017
 snowflake/ml/modeling/_internal/model_specifications.py,sha256=P9duVMP9-X7us_RZFPyXvWxOrm5K30sWDVYwSMEzG1M,4876
 snowflake/ml/modeling/_internal/model_trainer.py,sha256=RxpZ5ARy_3sfRMCvArkdK-KmsdbNXxEZTbXoaJ4c1ag,984
 snowflake/ml/modeling/_internal/model_trainer_builder.py,sha256=n1l9i9LFLcdbMFRvxkWNIs7kYnNNlUJnaToRvFBEjls,8062
@@ -401,14 +407,14 @@ snowflake/ml/monitoring/_client/queries/rmse.ssql,sha256=OEJiSStRz9-qKoZaFvmubtY
 snowflake/ml/monitoring/_manager/model_monitor_manager.py,sha256=_-vxqnHqohTHTrwfURjPXijyAeh1mTRdHCG436GaBik,10314
 snowflake/ml/monitoring/entities/model_monitor_config.py,sha256=IxEiee1HfBXCQGzJOZbrDrvoV8J1tDNk43ygNuN00Io,1793
 snowflake/ml/registry/__init__.py,sha256=XdPQK9ejYkSJVrSQ7HD3jKQO0hKq2mC4bPCB6qrtH3U,76
-snowflake/ml/registry/registry.py,sha256=zZUzE8gMvoegB1asxAVb6PtoB2Ycxg9YuynRwPZ6CEE,27790
-snowflake/ml/registry/_manager/model_manager.py,sha256=MjS0AnHQ8g2yn3svvhSPgmPB0j0bpw4gH-p5LjtYKds,16083
+snowflake/ml/registry/registry.py,sha256=QPiv52G6v2SuJsifcnRBotJjr8gB1PteoFXkihVgEco,30225
+snowflake/ml/registry/_manager/model_manager.py,sha256=XTvlxDdXujYsV4REwsVIkzmFFIiWSFY0UKJKwLpRKp8,16983
 snowflake/ml/utils/authentication.py,sha256=Wx1kVBZ9XBDuKkRHpPEB2pBxpiJepVLFAirDMx4m5Gk,2612
 snowflake/ml/utils/connection_params.py,sha256=JRpQppuWRk6bhdLzVDhMfz3Y6yInobFNLHmIBaXD7po,8005
 snowflake/ml/utils/sparse.py,sha256=XqDQkw39Ml6YIknswdkvFIwUwBk_GBXAbP8IACfPENg,3817
 snowflake/ml/utils/sql_client.py,sha256=z4Rhi7pQz3s9cyu_Uzfr3deCnrkCdFh9IYIvicsuwdc,692
-snowflake_ml_python-1.8.0.dist-info/licenses/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
-snowflake_ml_python-1.8.0.dist-info/METADATA,sha256=sflUdLTbz3T-qdOUeKSGo5g7NmS-LXN3CsVca7MJefI,80146
-snowflake_ml_python-1.8.0.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-snowflake_ml_python-1.8.0.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
-snowflake_ml_python-1.8.0.dist-info/RECORD,,
+snowflake_ml_python-1.8.2.dist-info/licenses/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
+snowflake_ml_python-1.8.2.dist-info/METADATA,sha256=-nSrqa3EXDuUaByFhQHhOp4BuOGY_XGfzwVX4VFaAgg,82231
+snowflake_ml_python-1.8.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+snowflake_ml_python-1.8.2.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
+snowflake_ml_python-1.8.2.dist-info/RECORD,,

{snowflake_ml_python-1.8.0.dist-info → snowflake_ml_python-1.8.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (77.0.3)
+Generator: setuptools (78.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any