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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (49) hide show
  1. snowflake/ml/_internal/platform_capabilities.py +0 -4
  2. snowflake/ml/_internal/utils/mixins.py +26 -1
  3. snowflake/ml/data/_internal/arrow_ingestor.py +5 -1
  4. snowflake/ml/data/data_connector.py +2 -2
  5. snowflake/ml/data/data_ingestor.py +2 -1
  6. snowflake/ml/experiment/_experiment_info.py +3 -3
  7. snowflake/ml/feature_store/__init__.py +2 -0
  8. snowflake/ml/feature_store/aggregation.py +367 -0
  9. snowflake/ml/feature_store/feature.py +366 -0
  10. snowflake/ml/feature_store/feature_store.py +234 -20
  11. snowflake/ml/feature_store/feature_view.py +189 -4
  12. snowflake/ml/feature_store/metadata_manager.py +425 -0
  13. snowflake/ml/feature_store/tile_sql_generator.py +1079 -0
  14. snowflake/ml/jobs/_interop/data_utils.py +8 -8
  15. snowflake/ml/jobs/_interop/dto_schema.py +52 -7
  16. snowflake/ml/jobs/_interop/protocols.py +124 -7
  17. snowflake/ml/jobs/_interop/utils.py +92 -33
  18. snowflake/ml/jobs/_utils/arg_protocol.py +7 -0
  19. snowflake/ml/jobs/_utils/constants.py +4 -0
  20. snowflake/ml/jobs/_utils/feature_flags.py +97 -13
  21. snowflake/ml/jobs/_utils/payload_utils.py +6 -40
  22. snowflake/ml/jobs/_utils/runtime_env_utils.py +12 -111
  23. snowflake/ml/jobs/_utils/scripts/mljob_launcher.py +204 -27
  24. snowflake/ml/jobs/decorators.py +17 -22
  25. snowflake/ml/jobs/job.py +25 -10
  26. snowflake/ml/jobs/job_definition.py +100 -8
  27. snowflake/ml/model/__init__.py +4 -0
  28. snowflake/ml/model/_client/model/batch_inference_specs.py +38 -2
  29. snowflake/ml/model/_client/model/model_version_impl.py +56 -28
  30. snowflake/ml/model/_client/ops/model_ops.py +2 -8
  31. snowflake/ml/model/_client/ops/service_ops.py +6 -11
  32. snowflake/ml/model/_client/service/model_deployment_spec.py +3 -0
  33. snowflake/ml/model/_client/service/model_deployment_spec_schema.py +1 -0
  34. snowflake/ml/model/_client/sql/service.py +21 -29
  35. snowflake/ml/model/_model_composer/model_method/model_method.py +2 -1
  36. snowflake/ml/model/_packager/model_handlers/huggingface.py +20 -0
  37. snowflake/ml/model/_packager/model_handlers/sentence_transformers.py +70 -14
  38. snowflake/ml/model/_signatures/utils.py +76 -1
  39. snowflake/ml/model/models/huggingface_pipeline.py +3 -0
  40. snowflake/ml/model/openai_signatures.py +154 -0
  41. snowflake/ml/registry/_manager/model_parameter_reconciler.py +2 -3
  42. snowflake/ml/version.py +1 -1
  43. {snowflake_ml_python-1.23.0.dist-info → snowflake_ml_python-1.25.0.dist-info}/METADATA +79 -2
  44. {snowflake_ml_python-1.23.0.dist-info → snowflake_ml_python-1.25.0.dist-info}/RECORD +47 -44
  45. {snowflake_ml_python-1.23.0.dist-info → snowflake_ml_python-1.25.0.dist-info}/WHEEL +1 -1
  46. snowflake/ml/jobs/_utils/function_payload_utils.py +0 -43
  47. snowflake/ml/jobs/_utils/spec_utils.py +0 -22
  48. {snowflake_ml_python-1.23.0.dist-info → snowflake_ml_python-1.25.0.dist-info}/licenses/LICENSE.txt +0 -0
  49. {snowflake_ml_python-1.23.0.dist-info → snowflake_ml_python-1.25.0.dist-info}/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: snowflake-ml-python
3
- Version: 1.23.0
3
+ Version: 1.25.0
4
4
  Summary: The machine learning client library that is used for interacting with Snowflake to build machine learning solutions.
5
5
  Author-email: "Snowflake, Inc" <support@snowflake.com>
6
6
  License:
@@ -417,6 +417,75 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
417
417
 
418
418
  # Release History
419
419
 
420
+ ## 1.25.0
421
+
422
+ ### New Features
423
+
424
+ * ML Job: Added support for creating ML job definitions and launching jobs with different
425
+ arguments without re-uploading payloads.
426
+
427
+ * Inference Autocapture (PuPr): The `create_service` API will now accept `autocapture` as a new argument to indicate
428
+ whether inference data will be captured.
429
+
430
+ * Model serving: Introduced the `min_instances` field in the `mv.create_service()` and
431
+ `HuggingFacePipelineModel.log_model_and_create_service()` APIs (defaulting to 0). The service now launches
432
+ with the `min_instances` and automatically scales between `min_instances` and `max_instances` based on
433
+ traffic and hardware utilization. When `min_instances` is set to 0, the service will automatically suspend
434
+ if no traffic is detected for a period of time.
435
+
436
+ ### Bug Fixes
437
+
438
+ ### Behavior Changes
439
+
440
+ * Inference Autocapture (PuPr): `list_services()` now shows `autocapture_enabled` column to indicate if model
441
+ service has autocapture enabled.
442
+
443
+ * Model serving: The `mv.create_service()` and `HuggingFacePipelineModel.log_model_and_create_service()` APIs now
444
+ include a `min_instances` field (defaulting to 0). When these APIs are called without specifying `min_instances`,
445
+ the system will now launch the service with 1 instance and enable auto scaling. This replaces the previous behavior,
446
+ where `min_instances` was automatically set to match `max_instances`, resulting in the immediate launch of the
447
+ maximum number of instances.
448
+
449
+ ### Deprecations
450
+
451
+ ## 1.24.0
452
+
453
+ ### New Features
454
+
455
+ * Feature Store: Added tile-based aggregation support with a new `Feature` API for efficient and
456
+ point-in-time correct time-series feature computation using pre-computed tiles stored in Dynamic Tables.
457
+
458
+ * Registry: Added auto-signature inference for SentenceTransformer models. When logging a SentenceTransformer
459
+ model, `sample_input_data` is now optional. If not provided, the signature is automatically inferred from
460
+ the model's embedding dimension. Supported methods: `encode`, `encode_query`, `encode_document`,
461
+ `encode_queries`, `encode_documents`.
462
+
463
+ ```python
464
+ import sentence_transformers
465
+ from snowflake.ml.registry import Registry
466
+
467
+ # Create model
468
+ model = sentence_transformers.SentenceTransformer("all-MiniLM-L6-v2")
469
+
470
+ # Log model without sample_input_data - signature is auto-inferred
471
+ registry = Registry(session)
472
+ mv = registry.log_model(
473
+ model=model,
474
+ model_name="my_sentence_transformer",
475
+ version_name="v1",
476
+ )
477
+
478
+ # Run inference with auto-inferred signature (input: "text", output: "output")
479
+ import pandas as pd
480
+ result = mv.run(pd.DataFrame({"text": ["Hello world"]}))
481
+ ```
482
+
483
+ ### Bug Fixes
484
+
485
+ ### Behavior Changes
486
+
487
+ ### Deprecations
488
+
420
489
  ## 1.23.0
