snowflake-ml-python 1.24.0__py3-none-any.whl → 1.25.0__py3-none-any.whl

snowflake/ml/model/openai_signatures.py

@@ -88,6 +88,96 @@ _OPENAI_CHAT_SIGNATURE_SPEC = core.ModelSignature(
     ],
 )
 
+_OPENAI_CHAT_SIGNATURE_WITH_PARAMS_SPEC = core.ModelSignature(
+    inputs=[
+        core.FeatureGroupSpec(
+            name="messages",
+            specs=[
+                core.FeatureGroupSpec(
+                    name="content",
+                    specs=[
+                        core.FeatureSpec(name="type", dtype=core.DataType.STRING),
+                        # Text prompts
+                        core.FeatureSpec(name="text", dtype=core.DataType.STRING),
+                        # Image URL prompts
+                        core.FeatureGroupSpec(
+                            name="image_url",
+                            specs=[
+                                # Base64 encoded image URL or image URL
+                                core.FeatureSpec(name="url", dtype=core.DataType.STRING),
+                                # Image detail level (e.g., "low", "high", "auto")
+                                core.FeatureSpec(name="detail", dtype=core.DataType.STRING),
+                            ],
+                        ),
+                        # Video URL prompts
+                        core.FeatureGroupSpec(
+                            name="video_url",
+                            specs=[
+                                # Base64 encoded video URL
+                                core.FeatureSpec(name="url", dtype=core.DataType.STRING),
+                            ],
+                        ),
+                        # Audio prompts
+                        core.FeatureGroupSpec(
+                            name="input_audio",
+                            specs=[
+                                core.FeatureSpec(name="data", dtype=core.DataType.STRING),
+                                core.FeatureSpec(name="format", dtype=core.DataType.STRING),
+                            ],
+                        ),
+                    ],
+                    shape=(-1,),
+                ),
+                core.FeatureSpec(name="name", dtype=core.DataType.STRING),
+                core.FeatureSpec(name="role", dtype=core.DataType.STRING),
+                core.FeatureSpec(name="title", dtype=core.DataType.STRING),
+            ],
+            shape=(-1,),
+        ),
+    ],
+    outputs=[
+        core.FeatureSpec(name="id", dtype=core.DataType.STRING),
+        core.FeatureSpec(name="object", dtype=core.DataType.STRING),
+        core.FeatureSpec(name="created", dtype=core.DataType.FLOAT),
+        core.FeatureSpec(name="model", dtype=core.DataType.STRING),
+        core.FeatureGroupSpec(
+            name="choices",
+            specs=[
+                core.FeatureSpec(name="index", dtype=core.DataType.INT32),
+                core.FeatureGroupSpec(
+                    name="message",
+                    specs=[
+                        core.FeatureSpec(name="content", dtype=core.DataType.STRING),
+                        core.FeatureSpec(name="name", dtype=core.DataType.STRING),
+                        core.FeatureSpec(name="role", dtype=core.DataType.STRING),
+                    ],
+                ),
+                core.FeatureSpec(name="logprobs", dtype=core.DataType.STRING),
+                core.FeatureSpec(name="finish_reason", dtype=core.DataType.STRING),
+            ],
+            shape=(-1,),
+        ),
+        core.FeatureGroupSpec(
+            name="usage",
+            specs=[
+                core.FeatureSpec(name="completion_tokens", dtype=core.DataType.INT32),
+                core.FeatureSpec(name="prompt_tokens", dtype=core.DataType.INT32),
+                core.FeatureSpec(name="total_tokens", dtype=core.DataType.INT32),
+            ],
+        ),
+    ],
+    params=[
+        core.ParamSpec(name="temperature", dtype=core.DataType.DOUBLE, default_value=1.0),
+        core.ParamSpec(name="max_completion_tokens", dtype=core.DataType.INT64, default_value=250),
+        core.ParamSpec(name="stop", dtype=core.DataType.STRING, default_value=""),
+        core.ParamSpec(name="n", dtype=core.DataType.INT32, default_value=1),
+        core.ParamSpec(name="stream", dtype=core.DataType.BOOL, default_value=False),
+        core.ParamSpec(name="top_p", dtype=core.DataType.DOUBLE, default_value=1.0),
+        core.ParamSpec(name="frequency_penalty", dtype=core.DataType.DOUBLE, default_value=0.0),
+        core.ParamSpec(name="presence_penalty", dtype=core.DataType.DOUBLE, default_value=0.0),
+    ],
+)
+
 _OPENAI_CHAT_SIGNATURE_SPEC_WITH_CONTENT_FORMAT_STRING = core.ModelSignature(
     inputs=[
         core.FeatureGroupSpec(
@@ -142,6 +232,62 @@ _OPENAI_CHAT_SIGNATURE_SPEC_WITH_CONTENT_FORMAT_STRING = core.ModelSignature(
     ],
 )
 
+_OPENAI_CHAT_SIGNATURE_WITH_PARAMS_SPEC_WITH_CONTENT_FORMAT_STRING = core.ModelSignature(
+    inputs=[
+        core.FeatureGroupSpec(
+            name="messages",
+            specs=[
+                core.FeatureSpec(name="content", dtype=core.DataType.STRING),
+                core.FeatureSpec(name="name", dtype=core.DataType.STRING),
+                core.FeatureSpec(name="role", dtype=core.DataType.STRING),
+                core.FeatureSpec(name="title", dtype=core.DataType.STRING),
+            ],
+            shape=(-1,),
+        ),
+    ],
+    outputs=[
+        core.FeatureSpec(name="id", dtype=core.DataType.STRING),
+        core.FeatureSpec(name="object", dtype=core.DataType.STRING),
+        core.FeatureSpec(name="created", dtype=core.DataType.FLOAT),
+        core.FeatureSpec(name="model", dtype=core.DataType.STRING),
+        core.FeatureGroupSpec(
+            name="choices",
+            specs=[
+                core.FeatureSpec(name="index", dtype=core.DataType.INT32),
+                core.FeatureGroupSpec(
+                    name="message",
+                    specs=[
+                        core.FeatureSpec(name="content", dtype=core.DataType.STRING),
+                        core.FeatureSpec(name="name", dtype=core.DataType.STRING),
+                        core.FeatureSpec(name="role", dtype=core.DataType.STRING),
+                    ],
+                ),
+                core.FeatureSpec(name="logprobs", dtype=core.DataType.STRING),
+                core.FeatureSpec(name="finish_reason", dtype=core.DataType.STRING),
+            ],
+            shape=(-1,),
+        ),
+        core.FeatureGroupSpec(
+            name="usage",
+            specs=[
+                core.FeatureSpec(name="completion_tokens", dtype=core.DataType.INT32),
+                core.FeatureSpec(name="prompt_tokens", dtype=core.DataType.INT32),
+                core.FeatureSpec(name="total_tokens", dtype=core.DataType.INT32),
+            ],
+        ),
+    ],
+    params=[
+        core.ParamSpec(name="temperature", dtype=core.DataType.DOUBLE, default_value=1.0),
+        core.ParamSpec(name="max_completion_tokens", dtype=core.DataType.INT64, default_value=250),
+        core.ParamSpec(name="stop", dtype=core.DataType.STRING, default_value=""),
+        core.ParamSpec(name="n", dtype=core.DataType.INT32, default_value=1),
+        core.ParamSpec(name="stream", dtype=core.DataType.BOOL, default_value=False),
+        core.ParamSpec(name="top_p", dtype=core.DataType.DOUBLE, default_value=1.0),
+        core.ParamSpec(name="frequency_penalty", dtype=core.DataType.DOUBLE, default_value=0.0),
+        core.ParamSpec(name="presence_penalty", dtype=core.DataType.DOUBLE, default_value=0.0),
+    ],
+)
+
 
 # Refer vLLM documentation: https://docs.vllm.ai/en/stable/serving/openai_compatible_server/#chat-template
 
@@ -152,3 +298,11 @@ OPENAI_CHAT_SIGNATURE_WITH_CONTENT_FORMAT_STRING = {"__call__": _OPENAI_CHAT_SIG
 # This is the default signature.
 # The content format allows vLLM to handler content parts like text, image, video, audio, file, etc.
 OPENAI_CHAT_SIGNATURE = {"__call__": _OPENAI_CHAT_SIGNATURE_SPEC}
+
+# Use this signature to leverage ParamSpec with the default ChatML template.
+OPENAI_CHAT_WITH_PARAMS_SIGNATURE = {"__call__": _OPENAI_CHAT_SIGNATURE_WITH_PARAMS_SPEC}
+
+# Use this signature to leverage ParamSpec with the content format string.
+OPENAI_CHAT_WITH_PARAMS_SIGNATURE_WITH_CONTENT_FORMAT_STRING = {
+    "__call__": _OPENAI_CHAT_SIGNATURE_WITH_PARAMS_SPEC_WITH_CONTENT_FORMAT_STRING
+}

snowflake/ml/registry/_manager/model_parameter_reconciler.py

@@ -193,12 +193,11 @@ class ModelParameterReconciler:
         if enable_explainability:
             if only_spcs or not is_warehouse_runnable:
                 raise ValueError(
-                    "`enable_explainability` cannot be set to True when the model is not runnable in WH "
-                    "or the target platforms include SPCS."
+                    "`enable_explainability` cannot be set to True when the model cannot run in Warehouse."
                 )
             elif has_both_platforms:
                 warnings.warn(
-                    ("Explain function will only be available for model deployed to warehouse."),
+                    ("Explain function will only be available for model deployed to Warehouse."),
                     category=UserWarning,
                     stacklevel=2,
                 )

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.24.0"
+VERSION = "1.25.0"

{snowflake_ml_python-1.24.0.dist-info → snowflake_ml_python-1.25.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.24.0
+Version: 1.25.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: 
@@ -417,6 +417,37 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
+## 1.25.0
+
+### New Features
+
+* ML Job: Added support for creating ML job definitions and launching jobs with different
+  arguments without re-uploading payloads.
+
+* Inference Autocapture (PuPr): The `create_service` API will now accept `autocapture` as a new argument to indicate
+  whether inference data will be captured.
+
+* Model serving: Introduced the `min_instances` field in the `mv.create_service()` and
+  `HuggingFacePipelineModel.log_model_and_create_service()` APIs (defaulting to 0). The service now launches
+  with the `min_instances` and automatically scales between `min_instances` and `max_instances` based on
+  traffic and hardware utilization. When `min_instances` is set to 0, the service will automatically suspend
+  if no traffic is detected for a period of time.
+
+### Bug Fixes
+
+### Behavior Changes
+
+* Inference Autocapture (PuPr): `list_services()` now shows `autocapture_enabled` column to indicate if model
+  service has autocapture enabled.
+
+* Model serving: The `mv.create_service()` and `HuggingFacePipelineModel.log_model_and_create_service()` APIs now
+  include a `min_instances` field (defaulting to 0). When these APIs are called without specifying `min_instances`,
+  the system will now launch the service with 1 instance and enable auto scaling. This replaces the previous behavior,
+  where `min_instances` was automatically set to match `max_instances`, resulting in the immediate launch of the
+  maximum number of instances.
+
+### Deprecations
+
 ## 1.24.0
 
 ### New Features
@@ -520,7 +551,15 @@ x_df = pd.DataFrame.from_records(
     [
         {
             "messages": [
-                {"role": "system", "content": "Complete the sentence."},
+                {
+                  "role": "system",
+                  "content": [
+                        {
+                            "type": "text",
+                            "text": "Complete the sentence."
+                        },
+                  ]
+                },
                 {
                     "role": "user",
                     "content": [

{snowflake_ml_python-1.24.0.dist-info → snowflake_ml_python-1.25.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=esv6fY3HPIokUy4DELkMBOWM2pRI2gTUZ-kkGwHiP-M,99
+snowflake/ml/version.py,sha256=V1sQKmkr-QMsegsFN8N1EoPDeU1CzZe6ofRecCIByco,99
 snowflake/ml/_internal/env.py,sha256=EY_2KVe8oR3LgKWdaeRb5rRU-NDNXJppPDsFJmMZUUY,265
 snowflake/ml/_internal/env_utils.py,sha256=Xx03pV_qEIVJJY--J3ZmnqK9Ugf0Os3O2vrF8xOyq_c,31500
 snowflake/ml/_internal/file_utils.py,sha256=7sA6loOeSfmGP4yx16P4usT9ZtRqG3ycnXu7_Tk7dOs,14206
@@ -40,7 +40,7 @@ snowflake/ml/_internal/utils/formatting.py,sha256=PswZ6Xas7sx3Ok1MBLoH2o7nfXOxaJ
 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=X8D_bjVRnpcSCuJFjrA71KBJDRFXD_73tVu4VL9agpE,5441
-snowflake/ml/_internal/utils/mixins.py,sha256=0_jJN_-iNvLim0-GsJ4geqK4Ja91O-M527uWzj3vBtw,3511
+snowflake/ml/_internal/utils/mixins.py,sha256=FxpB5ZJBJk9Cv3dI_YiJDkBzMRel60aOD_xn67TFR5s,4708
 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
@@ -53,19 +53,19 @@ snowflake/ml/_internal/utils/table_manager.py,sha256=Wf3JXLUzdCiffKF9PJj7edHY7us
 snowflake/ml/_internal/utils/temp_file_utils.py,sha256=eHyyvxHfj4Z3FIS6VWgNyw5bFjNi5cSGYmY1hzyqzwY,1534
 snowflake/ml/_internal/utils/url.py,sha256=V3Y5zwNhJouy_cyLTa2rogg5nQZ-Ag-7Rmq-qPPEjmg,1219
 snowflake/ml/data/__init__.py,sha256=nm5VhN98Lzxr4kb679kglQfqbDbHhd9zYsnFJiQiThg,351
-snowflake/ml/data/data_connector.py,sha256=ZAgnXWEmOjR_3tremS2gNgLen7Rzrt1Z3ZtkXJrZtCo,14154
-snowflake/ml/data/data_ingestor.py,sha256=0TFc8qo4TZwdHMaBUBTZ7T8kkZfLGVmStvEx9KrXPHU,1165
+snowflake/ml/data/data_connector.py,sha256=rzSW-z6YR5RQZH8aj7gbjyHhdwecCYKbrkk6-SReE68,14174
+snowflake/ml/data/data_ingestor.py,sha256=Fxy1wuw0-6dWjJhBn9o5ZhzXKIpaaM5Y6Ji9XZGO5g0,1203
 snowflake/ml/data/data_source.py,sha256=HjBO1xqTyJfAvEAGESUIdke0KvSj5S5-FcI2D2zgejI,512
 snowflake/ml/data/ingestor_utils.py,sha256=JOv7Kvs0DNhsXUjl940ZULDkeTjIcePCfQ9aL_NteV0,2721
 snowflake/ml/data/torch_utils.py,sha256=1IgXiqxLgUh0yyNqchOSps5gLqmMOglSctoifjJIDFI,3591
-snowflake/ml/data/_internal/arrow_ingestor.py,sha256=8QSHNrFfvOmlMHxbF0Uexnw0KBpfvqXuVKbqjxiLOFk,15901
+snowflake/ml/data/_internal/arrow_ingestor.py,sha256=0YfgMGDqq72O28nrV_LOgH-lpM1oljDhyn--BdNYKG0,16025
 snowflake/ml/dataset/__init__.py,sha256=nESj7YEI2u90Oxyit_hKCQMWb7N1BlEM3Ho2Fm0MfHo,274
 snowflake/ml/dataset/dataset.py,sha256=Uo99ZfAIpY9LZ4_gMsQfY_SwUpPnbfkuEcViHmSV6HA,21067
 snowflake/ml/dataset/dataset_factory.py,sha256=Fym4ICK-B1j6Om4ENwWxEvryq3ZKoCslBSZDBenmjOo,1615
 snowflake/ml/dataset/dataset_metadata.py,sha256=lcNvugBkP8YEkGMQqaV8SlHs5mwUKsUS8GgaPGNm6wM,4145
 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_info.py,sha256=B47YwCfp7qr1ZNnvDXzjFuk69R4_Qa3a-nVs5gE1OXc,2592
 snowflake/ml/experiment/experiment_tracking.py,sha256=X4R4S6TjWkRB6F6RkPoFY4iOO44of4YMj_whcKoSayk,21841
 snowflake/ml/experiment/utils.py,sha256=5lanWEq6tgWnOMHCX4FnfBmpIQEIgH57Cz0YtpaAa2Y,830
 snowflake/ml/experiment/_client/artifact.py,sha256=R2WB4Y_kqv43BWLfXv8SEDINn1Bnevzgb-mH5LyvgGk,3035
@@ -114,31 +114,30 @@ 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=SnkgCta6_5G6Ljl-Nzctr4yavhHUSlNKN3je0ojp54E,20685
 snowflake/ml/jobs/__init__.py,sha256=JypKzxERpcn4yJ7FILA98Gl0sFDEGkAIQ35b1iSzaXg,741
-snowflake/ml/jobs/decorators.py,sha256=mQgdWvvCwD7q79cSFKZHKegXGh2j1u8WM64UD3lCKr4,3428
-snowflake/ml/jobs/job.py,sha256=wEeIk5RZhkaJ56Zb1cBxsTAUJIo4QyhpRWaZaiYBuGY,27697
-snowflake/ml/jobs/job_definition.py,sha256=f1lZoTdn1AulJTHstAN8yDd-8Cx4kv4A0KBm-TAxJq4,10377
+snowflake/ml/jobs/decorators.py,sha256=ugaVGOLTlYFyohYbErCyWU5Aqh2gwSlNAmn7Yr55dsE,3241
+snowflake/ml/jobs/job.py,sha256=t1kk2Il8NQcDsVphw38_V6Hha5c0ToO9sI6M8lws-NE,28275
+snowflake/ml/jobs/job_definition.py,sha256=qTB2T9UrZXCS5xaznDCsFHG4mdyFzymOPOInHbIRRwc,14822
 snowflake/ml/jobs/manager.py,sha256=heuOXEn7Y5Gb5s2y55GEx6i9mr-Z2tTL-b0rPGYv3ao,20469
 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/data_utils.py,sha256=Cw0Mpa_bs-Co_z3EVf93fLjlXE9U6gMfILrU8BHjj_k,4109
+snowflake/ml/jobs/_interop/dto_schema.py,sha256=QlFYvlarJNI_gALZgVdNej08oBy2E7dPRdpMQTrvoHc,4272
 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=I-RGWUDUc_FsKsq_mtczO7n93O_IzUvmEKXTznSJyK8,18413
+snowflake/ml/jobs/_interop/protocols.py,sha256=YNCIT_FzjFPfEGp78QlolIYBxo3IMbsSRo9tBfeB0Ic,23026
 snowflake/ml/jobs/_interop/results.py,sha256=nQ07XJ1BZEkPB4xa12pbGyaKqR8sWCoSzx0IKQlub4w,1714
-snowflake/ml/jobs/_interop/utils.py,sha256=TWFkUcAYmb-fpTwEL8idkk3XxlZ8vLz4X_gyS78PSi4,5552
+snowflake/ml/jobs/_interop/utils.py,sha256=WkScRL88GmDD-cm_xqSIhCelK-793ERfwY3Y8U8zMMY,8102
 snowflake/ml/jobs/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-snowflake/ml/jobs/_utils/constants.py,sha256=7KMhM-7KEursLjFBlj4xaQ4uGFJtjJ12rlx55hcBCQ0,4677
-snowflake/ml/jobs/_utils/feature_flags.py,sha256=dLWBVIjyB2vsa4Vtm7Yhty6DOi8Nn73_YSjuYf73Y7A,1669
-snowflake/ml/jobs/_utils/function_payload_utils.py,sha256=4LBaStMdhRxcqwRkwFje-WwiEKRWnBfkaOYouF3N3Kg,1308
-snowflake/ml/jobs/_utils/payload_utils.py,sha256=X9nG3kKlxr-2EfnLXhLwHdXoAhQS_MaTPfXPydlE_JU,33824
+snowflake/ml/jobs/_utils/arg_protocol.py,sha256=SP60NjjxFEvGBWftfSSj2lU0PIrDMBdGzZ6fgA_VZOE,89
+snowflake/ml/jobs/_utils/constants.py,sha256=rLT2gTT_iv2Cia0b_fERb5U0Lc7Zuuu67QP8csNeR8I,4785
+snowflake/ml/jobs/_utils/feature_flags.py,sha256=j1PvXFTc83ds6I1575X0xnfIz-qRlnUwna8ZhQN1pUw,4831
+snowflake/ml/jobs/_utils/payload_utils.py,sha256=fAe9CN_mbbqtgfaUe61nKZTTPbeSPy4G1EB6wQImNwA,32681
 snowflake/ml/jobs/_utils/query_helper.py,sha256=DxezZzftVT7WZzf0uzEn0l6U7BFLNU4U4W_IRCzgbaI,1265
-snowflake/ml/jobs/_utils/runtime_env_utils.py,sha256=EHESq95LhgKi1E68u9vlp8Brc4Ff8R0dFlIXAXoB1EA,4472
-snowflake/ml/jobs/_utils/spec_utils.py,sha256=eC24LFiHtHQiswrZy7m94jixIInJFXNZ_-Km-xr9NJg,871
+snowflake/ml/jobs/_utils/runtime_env_utils.py,sha256=l7IegZo_UYeQQEXs8gptP8UBEnikGcxP49Ocp0t2d0I,751
 snowflake/ml/jobs/_utils/stage_utils.py,sha256=vUWBj5nAsdcYe5Gtv_JhHYMbE5sKGFW6w7qh6uPtBUI,5912
 snowflake/ml/jobs/_utils/types.py,sha256=UJ_WjiHcMxR3DZb5LQm-bUs7i74tZi7DHXrWRyvqlpg,3154
 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=D2yheAux0GrIcxzD7IPEuUaRb4JyDdDpxRXXNlxszWQ,19912
+snowflake/ml/jobs/_utils/scripts/mljob_launcher.py,sha256=HuhAai6JL1Zdbo50Pg257Zeyfog90fJKVIvujmCASPw,26904
 snowflake/ml/jobs/_utils/scripts/signal_workers.py,sha256=AR1Pylkm4-FGh10WXfrCtcxaV0rI7IQ2ZiO0Li7zZ3U,7433
 snowflake/ml/jobs/_utils/scripts/start_mlruntime.sh,sha256=EIhhpEeu1Ph3J-QsAvugkhESkTMSZP3lWubmBCw5Z4w,3671
 snowflake/ml/jobs/_utils/scripts/startup.sh,sha256=OwJjQczoNwqf6v2vq6MeChNoa79NHfLeZG8VngRQMvQ,4148
@@ -152,7 +151,7 @@ snowflake/ml/model/custom_model.py,sha256=sdyKhT-QNNtTeu3idu6BExZNVyjUD4YTU8cru3
 snowflake/ml/model/event_handler.py,sha256=pojleQVM9TPNeDvliTvon2Sfxqbf2WWxrOebo1SaEHo,7211
 snowflake/ml/model/inference_engine.py,sha256=L0nwySY2Qwp3JzuRpPS87r0--m3HTUNUgZXYyOPJjyk,66
 snowflake/ml/model/model_signature.py,sha256=ae1tkh3Rw9MzJSxmVT9kb0PwD3TANtKbWwp6b8-cItE,32847
-snowflake/ml/model/openai_signatures.py,sha256=wu7l9V6-fWNJoAdE3R25hYBAmVfND5oaWdsUXilxBDo,7329
+snowflake/ml/model/openai_signatures.py,sha256=ny3aAd301t3h_4Pb60MkrY0LZO6QxpyIlHYnjv120VQ,14755
 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=Xxa6b9ezbvXYvSIN5R4Zv6Dro4ZH74-eW4cno92VTJE,11475
@@ -160,15 +159,15 @@ snowflake/ml/model/volatility.py,sha256=qu-wqe9oKkRwXwE2qkKygxTWzUypQYEk3UjsqOGR
 snowflake/ml/model/_client/model/batch_inference_specs.py,sha256=6q3XzYWaO1CH-JqonJr12n6NuRhQDvPns_FAq3yvrN4,6114
 snowflake/ml/model/_client/model/inference_engine_utils.py,sha256=yPkdImi2qP1uG1WzLKCBZgXV-DiIBVpImEosIjYJk8Y,1958
 snowflake/ml/model/_client/model/model_impl.py,sha256=Yabrbir5vPMOnsVmQJ23YN7vqhi756Jcm6pfO8Aq92o,17469
-snowflake/ml/model/_client/model/model_version_impl.py,sha256=kJ9tG_MQJzmIXss4K7IISoB7uO9rkjcH8kef5ypBy2s,69613
+snowflake/ml/model/_client/model/model_version_impl.py,sha256=3R39U2xAWJdvZktJ_h1sAaxHmum7wd1dQP1dJE5PmbE,70403
 snowflake/ml/model/_client/ops/deployment_step.py,sha256=9kxKDr9xcD4KmVM-9O4_tm3ytkllQVoElJD793VI84Q,1428
 snowflake/ml/model/_client/ops/metadata_ops.py,sha256=qpK6PL3OyfuhyOmpvLCpHLy6vCxbZbp1HlEvakFGwv4,4884
 snowflake/ml/model/_client/ops/model_ops.py,sha256=s5-N9RlaJzndh0D1sQa4BnYb0N-FKYgx86O0d2H5mKg,53819
 snowflake/ml/model/_client/ops/param_utils.py,sha256=MPPerO8wYNYIuig4rFoG_YI5idD_dBlrWXgmAXcR2nM,5160
-snowflake/ml/model/_client/ops/service_ops.py,sha256=ZPZDJFEAaiQPS1Rni_cZ-mWCE_gxHixTV6cWmoTZZBE,48049
+snowflake/ml/model/_client/ops/service_ops.py,sha256=9eWK3R_iOCkqafR_bK2OpAy0fdpjqYAK447ozZZF_Lg,47910
 snowflake/ml/model/_client/service/import_model_spec_schema.py,sha256=SlEX1GiPlB8whMCmiwKUopnrGlm4fkQOQbTW2KyVTFU,554
-snowflake/ml/model/_client/service/model_deployment_spec.py,sha256=Qg7cR9jQTnbWHIV5FyxUZocxWH0nFEem9wdOt5osKIw,20149
-snowflake/ml/model/_client/service/model_deployment_spec_schema.py,sha256=2FCfhyg7F2XYVWKFybeEQ9Fq3goZ3XKmJdH_CuRqPpI,2649
+snowflake/ml/model/_client/service/model_deployment_spec.py,sha256=pRwwlD9h6V76fAOtFR1XgxXEv1fQjfdJCxBZgjgocE0,20286
+snowflake/ml/model/_client/service/model_deployment_spec_schema.py,sha256=axEX0gnUqnLav575xzpJ9lDWNwr9vyjyay7soo7Hwgk,2672
 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=SOYr13YEq0mxgIatsSchOq0aKUgdPhKO3clRQ6AMa7U,24766
@@ -230,7 +229,7 @@ snowflake/ml/model/_signatures/snowpark_handler.py,sha256=aNGPa2v0kTMuSZ80NBdHeA
 snowflake/ml/model/_signatures/tensorflow_handler.py,sha256=_yrvMg-w_jJoYuyrGXKPX4Dv7Vt8z1e6xIKiWGuZcc4,5660
 snowflake/ml/model/_signatures/utils.py,sha256=hoc_UuMxfPWkVmoMEE7U-XNRcSgDPyRIdyDVK0JLcfE,21685
 snowflake/ml/model/models/huggingface.py,sha256=VO84lBizmSALntWCnK4O_eY_Cq2uzMooyHtfJXuFkew,13791
-snowflake/ml/model/models/huggingface_pipeline.py,sha256=yJ7NW97EW5GRtOPYCYuNjoPIgXbAIyOEKW9P1trt9vY,14226
+snowflake/ml/model/models/huggingface_pipeline.py,sha256=tLTj1YJumNs_pdHGmA1Us5HIi8EYm38X6jqv2j85TMo,14378
 snowflake/ml/modeling/_internal/estimator_utils.py,sha256=dfPPWO-RHf5C3Tya3VQ4KEqoa32pm-WKwRrjzjDInLk,13956
 snowflake/ml/modeling/_internal/model_specifications.py,sha256=3wFMcKPCSoiEzU7Mx6RVem89BRlBBENpX__-Rd7GwdU,4851
 snowflake/ml/modeling/_internal/model_trainer.py,sha256=5Ck1lbdyzcd-TpzAxEyovIN9fjaaVIqugyMHXt0wzH0,971
@@ -459,14 +458,14 @@ snowflake/ml/monitoring/entities/model_monitor_config.py,sha256=auy9BD0IoyUpZPZX
 snowflake/ml/registry/__init__.py,sha256=XdPQK9ejYkSJVrSQ7HD3jKQO0hKq2mC4bPCB6qrtH3U,76
 snowflake/ml/registry/registry.py,sha256=_vtQCh4DmhnPusTKWJteRPJkDpLFEfG150cjED70sOA,34611
 snowflake/ml/registry/_manager/model_manager.py,sha256=splK5YGErt-eDIy6UbZAB3VKsGMZSJk2_MzfgrIQOhY,26306
-snowflake/ml/registry/_manager/model_parameter_reconciler.py,sha256=BTi8WNQCW1TiW8jd9LbR6mrIHE29ehocKSh6mVQX0vI,15325
+snowflake/ml/registry/_manager/model_parameter_reconciler.py,sha256=92YX67YMLmqbJN_emr49c3QWyFVGc1DUn6JvA3Y2qqo,15267
 snowflake/ml/utils/authentication.py,sha256=TQV3E8YDHAPXA3dS8JWDmb_Zm8P0d9c8kCexRI4nefo,3106
 snowflake/ml/utils/connection_params.py,sha256=NSBUgcs-DXPRHs1BKpxdSubbJx1yrFRlMPBp-bE3Ugc,8308
 snowflake/ml/utils/html_utils.py,sha256=4g1EPuD8EnOAK7BCYiY8Wp3ZrdDkNOcUDrDAbUYxLfs,9954
 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.24.0.dist-info/licenses/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
-snowflake_ml_python-1.24.0.dist-info/METADATA,sha256=sR-h6xi0sM7Z04GM-Jd3clq-_jE_I9fC-3VyPcUB8r0,105348
-snowflake_ml_python-1.24.0.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-snowflake_ml_python-1.24.0.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
-snowflake_ml_python-1.24.0.dist-info/RECORD,,
+snowflake_ml_python-1.25.0.dist-info/licenses/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
+snowflake_ml_python-1.25.0.dist-info/METADATA,sha256=hV6uhSQvACNGp9jEdujaC9Y_F5r5rf1eBI22Hh6FBe0,107037
+snowflake_ml_python-1.25.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+snowflake_ml_python-1.25.0.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
+snowflake_ml_python-1.25.0.dist-info/RECORD,,

{snowflake_ml_python-1.24.0.dist-info → snowflake_ml_python-1.25.0.dist-info}/WHEEL

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

snowflake/ml/jobs/_utils/function_payload_utils.py

@@ -1,43 +0,0 @@
-import inspect
-from typing import Any, Callable, Optional
-
-from snowflake import snowpark
-from snowflake.snowpark import context as sp_context
-
-
-class FunctionPayload:
-    def __init__(
-        self,
-        func: Callable[..., Any],
-        session: Optional[snowpark.Session] = None,
-        session_argument: str = "",
-        *args: Any,
-        **kwargs: Any
-    ) -> None:
-        self.function = func
-        self.args = args
-        self.kwargs = kwargs
-        self._session = session
-        self._session_argument = session_argument
-
-    @property
-    def session(self) -> Optional[snowpark.Session]:
-        return self._session
-
-    def __getstate__(self) -> dict[str, Any]:
-        """Customize pickling to exclude session."""
-        state = self.__dict__.copy()
-        state["_session"] = None
-        return state
-
-    def __setstate__(self, state: dict[str, Any]) -> None:
-        """Restore session from context during unpickling."""
-        self.__dict__.update(state)
-        self._session = sp_context.get_active_session()
-
-    def __call__(self) -> Any:
-        sig = inspect.signature(self.function)
-        bound = sig.bind_partial(*self.args, **self.kwargs)
-        bound.arguments[self._session_argument] = self._session
-
-        return self.function(*bound.args, **bound.kwargs)

snowflake/ml/jobs/_utils/spec_utils.py

@@ -1,22 +0,0 @@
-from snowflake import snowpark
-from snowflake.ml._internal.utils import snowflake_env
-from snowflake.ml.jobs._utils import constants, query_helper, types
-
-
-def _get_node_resources(session: snowpark.Session, compute_pool: str) -> types.ComputeResources:
-    """Extract resource information for the specified compute pool"""
-    # Get the instance family
-    rows = query_helper.run_query(
-        session,
-        "show compute pools like ?",
-        params=[compute_pool],
-    )
-    if not rows:
-        raise ValueError(f"Compute pool '{compute_pool}' not found")
-    instance_family: str = rows[0]["instance_family"]
-    cloud = snowflake_env.get_current_cloud(session, default=snowflake_env.SnowflakeCloudType.AWS)
-
-    return (
-        constants.COMMON_INSTANCE_FAMILIES.get(instance_family)
-        or constants.CLOUD_INSTANCE_FAMILIES[cloud][instance_family]
-    )