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

snowflake/ml/model/type_hints.py

@@ -24,6 +24,7 @@ if TYPE_CHECKING:
     import mlflow
     import numpy as np
     import pandas as pd
+    import prophet
     import sentence_transformers
     import sklearn.base
     import sklearn.pipeline
@@ -81,6 +82,7 @@ SupportedRequireSignatureModelType = Union[
     "catboost.CatBoost",
     "lightgbm.LGBMModel",
     "lightgbm.Booster",
+    "prophet.Prophet",
     "snowflake.ml.model.custom_model.CustomModel",
     "sklearn.base.BaseEstimator",
     "sklearn.pipeline.Pipeline",
@@ -113,6 +115,7 @@ Here is all acceptable types of Snowflake native model packaging and its handler
 | snowflake.ml.model.custom_model.CustomModel | custom.py    | _CustomModelHandler |
 | sklearn.base.BaseEstimator      | sklearn.py   | _SKLModelHandler    |
 | sklearn.pipeline.Pipeline       | sklearn.py   | _SKLModelHandler    |
+| prophet.Prophet       | prophet.py   | ProphetHandler    |
 | xgboost.XGBModel       | xgboost.py   | _XGBModelHandler    |
 | xgboost.Booster        | xgboost.py   | _XGBModelHandler    |
 | lightgbm.LGBMModel       | lightgbm.py   | _LGBMModelHandler    |
@@ -134,6 +137,7 @@ SupportedModelHandlerType = Literal[
     "huggingface_pipeline",
     "lightgbm",
     "mlflow",
+    "prophet",
     "pytorch",
     "sentence_transformers",
     "sklearn",
@@ -248,11 +252,18 @@ class KerasSaveOptions(BaseModelSaveOption):
     cuda_version: NotRequired[str]
 
 
+class ProphetSaveOptions(BaseModelSaveOption):
+    target_methods: NotRequired[Sequence[str]]
+    date_column: NotRequired[str]
+    target_column: NotRequired[str]
+
+
 ModelSaveOption = Union[
     BaseModelSaveOption,
     CatBoostModelSaveOptions,
     CustomModelSaveOption,
     LGBMModelSaveOptions,
+    ProphetSaveOptions,
     SKLModelSaveOptions,
     XGBModelSaveOptions,
     SNOWModelSaveOptions,
@@ -327,11 +338,16 @@ class KerasLoadOptions(BaseModelLoadOption):
     use_gpu: NotRequired[bool]
 
 
+class ProphetLoadOptions(BaseModelLoadOption):
+    ...
+
+
 ModelLoadOption = Union[
     BaseModelLoadOption,
     CatBoostModelLoadOptions,
     CustomModelLoadOption,
     LGBMModelLoadOptions,
+    ProphetLoadOptions,
     SKLModelLoadOptions,
     XGBModelLoadOptions,
     SNOWModelLoadOptions,

snowflake/ml/modeling/_internal/snowpark_implementations/xgboost_external_memory_trainer.py

@@ -93,7 +93,7 @@ def get_data_iterator(
             cache_dir_name = tempfile.mkdtemp()
             super().__init__(cache_prefix=os.path.join(cache_dir_name, "cache"))
 
-        def next(self, batch_consumer_fn) -> int:  # type: ignore[no-untyped-def]
+        def next(self, batch_consumer_fn) -> bool | int:  # type: ignore[no-untyped-def]
             """Advance the iterator by 1 step and pass the data to XGBoost's batch_consumer_fn.
             This function is called by XGBoost during the construction of ``DMatrix``
 
@@ -101,7 +101,7 @@ def get_data_iterator(
                 batch_consumer_fn: batch consumer function
 
             Returns:
-                0 if there is no more data, else 1.
+                False/0 if there is no more data, else True/1.
             """
             while (self._df is None) or (self._df.shape[0] < self._batch_size):
                 # Read files and append data to temp df until batch size is reached.
@@ -117,7 +117,7 @@ def get_data_iterator(
 
             if (self._df is None) or (self._df.shape[0] == 0):
                 # No more data
-                return 0
+                return False
 
             # Slice the temp df and save the remainder in the temp df
             batch_end_index = min(self._batch_size, self._df.shape[0])
@@ -133,8 +133,8 @@ def get_data_iterator(
                 func_args["weight"] = batch_df[self._sample_weight_col].squeeze()
 
             batch_consumer_fn(**func_args)
-            # Return 1 to let XGBoost know we haven't seen all the files yet.
-            return 1
+            # Return True to let XGBoost know we haven't seen all the files yet.
+            return True
 
         def reset(self) -> None:
             """Reset the iterator to its beginning"""

snowflake/ml/modeling/metrics/metrics_utils.py

@@ -4,6 +4,7 @@ from typing import Any, Collection, Iterable, Optional, Union
 
 import cloudpickle
 import numpy as np
+from packaging import version
 
 import snowflake.snowpark._internal.utils as snowpark_utils
 from snowflake import snowpark
@@ -59,7 +60,10 @@ def register_accumulator_udtf(*, session: Session, statement_params: dict[str, A
             ]
         ),
         input_types=[T.BinaryType()],
-        packages=[f"numpy=={np.__version__}", f"cloudpickle=={cloudpickle.__version__}"],
+        packages=[
+            f"numpy=={version.parse(np.__version__).major}.*",
+            f"cloudpickle=={version.parse(cloudpickle.__version__).major}.*",
+        ],
         imports=[],  # Prevents unnecessary import resolution.
         name=accumulator,
         is_permanent=False,
@@ -175,7 +179,10 @@ def register_sharded_dot_sum_computer(*, session: Session, statement_params: dic
             ]
         ),
         input_types=[T.ArrayType(), T.IntegerType(), T.IntegerType()],
-        packages=[f"numpy=={np.__version__}", f"cloudpickle=={cloudpickle.__version__}"],
+        packages=[
+            f"numpy=={version.parse(np.__version__).major}.*",
+            f"cloudpickle=={version.parse(cloudpickle.__version__).major}.*",
+        ],
         imports=[],  # Prevents unnecessary import resolution.
         name=sharded_dot_and_sum_computer,
         is_permanent=False,

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.16.0"
+VERSION = "1.18.0"

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.16.0
+Version: 1.18.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: 
@@ -253,19 +253,20 @@ Requires-Dist: s3fs<2026,>=2024.6.1
 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
+Requires-Dist: snowflake-connector-python[pandas]<4,>=3.17.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: tqdm<5
 Requires-Dist: typing-extensions<5,>=4.1.0
-Requires-Dist: xgboost<3,>=1.7.3
+Requires-Dist: xgboost<4
 Provides-Extra: all
 Requires-Dist: altair<6,>=5; extra == "all"
 Requires-Dist: catboost<2,>=1.2.0; extra == "all"
 Requires-Dist: keras<4,>=2.0.0; extra == "all"
 Requires-Dist: lightgbm<5,>=4.1.0; extra == "all"
 Requires-Dist: mlflow<3,>=2.16.0; extra == "all"
+Requires-Dist: prophet<2,>=1.1.0; extra == "all"
 Requires-Dist: sentence-transformers<4,>=2.7.0; extra == "all"
 Requires-Dist: sentencepiece<0.2.0,>=0.1.95; extra == "all"
 Requires-Dist: streamlit<2,>=1.30.0; extra == "all"
@@ -286,6 +287,8 @@ Provides-Extra: lightgbm
 Requires-Dist: lightgbm<5,>=4.1.0; extra == "lightgbm"
 Provides-Extra: mlflow
 Requires-Dist: mlflow<3,>=2.16.0; extra == "mlflow"
+Provides-Extra: prophet
+Requires-Dist: prophet<2,>=1.1.0; extra == "prophet"
 Provides-Extra: streamlit
 Requires-Dist: streamlit<2,>=1.30.0; extra == "streamlit"
 Provides-Extra: tensorflow
@@ -406,6 +409,48 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
+## 1.18.0
+
+### Bug Fixes
+
+* Registry: The create_service API now validates that a model has a GPU runtime configuration and will throw a
+  descriptive error if the configuration is missing.
+
+### Behavior Changes
+
+### New Features
+
+* Registry (PrPr): Introducing `ModelVersion.run_batch` for batch inference in Snowpark Container Services.
+* Experiment Tracking (PrPr): Added `version_name` argument to the autologging callbacks
+  to specify the version name for the autologged model.
+
+### Deprecations
+
+* `Python 3.9` is deprecated.
+
+## 1.17.0
+
+### Bug Fixes
+
+* ML Job: Added support for retrieving details of deleted jobs, including status, compute pool, and target instances.
+
+### Behavior Changes
+
+### New Features
+
+* Support xgboost 3.x.
+* ML Job: Overhauled the `MLJob.result()` API with broader cross-version
+  compatibility and support for additional data types, namely:
+  * Pandas DataFrames
+  * PyArrow Tables
+  * NumPy arrays
+  * NOTE: Requires `snowflake-ml-python>=1.17.0` to be installed inside remote container environment.
+* ML Job: Enabled job submission v2 by default
+  * Jobs submitted using v2 will automatically use the latest Container Runtime image
+  * v1 behavior can be restored by setting environment variable `MLRS_USE_SUBMIT_JOB_V2` to `false`
+
+### Deprecations
+
 ## 1.16.0
 
 ### Bug Fixes
@@ -440,9 +485,10 @@ options = {
             "function_type": "TABLE_FUNCTION",
             "volatility": Volatility.VOLATILE,
         },
+    },
 }
 
-````
+```
 
 ## 1.15.0 (09-29-2025)
 

{snowflake_ml_python-1.16.0.dist-info → snowflake_ml_python-1.18.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=zo4wv3_2on676LLh7nL4I4AC6CM54PzfNBPoXameD38,99
+snowflake/ml/version.py,sha256=DlOpeyxcS6xpAWs8-Ni5jfcj--OA0JCQSc2w2BFU3l8,99
 snowflake/ml/_internal/env.py,sha256=EY_2KVe8oR3LgKWdaeRb5rRU-NDNXJppPDsFJmMZUUY,265
 snowflake/ml/_internal/env_utils.py,sha256=x6ID94g6FYoMX3afp0zoUHzBvuvPyiE2F6RDpxx5Cq0,30967
 snowflake/ml/_internal/file_utils.py,sha256=7sA6loOeSfmGP4yx16P4usT9ZtRqG3ycnXu7_Tk7dOs,14206
@@ -18,7 +18,7 @@ snowflake/ml/_internal/init_utils.py,sha256=WhrlvS-xcmKErSpwg6cUk6XDQ5lQcwDqPJnU
 snowflake/ml/_internal/migrator_utils.py,sha256=k3erO8x3YJcX6nkKeyJAUNGg1qjE3RFmD-W6dtLzIH0,161
 snowflake/ml/_internal/platform_capabilities.py,sha256=5cpeKpsxCObjOsPIz38noIusWw4n5KXOvPqRPiF3Kj4,7627
 snowflake/ml/_internal/relax_version_strategy.py,sha256=MYEIZrx1HfKNhl9Na3GN50ipX8c0MKIj9nwxjB0IC0Y,484
-snowflake/ml/_internal/telemetry.py,sha256=GCut6xG7SvAV8JRCxuQjvno9t7cLGLByECpMNUY1q30,31867
+snowflake/ml/_internal/telemetry.py,sha256=SL5_yXvRoqvmIckTpTHVwGsgPyg06uIIPZHMZWfdkzo,31922
 snowflake/ml/_internal/type_utils.py,sha256=bNNW0I9rOvwhx-Y274vGd0qWA0fMIPA3SGnaDE09wvc,2198
 snowflake/ml/_internal/exceptions/dataset_error_messages.py,sha256=h7uGJbxBM6se-TW_64LKGGGdBCbwflzbBnmijWKX3Gc,285
 snowflake/ml/_internal/exceptions/dataset_errors.py,sha256=TqESe8cDfWurJdv5X0DOwgzBfHCEqga_F3WQipYbdqg,741
@@ -29,8 +29,8 @@ snowflake/ml/_internal/exceptions/fileset_error_messages.py,sha256=dqPpRu0cKyQA_
 snowflake/ml/_internal/exceptions/fileset_errors.py,sha256=pHwY7f5c6JH-RZDtkiWy8nICHKy4T5vvWs5cq5rPD_4,1030
 snowflake/ml/_internal/exceptions/modeling_error_messages.py,sha256=M1s_PNHcOGlSDKD2kvSUQYsSaKHdHdnE74609LvF27c,749
 snowflake/ml/_internal/exceptions/sql_error_codes.py,sha256=aEI3-gW7FeNahoPncdOaGGRBmPJmkCHK-a1o2e3c3PI,206
-snowflake/ml/_internal/human_readable_id/adjectives.txt,sha256=5o4MbVeHoELAqyLpyuKleOKR47jPjC_nKoziOIZMwT0,804
-snowflake/ml/_internal/human_readable_id/animals.txt,sha256=GDLzMwzxiL07PhIMxw4t89bhYqqg0bQfPiuQT8VNeME,837
+snowflake/ml/_internal/human_readable_id/adjectives.txt,sha256=n_nj9zvA09H3cn3DoZBIuu_4ki76S1hBiuhyfbjUr3g,812
+snowflake/ml/_internal/human_readable_id/animals.txt,sha256=IycV3Cs9XgkCFo24MB3wDye2iBZHhF5YN_PFxXs3AOc,841
 snowflake/ml/_internal/human_readable_id/hrid_generator.py,sha256=LYWB86qZgsVBvnc6Q5VjfDOmnGSQU3cTRKfId_nJSPY,1341
 snowflake/ml/_internal/human_readable_id/hrid_generator_base.py,sha256=_Egc-L0DKWgug1WaJebLCayKcljr2WdPuqH5uIoR1Kg,4469
 snowflake/ml/_internal/lineage/lineage_utils.py,sha256=-_PKuznsL_w38rVj3wXgbPdm6XkcbnABrU4v4GwZQcg,3426
@@ -65,21 +65,21 @@ snowflake/ml/dataset/dataset_metadata.py,sha256=lcNvugBkP8YEkGMQqaV8SlHs5mwUKsUS
 snowflake/ml/dataset/dataset_reader.py,sha256=mZsG9HyWUGgfotrGkLrunyEsOm_659mH-Sn2OyG6A-Q,5036
 snowflake/ml/experiment/__init__.py,sha256=r7qdyPd3jwxzqvksim2ju5j_LrnYQrta0ZI6XpWUqmc,109
 snowflake/ml/experiment/_experiment_info.py,sha256=iaJ65x6nzBYJ5djleSOzBtMpZUJCUDlRpaDw0pu-dcU,2533
-snowflake/ml/experiment/experiment_tracking.py,sha256=5WEZpI19MCbqbozpWG88Ba7Kd5v7JvfNDvJA-tscMwo,17418
+snowflake/ml/experiment/experiment_tracking.py,sha256=r3iu3RRCRW3JRzW2bz94micO37pNk2Yi7Qrgq4LWf-k,18131
 snowflake/ml/experiment/utils.py,sha256=3bpbkilc5vvFjnti-kcyhhjAd9Ga3LqiKqJDwORiATY,628
 snowflake/ml/experiment/_client/artifact.py,sha256=R2WB4Y_kqv43BWLfXv8SEDINn1Bnevzgb-mH5LyvgGk,3035
-snowflake/ml/experiment/_client/experiment_tracking_sql_client.py,sha256=0cR2aTQa9uB9D7s5AStZ9gOAJe3SL7rFebttvpqkZFk,7250
+snowflake/ml/experiment/_client/experiment_tracking_sql_client.py,sha256=ISST57CEs12zGb76X-aKaitJX-bEzf-MpWc8T5OoqxE,8317
 snowflake/ml/experiment/_entities/__init__.py,sha256=11XxkvAzosydf5owNmMzLwXZdQ2NtNKRM-MMra4ND2k,247
 snowflake/ml/experiment/_entities/experiment.py,sha256=lKmQj59K8fGDWVwRqeIesxorrChb-m78vX_WUmI7PV0,225
 snowflake/ml/experiment/_entities/run.py,sha256=JkhiS4UZWuRm3ZSLgc2uktedeag5Voih2r02YFr6DQk,1621
 snowflake/ml/experiment/_entities/run_metadata.py,sha256=25cIg8FnAYHk5SoTg_StzL10_BkomL7xrhMmWxUTU8E,366
-snowflake/ml/experiment/callback/keras.py,sha256=7oq23irYkBV7bLFBCxxKlf9pL4YuDFJDCZ8xtffVRFI,2547
-snowflake/ml/experiment/callback/lightgbm.py,sha256=5co7eR_t651cq1WTK9JCQjhSlYc2oIvxaf3aVnVOlR4,2613
-snowflake/ml/experiment/callback/xgboost.py,sha256=eZMRFAebMERwdqMFm7i6S9wkHD7_VLcwIP0OkWHDOMM,2621
+snowflake/ml/experiment/callback/keras.py,sha256=I_O2SBYttFNChO2Sc_C6xQh03r3ymSFB4eN2TS41Dgs,2680
+snowflake/ml/experiment/callback/lightgbm.py,sha256=qu4m8WV6Rqxa39X7g7ZBd1zJ8icYEkBBF3Kh3C1VpHU,2754
+snowflake/ml/experiment/callback/xgboost.py,sha256=F547AXZ7Gv39cyIrgRdxVE8MQ3VlNi5JqKKW0Z5RlQo,2754
 snowflake/ml/feature_store/__init__.py,sha256=MJr2Gp_EimDgDxD6DtenOEdLTzg6NYPfdNiPM-5rEtw,406
 snowflake/ml/feature_store/access_manager.py,sha256=Q5ImMXRY8WA5X5dpBMzHnIJmeyKVShjNAlbn3cQb4N8,10654
 snowflake/ml/feature_store/entity.py,sha256=ViOSlqCV17ouiO4iH-_KvkvJZqSzpf-nfsjijG6G1Uk,4047
-snowflake/ml/feature_store/feature_store.py,sha256=r51z1SC6e7Sz68dz8pIWQYscVgtG68yiOncBJe6Fcyc,161425
+snowflake/ml/feature_store/feature_store.py,sha256=dUrlXqZYFvdmBk5HRHmRUVcNWcUCZZAahKsrh4DSYqg,169501
 snowflake/ml/feature_store/feature_view.py,sha256=fCciDGC8qd-ySGVEHKHo-PYIFrAuFlVAFDu8ZiRTDIY,44141
 snowflake/ml/feature_store/examples/example_helper.py,sha256=eaD2vLe7y4C5hMZQTeMXylbTtLacbq9gJcAluGHrkug,12470
 snowflake/ml/feature_store/examples/airline_features/entities.py,sha256=mzHRS-InHpXON0eHds-QLmi7nK9ciOnCruhPZI4niLs,438
@@ -108,29 +108,36 @@ snowflake/ml/fileset/fileset.py,sha256=ApMpHiiyzGRkyxQfJPdXPuKtw_wOXbOfQCXSH6pDw
 snowflake/ml/fileset/sfcfs.py,sha256=FJFc9-gc0KXaNyc10ZovN_87aUCShb0WztVwa02t0io,15517
 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/__init__.py,sha256=h176wKqEylZs5cdWdzWHuUrSAcwctDdw4tUhIpy-mO4,657
 snowflake/ml/jobs/decorators.py,sha256=mQgdWvvCwD7q79cSFKZHKegXGh2j1u8WM64UD3lCKr4,3428
-snowflake/ml/jobs/job.py,sha256=VFBogPXXTWa0p-Jl10lSUFyKeqGQOtbzJIgrWTPA0rQ,22222
-snowflake/ml/jobs/manager.py,sha256=kRv4LyNP_dgKSpbPRfQWAOYUHxRj92PIujiJfgSf9Tc,28610
+snowflake/ml/jobs/job.py,sha256=Pwc71PcZ9t92Z1tj_-O-fZOOQhqgC3wjkK_uuS29uEs,27532
+snowflake/ml/jobs/manager.py,sha256=PuM3WZQrB3wbtBcQ7p7X3wxAz6hq6Of0p-_hkjg4sfs,29092
+snowflake/ml/jobs/_interop/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+snowflake/ml/jobs/_interop/data_utils.py,sha256=xUO5YlhUKFVCDtbjam5gP2lka3lfoknTLr7syNAVxK0,4074
+snowflake/ml/jobs/_interop/dto_schema.py,sha256=NhoQ6WJa7uLO9VJojEENVVZhZMfL_G1VPPSSUYmmhO8,2750
+snowflake/ml/jobs/_interop/exception_utils.py,sha256=ZCphBkaaNINFATXZmjPzzNLKZZxKvGPROZ2azU8w13g,16348
+snowflake/ml/jobs/_interop/legacy.py,sha256=8BuC197e6nPmAzh4urYiuBuCNP-RlOlvWnWpSHTduqQ,9238
+snowflake/ml/jobs/_interop/protocols.py,sha256=xfOXL25hxhhy3ULfZWOfEjX0XqSTxo5cOURPXt777W4,18175
+snowflake/ml/jobs/_interop/results.py,sha256=nQ07XJ1BZEkPB4xa12pbGyaKqR8sWCoSzx0IKQlub4w,1714
+snowflake/ml/jobs/_interop/utils.py,sha256=TWFkUcAYmb-fpTwEL8idkk3XxlZ8vLz4X_gyS78PSi4,5552
 snowflake/ml/jobs/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-snowflake/ml/jobs/_utils/constants.py,sha256=JB8i85VbI0nIYP_c2nyAADQcmEF8BscVHHKBGhruofY,3838
-snowflake/ml/jobs/_utils/feature_flags.py,sha256=cH_NyeOncL3_tzbk0WvL1siNyodxBgn1ziPk2yBW6wY,404
+snowflake/ml/jobs/_utils/constants.py,sha256=Wp2s_wBX5WZnxo3cdrsitnb9Ze0OUGmH26sofDFrdMI,4613
+snowflake/ml/jobs/_utils/feature_flags.py,sha256=c69OYFOZyXVmj87VKRh-rp_MP-3I1gJXhxBSiXAprbQ,1612
 snowflake/ml/jobs/_utils/function_payload_utils.py,sha256=4LBaStMdhRxcqwRkwFje-WwiEKRWnBfkaOYouF3N3Kg,1308
-snowflake/ml/jobs/_utils/interop_utils.py,sha256=7mODMTjKCLXkJloACG6_9b2wvmRgjXF0Jx3wpWYyJeA,21413
-snowflake/ml/jobs/_utils/payload_utils.py,sha256=1Xon3jlBgzfv1SQgQkJ1ir3xt9PVviP8-UC6P-FOmwc,30807
+snowflake/ml/jobs/_utils/payload_utils.py,sha256=INq_G1flV-Sa2riuqKwx5DOWTUegcDF01jfmJKpBcao,31101
 snowflake/ml/jobs/_utils/query_helper.py,sha256=1-XK-y4iukbR1693qAELprRbHmJDM4YoEBHov8IYbHU,1115
 snowflake/ml/jobs/_utils/runtime_env_utils.py,sha256=fqa3ctf_CAOSv1zT__01Qp9T058mKgMjXuEkBZqKUqA,2247
-snowflake/ml/jobs/_utils/spec_utils.py,sha256=dVhz4AVRqVOkULgc4PyQmtBRYw8SL21-YfpN27w4hVI,16269
-snowflake/ml/jobs/_utils/stage_utils.py,sha256=38-LsokaGx0NzlnP8CMRioClRz-3x6xhPiZIgl2CB9g,5224
-snowflake/ml/jobs/_utils/types.py,sha256=AGLu0kPTNRUki26rah_KBwWp0bBJEtUP3zcfxkj5kB0,2326
+snowflake/ml/jobs/_utils/spec_utils.py,sha256=Ch-3iKezKWXgSJm-xpHOW7ZpMBjIZvSNiEZGL9CyA2w,16346
+snowflake/ml/jobs/_utils/stage_utils.py,sha256=YjN7cQFRcAUT1JvNZDSiNw8KiCF4HJ6ymkOYMhYJwE0,5297
+snowflake/ml/jobs/_utils/types.py,sha256=uOf7TPPWfIRALZhD6Li3AgizPOktPXv8_6iVK2grkgc,2587
 snowflake/ml/jobs/_utils/scripts/constants.py,sha256=YyIWZqQPYOTtgCY6SfyJjk2A98I5RQVmrOuLtET5Pqg,173
 snowflake/ml/jobs/_utils/scripts/get_instance_ip.py,sha256=N2wJYMPlwg-hidwgHhDhiBWOE6TskqCfWLMRRNnZBQs,5776
-snowflake/ml/jobs/_utils/scripts/mljob_launcher.py,sha256=DPRupPuFqJ9TBbMCuomCf3jRqBJ_QQ1nCplg28HYt04,15752
+snowflake/ml/jobs/_utils/scripts/mljob_launcher.py,sha256=XcCyXYkwAAF3quPs0uoq_n-OiEYPYJtadKGsOFKBlTM,17005
 snowflake/ml/jobs/_utils/scripts/signal_workers.py,sha256=AR1Pylkm4-FGh10WXfrCtcxaV0rI7IQ2ZiO0Li7zZ3U,7433
 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=SA0rbbI67rMr1qTMs4bAVkvqVtuKNI4lIaO5w0S-IXE,5767
-snowflake/ml/model/__init__.py,sha256=78w63Y250_m2zsN6eamZAZ2ovPpxk2ZCYOClTwJS1-s,567
+snowflake/ml/model/__init__.py,sha256=TV9yOwEKUWXKHiZ-z7mIwS2S6-bhCHOHU43NhVgLaOw,1012
 snowflake/ml/model/custom_model.py,sha256=fDhMObqlyzD_qQG1Bq6HHkBN1w3Qzg9e81JWPiqRfc4,12249
 snowflake/ml/model/event_handler.py,sha256=pojleQVM9TPNeDvliTvon2Sfxqbf2WWxrOebo1SaEHo,7211
 snowflake/ml/model/inference_engine.py,sha256=L0nwySY2Qwp3JzuRpPS87r0--m3HTUNUgZXYyOPJjyk,66
@@ -138,44 +145,46 @@ snowflake/ml/model/model_signature.py,sha256=RH62vv4YmrQugTXLsh6kyuzfTs9_yz8a0TM
 snowflake/ml/model/openai_signatures.py,sha256=ZVnHDgaOA6RcvtSP3HIbHVgr3scJH2gG_9QvZidn41g,2630
 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=VmP8qr60V9mBZYuTeGRWIZP3w14NXERNExrXQLx3jWI,10836
+snowflake/ml/model/type_hints.py,sha256=hoIq3KOscvp9rqJnmgWHW3IGwCSwiCVbklFAqSQekr4,11225
 snowflake/ml/model/volatility.py,sha256=qu-wqe9oKkRwXwE2qkKygxTWzUypQYEk3UjsqOGRl_I,1144
-snowflake/ml/model/_client/model/batch_inference_specs.py,sha256=FFzNvP-OHO1gFBfqXz89T5HMheFGfsx7P5_5Ze_QYvM,957
-snowflake/ml/model/_client/model/inference_engine_utils.py,sha256=lOqZzySZygeWqHTNYGBYgpTRfEst9f7lX50Ku8k950g,1966
+snowflake/ml/model/_client/model/batch_inference_specs.py,sha256=0SlLTpZW_gzNP5IH_8cFnqjArxM0zVjA5nBLKnKAnz8,4396
+snowflake/ml/model/_client/model/inference_engine_utils.py,sha256=L8HnoAEbnN5YAcMlsgNbeqfyZbiOyrNMxj7rD4DcjyU,1878
 snowflake/ml/model/_client/model/model_impl.py,sha256=Yabrbir5vPMOnsVmQJ23YN7vqhi756Jcm6pfO8Aq92o,17469
-snowflake/ml/model/_client/model/model_version_impl.py,sha256=DgdbEWkgAvfvO-xO_Bg-qFzwfAGT6tst-7sW_TZzQ3Q,53232
+snowflake/ml/model/_client/model/model_version_impl.py,sha256=v03RFKRnu7Gr8lovgRjE-MUK1MY_-RD5zjrrrbliJq4,60470
 snowflake/ml/model/_client/ops/metadata_ops.py,sha256=qpK6PL3OyfuhyOmpvLCpHLy6vCxbZbp1HlEvakFGwv4,4884
-snowflake/ml/model/_client/ops/model_ops.py,sha256=XGQB62Nc4ld86I7wJzuJGkmbrI38wIBm4egsZptdrM8,50394
-snowflake/ml/model/_client/ops/service_ops.py,sha256=gUmYDvXnacachE9OK92HGYbZ7ec9WEsrDRsOgOckhbI,47036
-snowflake/ml/model/_client/service/model_deployment_spec.py,sha256=Ro2kAM_rfMccMjW23RpP6qDPq090vAIUv_he-8GE68k,19487
-snowflake/ml/model/_client/service/model_deployment_spec_schema.py,sha256=QpDso2bjx2eCRKIG4-ppc3z46B7hgYMZehOTRoR9IJs,2425
+snowflake/ml/model/_client/ops/model_ops.py,sha256=sQ0t6plS5o_2qqnA48quYwJOZ0NlhgnZMarwrzSKwew,50599
+snowflake/ml/model/_client/ops/service_ops.py,sha256=Ey_yvKQvFnD4dafjFtPA3aaU1GTGqrdlgIpjrfYC8Ew,47143
+snowflake/ml/model/_client/service/model_deployment_spec.py,sha256=n1i19opcjocXflWdr2jjtNk1GVqw8YSwip_ki2XyVMc,19628
+snowflake/ml/model/_client/service/model_deployment_spec_schema.py,sha256=Nlw-rwCGmiGqPYwWjZrowPtcRvgYMInpmWZvsEC4sTI,2464
 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
+snowflake/ml/model/_client/sql/model_version.py,sha256=JE974ehlquitpDK9YHv94QklyereYk_vPiz64WYNXSk,23673
 snowflake/ml/model/_client/sql/service.py,sha256=0aXyRDZIFCgBq6TEG6qdhc7wtCsmphpyBXuSoNyLmTw,11630
 snowflake/ml/model/_client/sql/stage.py,sha256=1TWYIVoWIeNwhVG9uqwmNpmKcC6x45LrbxCtzJW7fi4,1214
 snowflake/ml/model/_client/sql/tag.py,sha256=9sI0VoldKmsfToWSjMQddozPPGCxYUI6n0gPBiqd6x8,4333
 snowflake/ml/model/_model_composer/model_composer.py,sha256=Xqi-sxmkBoZl383LQAXhMQkq9KsAS0A3ythC5bN3EOU,8292
-snowflake/ml/model/_model_composer/model_manifest/model_manifest.py,sha256=y6lu1_4UC7pfosBKofc0dl-LWF8mpcTqhY5sKSkUH_I,9247
+snowflake/ml/model/_model_composer/model_manifest/model_manifest.py,sha256=MyNaOLRD94RiMf-GofqPP989ifTLrSOOOau2RHCHphg,9308
 snowflake/ml/model/_model_composer/model_manifest/model_manifest_schema.py,sha256=vfuZQ4YADfLcEvJYTSqVMsLfxCO-QWsFFzLXI1NoMXk,2950
 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
 snowflake/ml/model/_model_composer/model_method/infer_function.py_template,sha256=olysEb_bE2C8CjIRAhm7qdr2mtgk77Tx45gnLRVQGFw,1511
 snowflake/ml/model/_model_composer/model_method/infer_partitioned.py_template,sha256=8p8jkTOJA-mBt5cuGhcWSH4z7ySQ9xevC35UioCLkC8,1539
 snowflake/ml/model/_model_composer/model_method/infer_table_function.py_template,sha256=QT32N6akQDutLh00cXp2OD4WI6Gb7IGG1snsnrXNih8,1453
-snowflake/ml/model/_model_composer/model_method/model_method.py,sha256=4H1-1N8lQpkqgHXWlzvj7u8joF5okt1cwais08BUm7k,8229
+snowflake/ml/model/_model_composer/model_method/model_method.py,sha256=7rfNQPgWT3scRIFAgIGdzrJaWeUftKhI2MGG5PFBDfM,8700
+snowflake/ml/model/_model_composer/model_method/utils.py,sha256=RQi2qebBeE-0Y-jLYXiDWZU8nfvbnif9QbExeWiMmyI,1057
 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=6-1MnGUR8nxB86A13nCZcWbET_Q6fSEOlyfcbTv7xCI,6087
-snowflake/ml/model/_packager/model_env/model_env.py,sha256=sQeElFH7qW3-dXTHWO48xCtlBrKCNJgeNEorad4t2cU,20231
+snowflake/ml/model/_packager/model_env/model_env.py,sha256=xDDyRr8AzME0SRv2mQxzfh-blh2MH7Fz8H7R5HXiVJQ,21085
 snowflake/ml/model/_packager/model_handlers/_base.py,sha256=OZhGv7nyej3PqaoBz021uGa40T06d9rv-kDcKUY3VnM,7152
-snowflake/ml/model/_packager/model_handlers/_utils.py,sha256=DIN1oKCl4DytNcH1xP3fGl4BHaEmQ_RGoKuysFiWz7s,12599
+snowflake/ml/model/_packager/model_handlers/_utils.py,sha256=v_IjjbvzJDqrAYSq4_l7_CiN8vkMzLx5MlYDJ_oL970,15522
 snowflake/ml/model/_packager/model_handlers/catboost.py,sha256=dbI2QizGZS04l6ehgXb3oy5YSXrlwRHz8YENVefEbms,10676
 snowflake/ml/model/_packager/model_handlers/custom.py,sha256=fM_13N5ejT0Ta0-M_Uzsqr_TwGVk_3jSjsLJiMEfyR4,8514
 snowflake/ml/model/_packager/model_handlers/huggingface_pipeline.py,sha256=4k05ouWc7qxTMX-tXJ4z9KJHWPcVF62vIkoM1XwBTAc,37246
 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/prophet.py,sha256=MzmIkX2WukTApf85SUEmn8k_jeBAhfGa_8RTZK0-aCA,23913
 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=_D1YE7TmEJDsuOUt-mT-2Nza2Bz0sIzSGRKn9sxuDhI,18340
@@ -189,12 +198,12 @@ 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=CctjNVwdC7ghVIPqbhb62t43SOFsmk2j2FdoZMZ8KXs,20063
-snowflake/ml/model/_packager/model_meta/model_meta_schema.py,sha256=UGPTjzxLBUKn8XSAuMvGnJGdjEvlUZjjfvI9rDRAQl4,3759
+snowflake/ml/model/_packager/model_meta/model_meta.py,sha256=7FuL0nClHpd7kBLhSM6BJPd6JCdyXLo4dePGThpcxfg,20549
+snowflake/ml/model/_packager/model_meta/model_meta_schema.py,sha256=mtKRbHQb6Hq2xiH1fTHSlBGaZg94qfyJ90rYRg14l0Y,3992
 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
 snowflake/ml/model/_packager/model_meta_migrator/migrator_v1.py,sha256=cyZVvBGM3nF1IVqDKfYstLCchNO-ZhSkPvLM4aU7J5c,2066
-snowflake/ml/model/_packager/model_runtime/_snowml_inference_alternative_requirements.py,sha256=xtyKWzGDR9IrUsS1p63RCdFifP0bZqKR-Re8Hq6ZkB8,912
+snowflake/ml/model/_packager/model_runtime/_snowml_inference_alternative_requirements.py,sha256=xVdhOAl8aj6B_zWHjrqq0YUQrgjZXL7SRdlazq682jo,904
 snowflake/ml/model/_packager/model_runtime/model_runtime.py,sha256=xEf-S9QurEOeQzrNxlc-4-S_VkHsVO1eNS4UR0hWwHU,5495
 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
@@ -220,7 +229,7 @@ snowflake/ml/modeling/_internal/snowpark_implementations/distributed_hpo_trainer
 snowflake/ml/modeling/_internal/snowpark_implementations/distributed_search_udf_file.py,sha256=HnsSmsXAeJrH9zVeq3CSziIaCUDxeWWx6kRyAK4qajM,6601
 snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_handlers.py,sha256=oXumJxQFMokoxsrXZ03X8NKLWr3yGuUGB3OM8qTTH4E,16416
 snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_trainer.py,sha256=xem3xtoOHi_HFoi85wvSx7F1BhzxVrGYqMhuyrFz4Ik,32919
-snowflake/ml/modeling/_internal/snowpark_implementations/xgboost_external_memory_trainer.py,sha256=Cu_ywcFzlkflbUvJ5C8rNk1H3YwRDEhVdsyngNcjE2Q,17282
+snowflake/ml/modeling/_internal/snowpark_implementations/xgboost_external_memory_trainer.py,sha256=nfdUDS7tJY2qPbYCRG-p05v8l0vVZq010mOHfU90US0,17310
 snowflake/ml/modeling/calibration/__init__.py,sha256=rY5qSOkHj59bHiTV6LhBiEhUA0StoCb0ACNR2vkV4v0,297
 snowflake/ml/modeling/calibration/calibrated_classifier_cv.py,sha256=UfQ3LkyxPC2F8USLcofkiCPz_7uFXWx69pzr81JfE18,54177
 snowflake/ml/modeling/cluster/__init__.py,sha256=rY5qSOkHj59bHiTV6LhBiEhUA0StoCb0ACNR2vkV4v0,297
@@ -355,7 +364,7 @@ snowflake/ml/modeling/metrics/__init__.py,sha256=1lc1DCVNeo7D-gvvCjmpI5tFIIrOsEd
 snowflake/ml/modeling/metrics/classification.py,sha256=UOc2w9iGkLzuleTpxCbfhAWpbli0HvNsGsN-r8G0ztI,66433
 snowflake/ml/modeling/metrics/correlation.py,sha256=N7GIT-EVlvyh_WMC-zOUzDUUQeKU1IXu4ocOjnx-WQo,5187
 snowflake/ml/modeling/metrics/covariance.py,sha256=HxJK1mwyt6lMSg8yonHFQ8IxAEa62MHeb1M3eHEtqlk,4672
-snowflake/ml/modeling/metrics/metrics_utils.py,sha256=XuAjYfL437LCeBY8RMElunk8jgVzemAgln573JzS3Qk,13315
+snowflake/ml/modeling/metrics/metrics_utils.py,sha256=vNeyYJHJBKXGrL2P-5PBD22Y5umRBOAVKLCZ3obbOdI,13507
 snowflake/ml/modeling/metrics/ranking.py,sha256=NwMdH_nubwdpIcCAZFEyafw_46uS9ULGdWkMgstGwjk,17774
 snowflake/ml/modeling/metrics/regression.py,sha256=qHUdhRkRssl2BDLyUyn5vZQqcrSVxp3TgTWa1kh1Mso,26052
 snowflake/ml/modeling/mixture/__init__.py,sha256=rY5qSOkHj59bHiTV6LhBiEhUA0StoCb0ACNR2vkV4v0,297
@@ -442,8 +451,8 @@ snowflake/ml/utils/connection_params.py,sha256=NSBUgcs-DXPRHs1BKpxdSubbJx1yrFRlM
 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.16.0.dist-info/licenses/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
-snowflake_ml_python-1.16.0.dist-info/METADATA,sha256=sHObJCXSJ0RKqaUrM_jSwLZiR1VA65K_B4kumAwXIvg,95756
-snowflake_ml_python-1.16.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-snowflake_ml_python-1.16.0.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
-snowflake_ml_python-1.16.0.dist-info/RECORD,,
+snowflake_ml_python-1.18.0.dist-info/licenses/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
+snowflake_ml_python-1.18.0.dist-info/METADATA,sha256=HhlNlP1kqGKztNYHt4n9EB4Vs6nFglrHUyTdC7xYHuo,97170
+snowflake_ml_python-1.18.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+snowflake_ml_python-1.18.0.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
+snowflake_ml_python-1.18.0.dist-info/RECORD,,