421
490
 
422
491
  ### New Features
@@ -482,7 +551,15 @@ x_df = pd.DataFrame.from_records(
482
551
  [
483
552
  {
484
553
  "messages": [
485
- {"role": "system", "content": "Complete the sentence."},
554
+ {
555
+ "role": "system",
556
+ "content": [
557
+ {
558
+ "type": "text",
559
+ "text": "Complete the sentence."
560
+ },
561
+ ]
562
+ },
486
563
  {
487
564
  "role": "user",
488
565
  "content": [
@@ -10,13 +10,13 @@ snowflake/cortex/_sse_client.py,sha256=sLYgqAfTOPADCnaWH2RWAJi8KbU_7gSRsTUDcDD5T
10
10
  snowflake/cortex/_summarize.py,sha256=7GH8zqfIdOiHA5w4b6EvJEKEWhaTrL4YA6iDGbn7BNM,1307
11
11
  snowflake/cortex/_translate.py,sha256=9ZGjvAnJFisbzJ_bXnt4pyug5UzhHJRXW8AhGQEersM,1652
12
12
  snowflake/cortex/_util.py,sha256=krNTpbkFLXwdFqy1bd0xi7ZmOzOHRnIfHdQCPiLZJxk,3288
13
- snowflake/ml/version.py,sha256=4kkULHrWTWUklPOTCjLofNqUa56UDa5-aT7xzkSZmTQ,99
13
+ snowflake/ml/version.py,sha256=V1sQKmkr-QMsegsFN8N1EoPDeU1CzZe6ofRecCIByco,99
14
14
  snowflake/ml/_internal/env.py,sha256=EY_2KVe8oR3LgKWdaeRb5rRU-NDNXJppPDsFJmMZUUY,265
15
15
  snowflake/ml/_internal/env_utils.py,sha256=Xx03pV_qEIVJJY--J3ZmnqK9Ugf0Os3O2vrF8xOyq_c,31500
16
16
  snowflake/ml/_internal/file_utils.py,sha256=7sA6loOeSfmGP4yx16P4usT9ZtRqG3ycnXu7_Tk7dOs,14206
17
17
  snowflake/ml/_internal/init_utils.py,sha256=WhrlvS-xcmKErSpwg6cUk6XDQ5lQcwDqPJnU7cooMIg,2672
18
18
  snowflake/ml/_internal/migrator_utils.py,sha256=k3erO8x3YJcX6nkKeyJAUNGg1qjE3RFmD-W6dtLzIH0,161
19
- snowflake/ml/_internal/platform_capabilities.py,sha256=qKCNbZUSDEjn_Fi5H9GJzFk_8CuWkSXYkFxug1mY0j0,9274
19
+ snowflake/ml/_internal/platform_capabilities.py,sha256=MdTeUqT7URSkE_PJoMGFOsi_DzeFDzAVBJJ2-NqoINY,9064
20
20
  snowflake/ml/_internal/relax_version_strategy.py,sha256=MYEIZrx1HfKNhl9Na3GN50ipX8c0MKIj9nwxjB0IC0Y,484
21
21
  snowflake/ml/_internal/telemetry.py,sha256=oN0NDuKYabmc85IGPDz48ktFpPlo9Xlhw0qGJdOtFNY,33684
22
22
  snowflake/ml/_internal/type_utils.py,sha256=bNNW0I9rOvwhx-Y274vGd0qWA0fMIPA3SGnaDE09wvc,2198
@@ -40,7 +40,7 @@ snowflake/ml/_internal/utils/formatting.py,sha256=PswZ6Xas7sx3Ok1MBLoH2o7nfXOxaJ
40
40
  snowflake/ml/_internal/utils/identifier.py,sha256=HrcCBOyn93fRjMj4K1YJG37ONtw7e3EZnt29LzhEgLA,12586
41
41
  snowflake/ml/_internal/utils/import_utils.py,sha256=msvUDaCcJpAcNCS-5Ynz4F1CvUhXjRsuZyOv1rN6Yhk,3213
42
42
  snowflake/ml/_internal/utils/jwt_generator.py,sha256=X8D_bjVRnpcSCuJFjrA71KBJDRFXD_73tVu4VL9agpE,5441
43
- snowflake/ml/_internal/utils/mixins.py,sha256=0_jJN_-iNvLim0-GsJ4geqK4Ja91O-M527uWzj3vBtw,3511
43
+ snowflake/ml/_internal/utils/mixins.py,sha256=FxpB5ZJBJk9Cv3dI_YiJDkBzMRel60aOD_xn67TFR5s,4708
44
44
  snowflake/ml/_internal/utils/parallelize.py,sha256=l8Zjo-hp8zqoLgHxBlpz9Zmn2Z-MRQ0fS_NnrR4jWR8,4522
45
45
  snowflake/ml/_internal/utils/pkg_version_utils.py,sha256=EaY_3IsVOZ9BCH28F5VLjp-0AiEqDlL7L715vkPsgrY,5149
46
46
  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
53
53
  snowflake/ml/_internal/utils/temp_file_utils.py,sha256=eHyyvxHfj4Z3FIS6VWgNyw5bFjNi5cSGYmY1hzyqzwY,1534
54
54
  snowflake/ml/_internal/utils/url.py,sha256=V3Y5zwNhJouy_cyLTa2rogg5nQZ-Ag-7Rmq-qPPEjmg,1219
55
55
  snowflake/ml/data/__init__.py,sha256=nm5VhN98Lzxr4kb679kglQfqbDbHhd9zYsnFJiQiThg,351
56
- snowflake/ml/data/data_connector.py,sha256=ZAgnXWEmOjR_3tremS2gNgLen7Rzrt1Z3ZtkXJrZtCo,14154
57
- snowflake/ml/data/data_ingestor.py,sha256=0TFc8qo4TZwdHMaBUBTZ7T8kkZfLGVmStvEx9KrXPHU,1165
56
+ snowflake/ml/data/data_connector.py,sha256=rzSW-z6YR5RQZH8aj7gbjyHhdwecCYKbrkk6-SReE68,14174
57
+ snowflake/ml/data/data_ingestor.py,sha256=Fxy1wuw0-6dWjJhBn9o5ZhzXKIpaaM5Y6Ji9XZGO5g0,1203
58
58
  snowflake/ml/data/data_source.py,sha256=HjBO1xqTyJfAvEAGESUIdke0KvSj5S5-FcI2D2zgejI,512
59
59
  snowflake/ml/data/ingestor_utils.py,sha256=JOv7Kvs0DNhsXUjl940ZULDkeTjIcePCfQ9aL_NteV0,2721
60
60
  snowflake/ml/data/torch_utils.py,sha256=1IgXiqxLgUh0yyNqchOSps5gLqmMOglSctoifjJIDFI,3591
61
- snowflake/ml/data/_internal/arrow_ingestor.py,sha256=8QSHNrFfvOmlMHxbF0Uexnw0KBpfvqXuVKbqjxiLOFk,15901
61
+ snowflake/ml/data/_internal/arrow_ingestor.py,sha256=0YfgMGDqq72O28nrV_LOgH-lpM1oljDhyn--BdNYKG0,16025
62
62
  snowflake/ml/dataset/__init__.py,sha256=nESj7YEI2u90Oxyit_hKCQMWb7N1BlEM3Ho2Fm0MfHo,274
63
63
  snowflake/ml/dataset/dataset.py,sha256=Uo99ZfAIpY9LZ4_gMsQfY_SwUpPnbfkuEcViHmSV6HA,21067
64
64
  snowflake/ml/dataset/dataset_factory.py,sha256=Fym4ICK-B1j6Om4ENwWxEvryq3ZKoCslBSZDBenmjOo,1615
65
65
  snowflake/ml/dataset/dataset_metadata.py,sha256=lcNvugBkP8YEkGMQqaV8SlHs5mwUKsUS8GgaPGNm6wM,4145
66
66
  snowflake/ml/dataset/dataset_reader.py,sha256=mZsG9HyWUGgfotrGkLrunyEsOm_659mH-Sn2OyG6A-Q,5036
67
67
  snowflake/ml/experiment/__init__.py,sha256=r7qdyPd3jwxzqvksim2ju5j_LrnYQrta0ZI6XpWUqmc,109
68
- snowflake/ml/experiment/_experiment_info.py,sha256=iaJ65x6nzBYJ5djleSOzBtMpZUJCUDlRpaDw0pu-dcU,2533
68
+ snowflake/ml/experiment/_experiment_info.py,sha256=B47YwCfp7qr1ZNnvDXzjFuk69R4_Qa3a-nVs5gE1OXc,2592
69
69
  snowflake/ml/experiment/experiment_tracking.py,sha256=X4R4S6TjWkRB6F6RkPoFY4iOO44of4YMj_whcKoSayk,21841
70
70
  snowflake/ml/experiment/utils.py,sha256=5lanWEq6tgWnOMHCX4FnfBmpIQEIgH57Cz0YtpaAa2Y,830
71
71
  snowflake/ml/experiment/_client/artifact.py,sha256=R2WB4Y_kqv43BWLfXv8SEDINn1Bnevzgb-mH5LyvgGk,3035
@@ -77,11 +77,15 @@ snowflake/ml/experiment/_entities/run_metadata.py,sha256=25cIg8FnAYHk5SoTg_StzL1
77
77
  snowflake/ml/experiment/callback/keras.py,sha256=2kNYVWJjBKn9i2pioiGthLNjclNdSHn-6xHm6tlDc4E,4122
78
78
  snowflake/ml/experiment/callback/lightgbm.py,sha256=dftlOMT9Wxo-XPaSvgbMh5IwVfQEskz4Ev4jd4G3_ns,4249
79
79
  snowflake/ml/experiment/callback/xgboost.py,sha256=VLO3uW0Pot1b4nRaoi8tjLtDt3hrkTV0_6cqja3SqZQ,4197
80
- snowflake/ml/feature_store/__init__.py,sha256=MJr2Gp_EimDgDxD6DtenOEdLTzg6NYPfdNiPM-5rEtw,406
80
+ snowflake/ml/feature_store/__init__.py,sha256=QajLPBszCu4QRvpfhP_Fw8OfRuXMrB-IEug-SJrLV-I,450
81
81
  snowflake/ml/feature_store/access_manager.py,sha256=ZuLk2IQE2H-XSV96Z6mf_KzF2J-kjaDf-_t-0nCxnTg,10724
82
+ snowflake/ml/feature_store/aggregation.py,sha256=l-JzcNI3ftKDs_EgdYPMxDLJzsgRtOuwoeND76t3xnA,13049
82
83
  snowflake/ml/feature_store/entity.py,sha256=ViOSlqCV17ouiO4iH-_KvkvJZqSzpf-nfsjijG6G1Uk,4047
83
- snowflake/ml/feature_store/feature_store.py,sha256=9rb-CT1F-gI1hb8JzqD9CBRz2Q9Vd3c2HtRbl7pSdu4,172334
84
- snowflake/ml/feature_store/feature_view.py,sha256=uQ57uF0IhFj_FzOaqMvwlar6Kl-CBOhjh5NSE0ZFgMo,44439
84
+ snowflake/ml/feature_store/feature.py,sha256=7jJbhDzJg6Y5LKW7ON7itRxnsM3UQ2QSCLW0W2X2hok,12800
85
+ snowflake/ml/feature_store/feature_store.py,sha256=APGpfayuTrycgZmm4zg2yK4AfVQsr9UCsV_DrR4Htdg,182598
86
+ snowflake/ml/feature_store/feature_view.py,sha256=tNjqEfXc5U3VHAdNlpCyQrVBsPP3GPLPzxbXTkd8Imk,53433
87
+ snowflake/ml/feature_store/metadata_manager.py,sha256=Z8OGbT8bVpLs7BlNEAw76RD5r3pW5hcxKcn9I9dlDHs,14716
88
+ snowflake/ml/feature_store/tile_sql_generator.py,sha256=Sik6nhFLOqImiCmwcw3sJ1RktlUJNGmYJb9T2q-MqeQ,48217
85
89
  snowflake/ml/feature_store/examples/example_helper.py,sha256=eaD2vLe7y4C5hMZQTeMXylbTtLacbq9gJcAluGHrkug,12470
86
90
  snowflake/ml/feature_store/examples/airline_features/entities.py,sha256=mzHRS-InHpXON0eHds-QLmi7nK9ciOnCruhPZI4niLs,438
87
91
  snowflake/ml/feature_store/examples/airline_features/source.yaml,sha256=kzl8ukOK8OuSPsxChEgJ9SPyPnzC-fPHqZC4O6aqd5o,247
@@ -110,65 +114,64 @@ snowflake/ml/fileset/sfcfs.py,sha256=FJFc9-gc0KXaNyc10ZovN_87aUCShb0WztVwa02t0io
110
114
  snowflake/ml/fileset/snowfs.py,sha256=uF5QluYtiJ-HezGIhF55dONi3t0E6N7ByaVAIAlM3nk,5133
111
115
  snowflake/ml/fileset/stage_fs.py,sha256=SnkgCta6_5G6Ljl-Nzctr4yavhHUSlNKN3je0ojp54E,20685
112
116
  snowflake/ml/jobs/__init__.py,sha256=JypKzxERpcn4yJ7FILA98Gl0sFDEGkAIQ35b1iSzaXg,741
113
- snowflake/ml/jobs/decorators.py,sha256=mQgdWvvCwD7q79cSFKZHKegXGh2j1u8WM64UD3lCKr4,3428
114
- snowflake/ml/jobs/job.py,sha256=wEeIk5RZhkaJ56Zb1cBxsTAUJIo4QyhpRWaZaiYBuGY,27697
115
- snowflake/ml/jobs/job_definition.py,sha256=f1lZoTdn1AulJTHstAN8yDd-8Cx4kv4A0KBm-TAxJq4,10377
117
+ snowflake/ml/jobs/decorators.py,sha256=ugaVGOLTlYFyohYbErCyWU5Aqh2gwSlNAmn7Yr55dsE,3241
118
+ snowflake/ml/jobs/job.py,sha256=t1kk2Il8NQcDsVphw38_V6Hha5c0ToO9sI6M8lws-NE,28275
119
+ snowflake/ml/jobs/job_definition.py,sha256=qTB2T9UrZXCS5xaznDCsFHG4mdyFzymOPOInHbIRRwc,14822
116
120
  snowflake/ml/jobs/manager.py,sha256=heuOXEn7Y5Gb5s2y55GEx6i9mr-Z2tTL-b0rPGYv3ao,20469
117
121
  snowflake/ml/jobs/_interop/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
118
- snowflake/ml/jobs/_interop/data_utils.py,sha256=xUO5YlhUKFVCDtbjam5gP2lka3lfoknTLr7syNAVxK0,4074
119
- snowflake/ml/jobs/_interop/dto_schema.py,sha256=NhoQ6WJa7uLO9VJojEENVVZhZMfL_G1VPPSSUYmmhO8,2750
122
+ snowflake/ml/jobs/_interop/data_utils.py,sha256=Cw0Mpa_bs-Co_z3EVf93fLjlXE9U6gMfILrU8BHjj_k,4109
123
+ snowflake/ml/jobs/_interop/dto_schema.py,sha256=QlFYvlarJNI_gALZgVdNej08oBy2E7dPRdpMQTrvoHc,4272
120
124
  snowflake/ml/jobs/_interop/exception_utils.py,sha256=ZCphBkaaNINFATXZmjPzzNLKZZxKvGPROZ2azU8w13g,16348
121
125
  snowflake/ml/jobs/_interop/legacy.py,sha256=8BuC197e6nPmAzh4urYiuBuCNP-RlOlvWnWpSHTduqQ,9238
122
- snowflake/ml/jobs/_interop/protocols.py,sha256=I-RGWUDUc_FsKsq_mtczO7n93O_IzUvmEKXTznSJyK8,18413
126
+ snowflake/ml/jobs/_interop/protocols.py,sha256=YNCIT_FzjFPfEGp78QlolIYBxo3IMbsSRo9tBfeB0Ic,23026
123
127
  snowflake/ml/jobs/_interop/results.py,sha256=nQ07XJ1BZEkPB4xa12pbGyaKqR8sWCoSzx0IKQlub4w,1714
124
- snowflake/ml/jobs/_interop/utils.py,sha256=TWFkUcAYmb-fpTwEL8idkk3XxlZ8vLz4X_gyS78PSi4,5552
128
+ snowflake/ml/jobs/_interop/utils.py,sha256=WkScRL88GmDD-cm_xqSIhCelK-793ERfwY3Y8U8zMMY,8102
125
129
  snowflake/ml/jobs/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
126
- snowflake/ml/jobs/_utils/constants.py,sha256=7KMhM-7KEursLjFBlj4xaQ4uGFJtjJ12rlx55hcBCQ0,4677
127
- snowflake/ml/jobs/_utils/feature_flags.py,sha256=dLWBVIjyB2vsa4Vtm7Yhty6DOi8Nn73_YSjuYf73Y7A,1669
128
- snowflake/ml/jobs/_utils/function_payload_utils.py,sha256=4LBaStMdhRxcqwRkwFje-WwiEKRWnBfkaOYouF3N3Kg,1308
129
- snowflake/ml/jobs/_utils/payload_utils.py,sha256=X9nG3kKlxr-2EfnLXhLwHdXoAhQS_MaTPfXPydlE_JU,33824
130
+ snowflake/ml/jobs/_utils/arg_protocol.py,sha256=SP60NjjxFEvGBWftfSSj2lU0PIrDMBdGzZ6fgA_VZOE,89
131
+ snowflake/ml/jobs/_utils/constants.py,sha256=rLT2gTT_iv2Cia0b_fERb5U0Lc7Zuuu67QP8csNeR8I,4785
132
+ snowflake/ml/jobs/_utils/feature_flags.py,sha256=j1PvXFTc83ds6I1575X0xnfIz-qRlnUwna8ZhQN1pUw,4831
133
+ snowflake/ml/jobs/_utils/payload_utils.py,sha256=fAe9CN_mbbqtgfaUe61nKZTTPbeSPy4G1EB6wQImNwA,32681
130
134
  snowflake/ml/jobs/_utils/query_helper.py,sha256=DxezZzftVT7WZzf0uzEn0l6U7BFLNU4U4W_IRCzgbaI,1265
131
- snowflake/ml/jobs/_utils/runtime_env_utils.py,sha256=EHESq95LhgKi1E68u9vlp8Brc4Ff8R0dFlIXAXoB1EA,4472
132
- snowflake/ml/jobs/_utils/spec_utils.py,sha256=eC24LFiHtHQiswrZy7m94jixIInJFXNZ_-Km-xr9NJg,871
135
+ snowflake/ml/jobs/_utils/runtime_env_utils.py,sha256=l7IegZo_UYeQQEXs8gptP8UBEnikGcxP49Ocp0t2d0I,751
133
136
  snowflake/ml/jobs/_utils/stage_utils.py,sha256=vUWBj5nAsdcYe5Gtv_JhHYMbE5sKGFW6w7qh6uPtBUI,5912
134
137
  snowflake/ml/jobs/_utils/types.py,sha256=UJ_WjiHcMxR3DZb5LQm-bUs7i74tZi7DHXrWRyvqlpg,3154
135
138
  snowflake/ml/jobs/_utils/scripts/constants.py,sha256=YyIWZqQPYOTtgCY6SfyJjk2A98I5RQVmrOuLtET5Pqg,173
136
139
  snowflake/ml/jobs/_utils/scripts/get_instance_ip.py,sha256=N2wJYMPlwg-hidwgHhDhiBWOE6TskqCfWLMRRNnZBQs,5776
137
- snowflake/ml/jobs/_utils/scripts/mljob_launcher.py,sha256=D2yheAux0GrIcxzD7IPEuUaRb4JyDdDpxRXXNlxszWQ,19912
140
+ snowflake/ml/jobs/_utils/scripts/mljob_launcher.py,sha256=HuhAai6JL1Zdbo50Pg257Zeyfog90fJKVIvujmCASPw,26904
138
141
  snowflake/ml/jobs/_utils/scripts/signal_workers.py,sha256=AR1Pylkm4-FGh10WXfrCtcxaV0rI7IQ2ZiO0Li7zZ3U,7433
139
142
  snowflake/ml/jobs/_utils/scripts/start_mlruntime.sh,sha256=EIhhpEeu1Ph3J-QsAvugkhESkTMSZP3lWubmBCw5Z4w,3671
140
143
  snowflake/ml/jobs/_utils/scripts/startup.sh,sha256=OwJjQczoNwqf6v2vq6MeChNoa79NHfLeZG8VngRQMvQ,4148
141
144
  snowflake/ml/jobs/_utils/scripts/worker_shutdown_listener.py,sha256=SeJ8v5XDriwHAjIGpcQkwVP-f-lO9QIdVjVD7Fkgafs,7893
142
145
  snowflake/ml/lineage/__init__.py,sha256=8p1YGynC-qOxAZ8jZX2z84Reg5bv1NoJMoJmNJCrzI4,65
143
146
  snowflake/ml/lineage/lineage_node.py,sha256=kL3b8VXhhpwnDl1r7UT27QEJkXEoSIavBGG-_3BE8t8,5775
144
- snowflake/ml/model/__init__.py,sha256=gTZ_w-c57d1VrlxBXBGGt2384xenmQ193-xF8x6Vfz4,1172
147
+ snowflake/ml/model/__init__.py,sha256=Am6DlddRnxeiG_Thn72iF-SiUt7RFP79oD4mndqGu5s,1240
145
148
  snowflake/ml/model/code_path.py,sha256=erQhLkW6EY8V8FqDBBCFgMr4iImfMH0G6Mvw1cbDwws,4352
146
149
  snowflake/ml/model/compute_pool.py,sha256=a54DQ3VRnOU0tSc8_IBwADLOKJMxyqe5R1jlKxnMMvs,106
147
150
  snowflake/ml/model/custom_model.py,sha256=sdyKhT-QNNtTeu3idu6BExZNVyjUD4YTU8cru3iy4AY,14184
148
151
  snowflake/ml/model/event_handler.py,sha256=pojleQVM9TPNeDvliTvon2Sfxqbf2WWxrOebo1SaEHo,7211
149
152
  snowflake/ml/model/inference_engine.py,sha256=L0nwySY2Qwp3JzuRpPS87r0--m3HTUNUgZXYyOPJjyk,66
150
153
  snowflake/ml/model/model_signature.py,sha256=ae1tkh3Rw9MzJSxmVT9kb0PwD3TANtKbWwp6b8-cItE,32847
151
- snowflake/ml/model/openai_signatures.py,sha256=wu7l9V6-fWNJoAdE3R25hYBAmVfND5oaWdsUXilxBDo,7329
154
+ snowflake/ml/model/openai_signatures.py,sha256=ny3aAd301t3h_4Pb60MkrY0LZO6QxpyIlHYnjv120VQ,14755
152
155
  snowflake/ml/model/target_platform.py,sha256=H5d-wtuKQyVlq9x33vPtYZAlR5ka0ytcKRYgwlKl0bQ,390
153
156
  snowflake/ml/model/task.py,sha256=Zp5JaLB-YfX5p_HSaw81P3J7UnycQq5EMa87A35VOaQ,286
154
157
  snowflake/ml/model/type_hints.py,sha256=Xxa6b9ezbvXYvSIN5R4Zv6Dro4ZH74-eW4cno92VTJE,11475
155
158
  snowflake/ml/model/volatility.py,sha256=qu-wqe9oKkRwXwE2qkKygxTWzUypQYEk3UjsqOGRl_I,1144
156
- snowflake/ml/model/_client/model/batch_inference_specs.py,sha256=BA_pF9Hnwz9iIywi4CUaNMLPqWE4R7Q8QE9kWFU369E,4797
159
+ snowflake/ml/model/_client/model/batch_inference_specs.py,sha256=6q3XzYWaO1CH-JqonJr12n6NuRhQDvPns_FAq3yvrN4,6114
157
160
  snowflake/ml/model/_client/model/inference_engine_utils.py,sha256=yPkdImi2qP1uG1WzLKCBZgXV-DiIBVpImEosIjYJk8Y,1958
158
161
  snowflake/ml/model/_client/model/model_impl.py,sha256=Yabrbir5vPMOnsVmQJ23YN7vqhi756Jcm6pfO8Aq92o,17469
159
- snowflake/ml/model/_client/model/model_version_impl.py,sha256=Un8AiOjIxee8RmieXCMSeCfVWXjThurT5ABpXUDxjoM,69024
162
+ snowflake/ml/model/_client/model/model_version_impl.py,sha256=3R39U2xAWJdvZktJ_h1sAaxHmum7wd1dQP1dJE5PmbE,70403
160
163
  snowflake/ml/model/_client/ops/deployment_step.py,sha256=9kxKDr9xcD4KmVM-9O4_tm3ytkllQVoElJD793VI84Q,1428
161
164
  snowflake/ml/model/_client/ops/metadata_ops.py,sha256=qpK6PL3OyfuhyOmpvLCpHLy6vCxbZbp1HlEvakFGwv4,4884
162
- snowflake/ml/model/_client/ops/model_ops.py,sha256=QzUFshidzdHkWIalbI0iLLA46OUgt4DWQVYQnHsWF3A,54272
165
+ snowflake/ml/model/_client/ops/model_ops.py,sha256=s5-N9RlaJzndh0D1sQa4BnYb0N-FKYgx86O0d2H5mKg,53819
163
166
  snowflake/ml/model/_client/ops/param_utils.py,sha256=MPPerO8wYNYIuig4rFoG_YI5idD_dBlrWXgmAXcR2nM,5160
164
- snowflake/ml/model/_client/ops/service_ops.py,sha256=Ew86CVvKiJtAMcoDw_1QmQqwdoVRxKH85yXmHr5PpEs,48414
167
+ snowflake/ml/model/_client/ops/service_ops.py,sha256=9eWK3R_iOCkqafR_bK2OpAy0fdpjqYAK447ozZZF_Lg,47910
165
168
  snowflake/ml/model/_client/service/import_model_spec_schema.py,sha256=SlEX1GiPlB8whMCmiwKUopnrGlm4fkQOQbTW2KyVTFU,554
166
- snowflake/ml/model/_client/service/model_deployment_spec.py,sha256=Qg7cR9jQTnbWHIV5FyxUZocxWH0nFEem9wdOt5osKIw,20149
167
- snowflake/ml/model/_client/service/model_deployment_spec_schema.py,sha256=2FCfhyg7F2XYVWKFybeEQ9Fq3goZ3XKmJdH_CuRqPpI,2649
169
+ snowflake/ml/model/_client/service/model_deployment_spec.py,sha256=pRwwlD9h6V76fAOtFR1XgxXEv1fQjfdJCxBZgjgocE0,20286
170
+ snowflake/ml/model/_client/service/model_deployment_spec_schema.py,sha256=axEX0gnUqnLav575xzpJ9lDWNwr9vyjyay7soo7Hwgk,2672
168
171
  snowflake/ml/model/_client/sql/_base.py,sha256=Qrm8M92g3MHb-QnSLUlbd8iVKCRxLhG_zr5M2qmXwJ8,1473
169
172
  snowflake/ml/model/_client/sql/model.py,sha256=nstZ8zR7MkXVEfhqLt7PWMik6dZr06nzq7VsF5NVNow,5840
170
173
  snowflake/ml/model/_client/sql/model_version.py,sha256=SOYr13YEq0mxgIatsSchOq0aKUgdPhKO3clRQ6AMa7U,24766
171
- snowflake/ml/model/_client/sql/service.py,sha256=e_qnt-mGlac1vCGjqOHjYDpplvZy8YiC82Ir0VyH1Tc,14882
174
+ snowflake/ml/model/_client/sql/service.py,sha256=qTQ-_-Yul1DKIKs9agckt-qvE_IXJmcFgksMe_Co3tQ,14484
172
175
  snowflake/ml/model/_client/sql/stage.py,sha256=1TWYIVoWIeNwhVG9uqwmNpmKcC6x45LrbxCtzJW7fi4,1214
173
176
  snowflake/ml/model/_client/sql/tag.py,sha256=9sI0VoldKmsfToWSjMQddozPPGCxYUI6n0gPBiqd6x8,4333
174
177
  snowflake/ml/model/_model_composer/model_composer.py,sha256=1iMJe9b71xHC4K8VGTreQMJhI3ryAeQl3SpSK77uK_g,8313
@@ -179,7 +182,7 @@ snowflake/ml/model/_model_composer/model_method/function_generator.py,sha256=nnU
179
182
  snowflake/ml/model/_model_composer/model_method/infer_function.py_template,sha256=YBVm2KL6lUNuuyTwvYDrvc6x94SeqLPx7G_snRoUEMI,2193
180
183
  snowflake/ml/model/_model_composer/model_method/infer_partitioned.py_template,sha256=QsnDLnPKYgm01HSAflxWUUr5rf_tDnEyI6d8wgXd3fE,2253
181
184
  snowflake/ml/model/_model_composer/model_method/infer_table_function.py_template,sha256=p8ymUzEiZG6Rd7091HxRSEaxxNru6lT0SQAE46GcHxQ,2167
182
- snowflake/ml/model/_model_composer/model_method/model_method.py,sha256=txvypSvsNh9mvJLwP01B4WiosFNzYjqLS-dSgD2bDYw,11916
185
+ snowflake/ml/model/_model_composer/model_method/model_method.py,sha256=6iHPhGGocF5NAlr3LRuVnlzjOP0uF3Cl9k6B0c0BjEk,11986
183
186
  snowflake/ml/model/_model_composer/model_method/utils.py,sha256=RQi2qebBeE-0Y-jLYXiDWZU8nfvbnif9QbExeWiMmyI,1057
184
187
  snowflake/ml/model/_model_composer/model_user_file/model_user_file.py,sha256=dYNgg8P9p6nRH47-OLxZIbt_Ja3t1VPGNQ0qJtpGuAw,1018
185
188
  snowflake/ml/model/_packager/model_handler.py,sha256=xbqTFDC7ArQyYuOk6zMV_dSttWbsKKVuaOUAY3ddQmE,2846
@@ -189,13 +192,13 @@ snowflake/ml/model/_packager/model_handlers/_base.py,sha256=OZhGv7nyej3PqaoBz021
189
192
  snowflake/ml/model/_packager/model_handlers/_utils.py,sha256=v_IjjbvzJDqrAYSq4_l7_CiN8vkMzLx5MlYDJ_oL970,15522
190
193
  snowflake/ml/model/_packager/model_handlers/catboost.py,sha256=dbI2QizGZS04l6ehgXb3oy5YSXrlwRHz8YENVefEbms,10676
191
194
  snowflake/ml/model/_packager/model_handlers/custom.py,sha256=eOPTt3BoT5r9Z6rzvKaLhbffND647-j8aQ8fD50-cmA,10595
192
- snowflake/ml/model/_packager/model_handlers/huggingface.py,sha256=rp4W4ZlkeGjaUq3xZHZuUVfroWggVDSXOGV5R0fbet8,45216
195
+ snowflake/ml/model/_packager/model_handlers/huggingface.py,sha256=2I3IqX3jilT8vWU4-7Z3mhmBN2l4HPTJmsYkOg_qP5Y,46345
193
196
  snowflake/ml/model/_packager/model_handlers/keras.py,sha256=JKBCiJEjc41zaoEhsen7rnlyPo2RBuEqG9Vq6JR_Cq0,8696
194
197
  snowflake/ml/model/_packager/model_handlers/lightgbm.py,sha256=DAFMiqpXEUmKqeq5rgn5j6rtuwScNnuiMUBwS4OyC7Q,11074
195
198
  snowflake/ml/model/_packager/model_handlers/mlflow.py,sha256=6ZkRRiaTtFatIv1zaWAJ-TGkTRTmrj0P-YCSgpYnRrQ,9434
196
199
  snowflake/ml/model/_packager/model_handlers/prophet.py,sha256=MzmIkX2WukTApf85SUEmn8k_jeBAhfGa_8RTZK0-aCA,23913
197
200
  snowflake/ml/model/_packager/model_handlers/pytorch.py,sha256=mF-pzH1kqL7egpYA3kP1NVwOLNPYdOViEkywdzRXYJc,9867
198
- snowflake/ml/model/_packager/model_handlers/sentence_transformers.py,sha256=ER3FEZB-iq3_rGxofOsBGDFSuxC2OUOm020rP97SEhs,13023
201
+ snowflake/ml/model/_packager/model_handlers/sentence_transformers.py,sha256=xTJUcouBYmg1ZgOQ2tENOtmXGMHZRKoTJcDz36-gvdc,15932
199
202
  snowflake/ml/model/_packager/model_handlers/sklearn.py,sha256=_D1YE7TmEJDsuOUt-mT-2Nza2Bz0sIzSGRKn9sxuDhI,18340
200
203
  snowflake/ml/model/_packager/model_handlers/snowmlmodel.py,sha256=uvz-hosuNbtcQFprnS8GzjnM8fWULBDMRbXq8immW9Q,18352
201
204
  snowflake/ml/model/_packager/model_handlers/tensorflow.py,sha256=2J2XWYOC70axWaoNJa9aQLMyjLAKIskrT31t_LgqcIk,11350
@@ -224,9 +227,9 @@ snowflake/ml/model/_signatures/pandas_handler.py,sha256=Gz2olwWzT4Kb3yBH0uYn3o92
224
227
  snowflake/ml/model/_signatures/pytorch_handler.py,sha256=Xy-ITCCX_EgHcyIIqeYSDUIvE2kiqECa8swy1hmohyc,5036
225
228
  snowflake/ml/model/_signatures/snowpark_handler.py,sha256=aNGPa2v0kTMuSZ80NBdHeAWYva0Nc1vo17ZjQwIjf2E,7621
226
229
  snowflake/ml/model/_signatures/tensorflow_handler.py,sha256=_yrvMg-w_jJoYuyrGXKPX4Dv7Vt8z1e6xIKiWGuZcc4,5660
227
- snowflake/ml/model/_signatures/utils.py,sha256=6iu52gcbaZXQb6rDgzGJDNRhKQk0MqwJnt9ihDZQHC4,19014
230
+ snowflake/ml/model/_signatures/utils.py,sha256=hoc_UuMxfPWkVmoMEE7U-XNRcSgDPyRIdyDVK0JLcfE,21685
228
231
  snowflake/ml/model/models/huggingface.py,sha256=VO84lBizmSALntWCnK4O_eY_Cq2uzMooyHtfJXuFkew,13791
229
- snowflake/ml/model/models/huggingface_pipeline.py,sha256=yJ7NW97EW5GRtOPYCYuNjoPIgXbAIyOEKW9P1trt9vY,14226
232
+ snowflake/ml/model/models/huggingface_pipeline.py,sha256=tLTj1YJumNs_pdHGmA1Us5HIi8EYm38X6jqv2j85TMo,14378
230
233
  snowflake/ml/modeling/_internal/estimator_utils.py,sha256=dfPPWO-RHf5C3Tya3VQ4KEqoa32pm-WKwRrjzjDInLk,13956
231
234
  snowflake/ml/modeling/_internal/model_specifications.py,sha256=3wFMcKPCSoiEzU7Mx6RVem89BRlBBENpX__-Rd7GwdU,4851
232
235
  snowflake/ml/modeling/_internal/model_trainer.py,sha256=5Ck1lbdyzcd-TpzAxEyovIN9fjaaVIqugyMHXt0wzH0,971
@@ -455,14 +458,14 @@ snowflake/ml/monitoring/entities/model_monitor_config.py,sha256=auy9BD0IoyUpZPZX
455
458
  snowflake/ml/registry/__init__.py,sha256=XdPQK9ejYkSJVrSQ7HD3jKQO0hKq2mC4bPCB6qrtH3U,76
456
459
  snowflake/ml/registry/registry.py,sha256=_vtQCh4DmhnPusTKWJteRPJkDpLFEfG150cjED70sOA,34611
457
460
  snowflake/ml/registry/_manager/model_manager.py,sha256=splK5YGErt-eDIy6UbZAB3VKsGMZSJk2_MzfgrIQOhY,26306
458
- snowflake/ml/registry/_manager/model_parameter_reconciler.py,sha256=BTi8WNQCW1TiW8jd9LbR6mrIHE29ehocKSh6mVQX0vI,15325
461
+ snowflake/ml/registry/_manager/model_parameter_reconciler.py,sha256=92YX67YMLmqbJN_emr49c3QWyFVGc1DUn6JvA3Y2qqo,15267
459
462
  snowflake/ml/utils/authentication.py,sha256=TQV3E8YDHAPXA3dS8JWDmb_Zm8P0d9c8kCexRI4nefo,3106
460
463
  snowflake/ml/utils/connection_params.py,sha256=NSBUgcs-DXPRHs1BKpxdSubbJx1yrFRlMPBp-bE3Ugc,8308
461
464
  snowflake/ml/utils/html_utils.py,sha256=4g1EPuD8EnOAK7BCYiY8Wp3ZrdDkNOcUDrDAbUYxLfs,9954
462
465
  snowflake/ml/utils/sparse.py,sha256=zLBNh-ynhGpKH5TFtopk0YLkHGvv0yq1q-sV59YQKgg,3819
463
466
  snowflake/ml/utils/sql_client.py,sha256=pSe2od6Pkh-8NwG3D-xqN76_uNf-ohOtVbT55HeQg1Y,668
464
- snowflake_ml_python-1.23.0.dist-info/licenses/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
465
- snowflake_ml_python-1.23.0.dist-info/METADATA,sha256=xQBz4OlefNPg0IGBeFQ3iFWZA3WYziR-xmnaVaiyQE8,104165
466
- snowflake_ml_python-1.23.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
467
- snowflake_ml_python-1.23.0.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
468
- snowflake_ml_python-1.23.0.dist-info/RECORD,,
467
+ snowflake_ml_python-1.25.0.dist-info/licenses/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
468
+ snowflake_ml_python-1.25.0.dist-info/METADATA,sha256=hV6uhSQvACNGp9jEdujaC9Y_F5r5rf1eBI22Hh6FBe0,107037
469
+ snowflake_ml_python-1.25.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
470
+ snowflake_ml_python-1.25.0.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
471
+ snowflake_ml_python-1.25.0.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (80.9.0)
2
+ Generator: setuptools (80.10.2)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5
 
@@ -1,43 +0,0 @@
1
- import inspect
2
- from typing import Any, Callable, Optional
3
-
4
- from snowflake import snowpark
5
- from snowflake.snowpark import context as sp_context
6
-
7
-
8
- class FunctionPayload:
9
- def __init__(
10
- self,
11
- func: Callable[..., Any],
12
- session: Optional[snowpark.Session] = None,
13
- session_argument: str = "",
14
- *args: Any,
15
- **kwargs: Any
16
- ) -> None:
17
- self.function = func
18
- self.args = args
19
- self.kwargs = kwargs
20
- self._session = session
21
- self._session_argument = session_argument
22
-
23
- @property
24
- def session(self) -> Optional[snowpark.Session]:
25
- return self._session
26
-
27
- def __getstate__(self) -> dict[str, Any]:
28
- """Customize pickling to exclude session."""
29
- state = self.__dict__.copy()
30
- state["_session"] = None
31
- return state
32
-
33
- def __setstate__(self, state: dict[str, Any]) -> None:
34
- """Restore session from context during unpickling."""
35
- self.__dict__.update(state)
36
- self._session = sp_context.get_active_session()
37
-
38
- def __call__(self) -> Any:
39
- sig = inspect.signature(self.function)
40
- bound = sig.bind_partial(*self.args, **self.kwargs)
41
- bound.arguments[self._session_argument] = self._session
42
-
43
- return self.function(*bound.args, **bound.kwargs)
@@ -1,22 +0,0 @@
1
- from snowflake import snowpark
2
- from snowflake.ml._internal.utils import snowflake_env
3
- from snowflake.ml.jobs._utils import constants, query_helper, types
4
-
5
-
6
- def _get_node_resources(session: snowpark.Session, compute_pool: str) -> types.ComputeResources:
7
- """Extract resource information for the specified compute pool"""
8
- # Get the instance family
9
- rows = query_helper.run_query(
10
- session,
11
- "show compute pools like ?",
12
- params=[compute_pool],
13
- )
14
- if not rows:
15
- raise ValueError(f"Compute pool '{compute_pool}' not found")
16
- instance_family: str = rows[0]["instance_family"]
17
- cloud = snowflake_env.get_current_cloud(session, default=snowflake_env.SnowflakeCloudType.AWS)
18
-
19
- return (
20
- constants.COMMON_INSTANCE_FAMILIES.get(instance_family)
21
- or constants.CLOUD_INSTANCE_FAMILIES[cloud][instance_family]
22
- )