snowflake-ml-python 1.5.0__py3-none-any.whl → 1.5.1__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 (38) hide show
  1. snowflake/ml/_internal/env_utils.py +6 -0
  2. snowflake/ml/_internal/lineage/lineage_utils.py +95 -0
  3. snowflake/ml/_internal/telemetry.py +1 -0
  4. snowflake/ml/_internal/utils/identifier.py +1 -1
  5. snowflake/ml/_internal/utils/sql_identifier.py +14 -1
  6. snowflake/ml/dataset/__init__.py +2 -1
  7. snowflake/ml/dataset/dataset.py +4 -3
  8. snowflake/ml/dataset/dataset_reader.py +5 -8
  9. snowflake/ml/feature_store/__init__.py +6 -0
  10. snowflake/ml/feature_store/access_manager.py +279 -0
  11. snowflake/ml/feature_store/feature_store.py +159 -99
  12. snowflake/ml/feature_store/feature_view.py +18 -8
  13. snowflake/ml/fileset/embedded_stage_fs.py +15 -12
  14. snowflake/ml/fileset/snowfs.py +3 -2
  15. snowflake/ml/fileset/stage_fs.py +25 -7
  16. snowflake/ml/model/_client/model/model_impl.py +46 -39
  17. snowflake/ml/model/_client/model/model_version_impl.py +24 -2
  18. snowflake/ml/model/_client/ops/metadata_ops.py +27 -4
  19. snowflake/ml/model/_client/ops/model_ops.py +131 -16
  20. snowflake/ml/model/_client/sql/_base.py +34 -0
  21. snowflake/ml/model/_client/sql/model.py +32 -39
  22. snowflake/ml/model/_client/sql/model_version.py +60 -43
  23. snowflake/ml/model/_client/sql/stage.py +6 -32
  24. snowflake/ml/model/_client/sql/tag.py +32 -56
  25. snowflake/ml/model/_model_composer/model_composer.py +2 -2
  26. snowflake/ml/model/_packager/model_handlers/mlflow.py +2 -1
  27. snowflake/ml/modeling/_internal/snowpark_implementations/distributed_hpo_trainer.py +50 -21
  28. snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_trainer.py +81 -3
  29. snowflake/ml/modeling/framework/base.py +4 -3
  30. snowflake/ml/modeling/pipeline/pipeline.py +27 -7
  31. snowflake/ml/registry/_manager/model_manager.py +36 -7
  32. snowflake/ml/version.py +1 -1
  33. {snowflake_ml_python-1.5.0.dist-info → snowflake_ml_python-1.5.1.dist-info}/METADATA +54 -10
  34. {snowflake_ml_python-1.5.0.dist-info → snowflake_ml_python-1.5.1.dist-info}/RECORD +37 -35
  35. snowflake/ml/_internal/lineage/dataset_dataframe.py +0 -44
  36. {snowflake_ml_python-1.5.0.dist-info → snowflake_ml_python-1.5.1.dist-info}/LICENSE.txt +0 -0
  37. {snowflake_ml_python-1.5.0.dist-info → snowflake_ml_python-1.5.1.dist-info}/WHEEL +0 -0
  38. {snowflake_ml_python-1.5.0.dist-info → snowflake_ml_python-1.5.1.dist-info}/top_level.txt +0 -0
@@ -5,13 +5,13 @@ snowflake/cortex/_sentiment.py,sha256=7X_a8qJNFFgn-Y1tjwMDkyNJHz5yYl0PvnezVCc4Ts
5
5
  snowflake/cortex/_summarize.py,sha256=DJRxUrPrTVmtQNgus0ZPF1z8nPmn4Rs5oL3U25CfXxQ,1075
6
6
  snowflake/cortex/_translate.py,sha256=JPMIXxHTgJPfJqT5Hw_WtYM6FZ8NuQufZ4XR-M8wnyo,1420
7
7
  snowflake/cortex/_util.py,sha256=0xDaDSctenhuj59atZenZp5q9zuhji0WQ77KPjqqNoc,1557
8
- snowflake/ml/version.py,sha256=3R9EThHATDpTgUWWErtoGeHWTDLDz1F9kgtQ7lMuM80,16
8
+ snowflake/ml/version.py,sha256=WiE22xydACC6rXf2dqEDDHIINeQkQXbfF2V05DUpD-o,16
9
9
  snowflake/ml/_internal/env.py,sha256=kCrJTRnqQ97VGUVI1cWUPD8HuBWeL5vOOtwUR0NB9Mg,161
10
- snowflake/ml/_internal/env_utils.py,sha256=aqaF-bXPUvXxONfMRxIuH-JKyu3oWkRqyC_1jDEmK4Y,27629
10
+ snowflake/ml/_internal/env_utils.py,sha256=HK5Ug5-gChiUv_z84BDjAuE9eHImrWRsX4Y7wJFApfk,27758
11
11
  snowflake/ml/_internal/file_utils.py,sha256=OyXHv-UcItiip1YgLnab6etonUQkYuyDtmplZA0CaoU,13622
12
12
  snowflake/ml/_internal/init_utils.py,sha256=U-oPOtyVf22hCwDH_CH2uDr9yuN6Mr3kwQ_yRAs1mcM,2696
13
13
  snowflake/ml/_internal/migrator_utils.py,sha256=k3erO8x3YJcX6nkKeyJAUNGg1qjE3RFmD-W6dtLzIH0,161
14
- snowflake/ml/_internal/telemetry.py,sha256=oM7dDcs1GKgxKP2UM7va1j1YfQGISFiGYyiT9zM7Yxc,22763
14
+ snowflake/ml/_internal/telemetry.py,sha256=E8AEeXgmSKzYx709WYMaTBMWF8VAr259cgmsIFs3IAw,22820
15
15
  snowflake/ml/_internal/type_utils.py,sha256=0AjimiQoAPHGnpLV_zCR6vlMR5lJ8CkZkKFwiUHYDCo,2168
16
16
  snowflake/ml/_internal/container_services/image_registry/credential.py,sha256=nShNgIb2yNu9w6vceOY3aSgjpuOoi0spWWmvgEafPSk,3291
17
17
  snowflake/ml/_internal/container_services/image_registry/http_client.py,sha256=_zqPPp76Vk0jQ8eVK0OJ4mJgcWsdY4suUd1P7Orqmm8,5214
@@ -30,9 +30,9 @@ snowflake/ml/_internal/human_readable_id/animals.txt,sha256=GDLzMwzxiL07PhIMxw4t
30
30
  snowflake/ml/_internal/human_readable_id/hrid_generator.py,sha256=LYWB86qZgsVBvnc6Q5VjfDOmnGSQU3cTRKfId_nJSPY,1341
31
31
  snowflake/ml/_internal/human_readable_id/hrid_generator_base.py,sha256=D1yoVG1vmAFUhWQ5xCRRU6HCCBPbXHpOXagFd0jK0O8,4519
32
32
  snowflake/ml/_internal/lineage/data_source.py,sha256=D24FdR6Wq_PdUuCsBDvSMCr5CfHqpMamrc8-F5iZVJ0,214
33
- snowflake/ml/_internal/lineage/dataset_dataframe.py,sha256=qbEXgkHxzx6zZzJCGpIhFV7-OdAuA_qrO9AixxyxHSk,1712
33
+ snowflake/ml/_internal/lineage/lineage_utils.py,sha256=4BNoyg3GFeUY5tDNdjDvN129rc6JymOt6PUdWP_Vhj4,3007
34
34
  snowflake/ml/_internal/utils/formatting.py,sha256=PswZ6Xas7sx3Ok1MBLoH2o7nfXOxaJqpUPg_UqXrQb8,3676
35
- snowflake/ml/_internal/utils/identifier.py,sha256=eokEDF7JIML2gm_3FfknPdPR9aBT3woweA5S4z_46-E,10925
35
+ snowflake/ml/_internal/utils/identifier.py,sha256=7dV6dN_KAoupT-xJS8f19K69GVWa4069RmKVWMuWH9k,10926
36
36
  snowflake/ml/_internal/utils/import_utils.py,sha256=eexwIe7auT17s4aVxAns7se0_K15rcq3O17MkIvDpPI,2068
37
37
  snowflake/ml/_internal/utils/log_stream_processor.py,sha256=pBf8ycEamhHjEzUT55Rx_tFqSkYRpD5Dt71Mx9ZdaS8,1001
38
38
  snowflake/ml/_internal/utils/parallelize.py,sha256=Q6_-P2t4DoYNO8DyC1kOl7H3qNL-bUK6EgtlQ_b5ThY,4534
@@ -44,25 +44,26 @@ snowflake/ml/_internal/utils/session_token_manager.py,sha256=qXRlE7pyw-Gb0q_BmTd
44
44
  snowflake/ml/_internal/utils/snowflake_env.py,sha256=Mrov0v95pzVUeAe7r1e1PtlIco9ytj5SGAuUWORQaKs,2927
45
45
  snowflake/ml/_internal/utils/snowpark_dataframe_utils.py,sha256=HPyWxj-SwgvWUrYR38BkBtx813eMqz5wmQosgc1sce0,5403
46
46
  snowflake/ml/_internal/utils/spcs_attribution_utils.py,sha256=9XPKe1BDkWhnGuHDXBHE4FP-m3U22lTZnrQLsHGFhWU,4292
47
- snowflake/ml/_internal/utils/sql_identifier.py,sha256=CHTxr3qtc1ygNkA5oOQQa-XEoosw5sjfHe7J4WZlkDQ,3270
47
+ snowflake/ml/_internal/utils/sql_identifier.py,sha256=ZcRjSfpovsqaY7S8bFB6z44z28XICncHGwOIzs8rLDI,3729
48
48
  snowflake/ml/_internal/utils/table_manager.py,sha256=jHGfl0YSqhFLL7DOOQkjUMzTmLkqFDIM7Gs0LBQw8BM,4384
49
49
  snowflake/ml/_internal/utils/temp_file_utils.py,sha256=7JNib0DvjxW7Eu3bimwAHibGosf0u8W49HEc49BghF8,1402
50
50
  snowflake/ml/_internal/utils/uri.py,sha256=pvskcWoeS0M66DaU2XlJzK9wce55z4J5dn5kTy_-Tqs,2828
51
- snowflake/ml/dataset/__init__.py,sha256=jyoLosJL9mTWTULTqJB3WaDS77-zsPcdoWDTNxyN-rY,236
52
- snowflake/ml/dataset/dataset.py,sha256=V6QeEN8WatRnpoXRss950ISKiE0hKrJg7PCFkHKmfYo,20863
51
+ snowflake/ml/dataset/__init__.py,sha256=nESj7YEI2u90Oxyit_hKCQMWb7N1BlEM3Ho2Fm0MfHo,274
52
+ snowflake/ml/dataset/dataset.py,sha256=LbiYP2S-dnw8a2ALswSLqZs7AittzSejMC9Hzipkpn0,21013
53
53
  snowflake/ml/dataset/dataset_factory.py,sha256=qdS6jX8uiCpW5TIKnZ-_2HRfWN3c_N1bZ6lBC1bLy5g,1712
54
54
  snowflake/ml/dataset/dataset_metadata.py,sha256=lvaYd1sNOgWcXD1q_-J7fQZ0ndOC8guR9IgKpChBcFA,3992
55
- snowflake/ml/dataset/dataset_reader.py,sha256=0Vgr_xV4YIBewPeymGNpaSCA7AVZcCgyq8pSuSwc2Ys,8205
56
- snowflake/ml/feature_store/__init__.py,sha256=dYtqk_GD_hAAZjGfH1maWlZQ30h4hu_KGaf-_y9_AD8,298
55
+ snowflake/ml/dataset/dataset_reader.py,sha256=TKitOC7YBk3yZ9axL9nI1paSI2ooSqBn4zw5eOYpCGY,8061
56
+ snowflake/ml/feature_store/__init__.py,sha256=VKBVkS050WNF8rcqNqwFfNXa_B9GZjcUpuibOGsUSls,423
57
+ snowflake/ml/feature_store/access_manager.py,sha256=TiMHu5ds4ZsjvGTOPWum7zgb4A-m3LNFDSktBFNezdk,10442
57
58
  snowflake/ml/feature_store/entity.py,sha256=dCpzLC3jrt5wDHqFYJXbAYkMiZ0zEmiVDMGkks6MXkA,3378
58
- snowflake/ml/feature_store/feature_store.py,sha256=SrJkB_9Y6clbw6jCWmKvjGoZ78PU2hzUSnHTVnTfDKo,76062
59
- snowflake/ml/feature_store/feature_view.py,sha256=Epqmm5QiK6jdgrgE-x54Lq1B6VBocFLAVk5bt13eyYg,18624
60
- snowflake/ml/fileset/embedded_stage_fs.py,sha256=gzMX6RbU_K9jCy1zEfF0YZ7nZSVHW4b3UTVAuAn-etY,5740
59
+ snowflake/ml/feature_store/feature_store.py,sha256=8SH806Dwy89TbKItTdnPi0pAn09pRObgjYSpmBnEjrI,78669
60
+ snowflake/ml/feature_store/feature_view.py,sha256=B3oYaRuChQaLo8c8sdUF6McswVUryda5GWMK22b3Ipg,19274
61
+ snowflake/ml/fileset/embedded_stage_fs.py,sha256=90nCRvRm2EZpDlx-Hu-NLI5s9fYbEFHdf0ggwjdrkQM,5919
61
62
  snowflake/ml/fileset/fileset.py,sha256=QRhxLeKf1QBqvXO4RyyRd1c8TixhYpHuBEII8Qi3C_M,26201
62
63
  snowflake/ml/fileset/parquet_parser.py,sha256=sjyRB59cGBzSzvbcYLvu_ApMPtrR-zwZsQkxekMR4FA,6884
63
64
  snowflake/ml/fileset/sfcfs.py,sha256=a77UJFz5Ve9s_26QpcOOoFNOBIKN91KmhYVTQkafn0c,15344
64
- snowflake/ml/fileset/snowfs.py,sha256=qwBbmMh5mGneO3scriERbdWJ8e717NfkUMXAz2Y4Dhg,6914
65
- snowflake/ml/fileset/stage_fs.py,sha256=yKMbPlc4chLG2svHKYy7tlefq6hy7kvnefHGunE7DPA,18408
65
+ snowflake/ml/fileset/snowfs.py,sha256=RXCtZ43_e_Kq_vc-1tJABNo0stpwmHQI2MSCLeFhGfI,6948
66
+ snowflake/ml/fileset/stage_fs.py,sha256=IebRjgPlJdwdAlpg_99DGbgIBD3XJb2p9N36O0tU3wI,19532
66
67
  snowflake/ml/fileset/tf_dataset.py,sha256=K8jafWBsyRaIYEmxaYAYNDj3dLApK82cg0Mlx52jX8I,3849
67
68
  snowflake/ml/fileset/torch_datapipe.py,sha256=O2irHckqLzPDnXemEbAEjc3ZCVnLufPdPbt9WKYiBp0,2386
68
69
  snowflake/ml/model/__init__.py,sha256=KgZmgLHXmkmEU5Q7pzYQlpfvIll4SRTSiT9s4RjeleI,393
@@ -71,14 +72,15 @@ snowflake/ml/model/custom_model.py,sha256=xvu7WZ1YmOdvuPePyAj6qMwKq-HNeVV9bNfkOT
71
72
  snowflake/ml/model/deploy_platforms.py,sha256=r6cS3gTNWG9i4P00fHehY6Q8eBiNva6501OTyp_E5m0,144
72
73
  snowflake/ml/model/model_signature.py,sha256=UQSGieGJcnmC02V4feCYMdhMXnGoOUa9KBuDrbeivBM,29342
73
74
  snowflake/ml/model/type_hints.py,sha256=aUg_1xNtzdH2_kH48v918jbpEnHPNIn6MmfrwdvYvdg,12705
74
- snowflake/ml/model/_client/model/model_impl.py,sha256=vyjPVoiEUFLW_XGz2tMXHBOk0_yDI5DgnDq3RTmQuW0,13623
75
- snowflake/ml/model/_client/model/model_version_impl.py,sha256=92uLtim0rZAngg1D0GSFNhPH1QDwZaT6jxOg_rRKcL4,17320
76
- snowflake/ml/model/_client/ops/metadata_ops.py,sha256=XFNolmueu0nC3nAjb2Lj3v1NffDAhAq0JWMek9JVO38,4094
77
- snowflake/ml/model/_client/ops/model_ops.py,sha256=J96RRYHOxtO3ScQZYS176Mr4B1BDuWh7UwH-2cSIgng,23490
78
- snowflake/ml/model/_client/sql/model.py,sha256=dem_jSDQb16bW0U0PvtNbR48XEiceo-sKn86dNFoXBs,5687
79
- snowflake/ml/model/_client/sql/model_version.py,sha256=HKAiIWvqWtcADUSMXUtAISGj-z8AJFbxGSBackuo3N0,14290
80
- snowflake/ml/model/_client/sql/stage.py,sha256=4zP8aO6cv0IDrZEqhkheNWwy4qBuv1qyGLwMFSW-7EI,1497
81
- snowflake/ml/model/_client/sql/tag.py,sha256=RYvblBfQmK4xmLF0pz0BNUd9wddqlfHtEK1JRRpJGPE,4646
75
+ snowflake/ml/model/_client/model/model_impl.py,sha256=hVtAHejB2pTDquWs4XNS7E7XZS1DI7nH7EILbd0btbc,13655
76
+ snowflake/ml/model/_client/model/model_version_impl.py,sha256=Li9JtKwZvNqKjpAQM4qA52-F0fu-HASt0RWPDEJGFPE,17994
77
+ snowflake/ml/model/_client/ops/metadata_ops.py,sha256=7cGx8zYzye2_cvZnyGxoukPtT6Q-Kexd-s4yeZmpmj8,4890
78
+ snowflake/ml/model/_client/ops/model_ops.py,sha256=bn2dB9N_OHP1yMNoFwR4OFxID2_pFQytpWnfZ_195is,28714
79
+ snowflake/ml/model/_client/sql/_base.py,sha256=pN5hxyC0gGzEJgZh2FBHLU0Y6iIoLcebHoE7wTpoUZQ,1252
80
+ snowflake/ml/model/_client/sql/model.py,sha256=dKgrkYKuuAIaOcAC1K7_wxWgrtGF1r89sItcP00hUzY,5736
81
+ snowflake/ml/model/_client/sql/model_version.py,sha256=eQsvfub4Vduy0t3NY4PTHzxwwgKl16nRVhzY7lq-ehk,15685
82
+ snowflake/ml/model/_client/sql/stage.py,sha256=hrCh9P9F4l5R0hLr2r-wLDIEc4XYHMFdX1wNRveMVt0,819
83
+ snowflake/ml/model/_client/sql/tag.py,sha256=pwwrcyPtSnkUfDzL3M8kqM0KSx7CaTtgty3HDhVC9vg,4345
82
84
  snowflake/ml/model/_deploy_client/image_builds/base_image_builder.py,sha256=clCaoO0DZam4X79UtQV1ZuMQtTezAJkhLu9ViAX18Xk,302
83
85
  snowflake/ml/model/_deploy_client/image_builds/client_image_builder.py,sha256=G74D9lV2B3d544YzFN-YrjPkaST7tbQeh-rM17dtoJc,10681
84
86
  snowflake/ml/model/_deploy_client/image_builds/docker_context.py,sha256=7uhAJsHsk7LbiZv_w3xOCE2O88rTUVnS3_B6OAz-JG4,6129
@@ -97,7 +99,7 @@ snowflake/ml/model/_deploy_client/utils/constants.py,sha256=Ip_2GgsCYRXj_mD4MUdk
97
99
  snowflake/ml/model/_deploy_client/utils/snowservice_client.py,sha256=k0SulzWdttRvJkyuXM59aluEVgQg8Qd7XZUUpEBKuO4,11671
98
100
  snowflake/ml/model/_deploy_client/warehouse/deploy.py,sha256=yZR9M76oh6JbPQJHb6t3wGO3wuD04w0zLEXiEyZW_tg,8358
99
101
  snowflake/ml/model/_deploy_client/warehouse/infer_template.py,sha256=1THMd6JX1nW-OozECyxXbn9HJXDgNBUIdhfC9ODPDWY,3011
100
- snowflake/ml/model/_model_composer/model_composer.py,sha256=QSliEFs-wr9LguSTkQWIA9Nbw_9mOsHpUnanVsgV3Qs,7325
102
+ snowflake/ml/model/_model_composer/model_composer.py,sha256=Ld11EWtryUMM0QhbLZmZEgNtyysEtxP1aG1Vfk-oSNk,7356
101
103
  snowflake/ml/model/_model_composer/model_manifest/model_manifest.py,sha256=5tMz0d7t9f0oJAEAOXC4BDDpMNAV4atKoK9C66ZHgvU,5667
102
104
  snowflake/ml/model/_model_composer/model_manifest/model_manifest_schema.py,sha256=PsRVrOt15Zr-t2K64_GK5aHjTWN4yLgixRqaYchY2rA,2530
103
105
  snowflake/ml/model/_model_composer/model_method/function_generator.py,sha256=2B-fykyanYlGWA4Ie2nOwXx2N5D2qZEvTbbPuSSreeI,1837
@@ -114,7 +116,7 @@ snowflake/ml/model/_packager/model_handlers/custom.py,sha256=y5CHdEeKWAO08uor2Ot
114
116
  snowflake/ml/model/_packager/model_handlers/huggingface_pipeline.py,sha256=Z7vZ5zhZByLVPfNdSkhgzBot1Y8UBOM3ITj3Qfway3A,19985
115
117
  snowflake/ml/model/_packager/model_handlers/lightgbm.py,sha256=PWPdpOdden2vfloXZA5sA20b2dCBiGO1-NfJ8atH-Uc,8445
116
118
  snowflake/ml/model/_packager/model_handlers/llm.py,sha256=SgCgy9Ys5KivNymjF35ufCpPOtMtSby2Zu4Tllir8Mg,10772
117
- snowflake/ml/model/_packager/model_handlers/mlflow.py,sha256=Itw1fPiBdU2euOmjLU3P6Vyfj9Go3jSx1c-yHlQRYpU,8993
119
+ snowflake/ml/model/_packager/model_handlers/mlflow.py,sha256=5Kyfg962x_kJQyabIQXf72bO0cAgNgj1vfy48RD9osw,9022
118
120
  snowflake/ml/model/_packager/model_handlers/pytorch.py,sha256=dSxKO530_IlF1OK3t9_UYpVntdPiszKy-x_7XGk0bzQ,8033
119
121
  snowflake/ml/model/_packager/model_handlers/sentence_transformers.py,sha256=JRPargMNEJaDFQIpzqEVvOml62G_UVVvJdqBH8Lhu_Y,9051
120
122
  snowflake/ml/model/_packager/model_handlers/sklearn.py,sha256=u4ino0SxjAMxEl2jgTqt6Mqs1dKGZmSE90mIp3qHErU,8218
@@ -155,9 +157,9 @@ snowflake/ml/modeling/_internal/local_implementations/pandas_handlers.py,sha256=
155
157
  snowflake/ml/modeling/_internal/local_implementations/pandas_trainer.py,sha256=MyTRkBV3zbDeO7HJaWrKYT3KkVqW51Q0AX2BbUtN4og,5737
156
158
  snowflake/ml/modeling/_internal/ml_runtime_implementations/ml_runtime_handlers.py,sha256=fgm1DpBBO0qUo2fXFwuN2uFAyTFhcIhT5_bC326VTVw,5544
157
159
  snowflake/ml/modeling/_internal/ml_runtime_implementations/ml_runtime_trainer.py,sha256=lM1vYwpJ1jgTh8vnuyMp4tnFibM6UFf50W1IpPWwUWE,2535
158
- snowflake/ml/modeling/_internal/snowpark_implementations/distributed_hpo_trainer.py,sha256=baETLCVGDcaaGXwiOx6QyhaMX_zQ1Kt7xGjotd_MSKo,54368
160
+ snowflake/ml/modeling/_internal/snowpark_implementations/distributed_hpo_trainer.py,sha256=zBCEXRbO3w8BqL2ASpq09z9R_DXRTogOoml2Cdbggt0,55706
159
161
  snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_handlers.py,sha256=AgNupOLqXJytkbdQ1p5Nj1L5QShwi8PSUSYj506SxhM,14539
160
- snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_trainer.py,sha256=MYLvpNscLKI0AOuJIdKprtw3VinO0MtSVPj376k-ILo,31819
162
+ snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_trainer.py,sha256=WiRekLxqcnnEJV3dHyjyU797tnKsgxj_g-ZAjmIVWVk,35283
161
163
  snowflake/ml/modeling/_internal/snowpark_implementations/xgboost_external_memory_trainer.py,sha256=VBYWGTy6ajQ-u2aiEvVU6NnKobEqJyz65oaHJS-ZjBs,17208
162
164
  snowflake/ml/modeling/calibration/__init__.py,sha256=rY5qSOkHj59bHiTV6LhBiEhUA0StoCb0ACNR2vkV4v0,297
163
165
  snowflake/ml/modeling/calibration/calibrated_classifier_cv.py,sha256=AGQub8A5L_xTB1gEJsbzTSZdsISnhdsAp3OmbEwRutw,51278
@@ -228,7 +230,7 @@ snowflake/ml/modeling/feature_selection/select_percentile.py,sha256=hnTQ5DtNot9O
228
230
  snowflake/ml/modeling/feature_selection/sequential_feature_selector.py,sha256=-0ZEZboABCnbfguy1YmopbjVZ0-eut24E9XaPol97Sk,50387
229
231
  snowflake/ml/modeling/feature_selection/variance_threshold.py,sha256=KwTTDc1WLWz6SiXWz3Bks7u-zP53kdqPRLB3RHtvodw,47323
230
232
  snowflake/ml/modeling/framework/_utils.py,sha256=7k9iU5zAWa4ZpMZlg8KfSMi4vH3o69w5aAh5RTRNdZ4,10203
231
- snowflake/ml/modeling/framework/base.py,sha256=K6qW43lGX99a5v9qIhOTJptPrkMSzzZDa5sVgqii0dM,31359
233
+ snowflake/ml/modeling/framework/base.py,sha256=ifv9wAOhPtd8kQT2TleIV0y3Ftw3xlULyvxGolyWn7w,31439
232
234
  snowflake/ml/modeling/gaussian_process/__init__.py,sha256=rY5qSOkHj59bHiTV6LhBiEhUA0StoCb0ACNR2vkV4v0,297
233
235
  snowflake/ml/modeling/gaussian_process/gaussian_process_classifier.py,sha256=1nDZ_nda4WBHqi4tW4avY2b0fNwV5ZF-DI1-BwQjtNg,53029
234
236
  snowflake/ml/modeling/gaussian_process/gaussian_process_regressor.py,sha256=zkHptpQKCEGJspDnZtNTBswzn-0wVDrOJRJ34ach8dM,52094
@@ -328,7 +330,7 @@ snowflake/ml/modeling/neural_network/mlp_classifier.py,sha256=XYFnxnnEmMa3jlUfKb
328
330
  snowflake/ml/modeling/neural_network/mlp_regressor.py,sha256=JYb6MziLoTHJvUvnInszDG9RlK17Nxw0SglJqydaP-k,55190
329
331
  snowflake/ml/modeling/parameters/disable_distributed_hpo.py,sha256=jyjlLPrtnDSQxlTTM0ayMjWKVL_IP3snd--yeXK5htY,221
330
332
  snowflake/ml/modeling/pipeline/__init__.py,sha256=dYtqk_GD_hAAZjGfH1maWlZQ30h4hu_KGaf-_y9_AD8,298
331
- snowflake/ml/modeling/pipeline/pipeline.py,sha256=KyA9a7CTTWYaONuB-BAn7dJ1PfXFKReNBKeM398qvi4,45360
333
+ snowflake/ml/modeling/pipeline/pipeline.py,sha256=c-TbJKNlLJ7JSmFEmnVRkmrisqfvrVDhB62xHbgLEH0,46223
332
334
  snowflake/ml/modeling/preprocessing/__init__.py,sha256=dYtqk_GD_hAAZjGfH1maWlZQ30h4hu_KGaf-_y9_AD8,298
333
335
  snowflake/ml/modeling/preprocessing/binarizer.py,sha256=noHrlTqpI7RRzYbCSuCjKHxhL8NUCDKNw-kDNTwyY_U,6999
334
336
  snowflake/ml/modeling/preprocessing/k_bins_discretizer.py,sha256=g7kY0LHjnCaBzkslCkjdPV06eL2KRYwZuYKRmDef3ew,20970
@@ -370,11 +372,11 @@ snowflake/ml/registry/_schema_upgrade_plans.py,sha256=LxZNXYGjGG-NmB7w7_SxgaJpZu
370
372
  snowflake/ml/registry/_schema_version_manager.py,sha256=-9wGH-7ELSZxp7-fW7hXTMqkJSIebXdSpwwgzdvnoYs,6922
371
373
  snowflake/ml/registry/model_registry.py,sha256=x42wR2lEyW99NnG8auNPOowg34bF87ksXQqrjMFd7Pw,84795
372
374
  snowflake/ml/registry/registry.py,sha256=RxEM0xLWdF3kIPf5upJffaPPP9liNMMZOnVeSyYNIb8,10949
373
- snowflake/ml/registry/_manager/model_manager.py,sha256=LYX_nS_egwum7F_LCbz_a3hibIHOTDK8LO1DPOWxPrE,5809
375
+ snowflake/ml/registry/_manager/model_manager.py,sha256=OOXPAOL8XZGWY0YiKD8vFfJHxJl1DXjk-oISig-bxmQ,7126
374
376
  snowflake/ml/utils/connection_params.py,sha256=JRpQppuWRk6bhdLzVDhMfz3Y6yInobFNLHmIBaXD7po,8005
375
377
  snowflake/ml/utils/sparse.py,sha256=XqDQkw39Ml6YIknswdkvFIwUwBk_GBXAbP8IACfPENg,3817
376
- snowflake_ml_python-1.5.0.dist-info/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
377
- snowflake_ml_python-1.5.0.dist-info/METADATA,sha256=4_GHqJoiNYXF-WBWL6qwV6wA13ZlB6ySiGQZBLCPwRY,50050
378
- snowflake_ml_python-1.5.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
379
- snowflake_ml_python-1.5.0.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
380
- snowflake_ml_python-1.5.0.dist-info/RECORD,,
378
+ snowflake_ml_python-1.5.1.dist-info/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
379
+ snowflake_ml_python-1.5.1.dist-info/METADATA,sha256=R4CXn-UPNIGbbSaVg1CXioUMX9yIXVNeLII-aGVEzuE,52106
380
+ snowflake_ml_python-1.5.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
381
+ snowflake_ml_python-1.5.1.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
382
+ snowflake_ml_python-1.5.1.dist-info/RECORD,,
@@ -1,44 +0,0 @@
1
- import copy
2
- from typing import List
3
-
4
- from snowflake import snowpark
5
- from snowflake.ml._internal.lineage import data_source
6
-
7
-
8
- class DatasetDataFrame(snowpark.DataFrame):
9
- """
10
- Represents a lazily-evaluated dataset. It extends :class:`snowpark.DataFrame` so all
11
- :class:`snowpark.DataFrame` operations can be applied to it. It holds additional information
12
- related to the :class`Dataset`.
13
-
14
- It will be created by dataset.read.to_snowpark_dataframe() API and by the transformations
15
- that produce a new dataframe.
16
- """
17
-
18
- @staticmethod
19
- def from_dataframe(
20
- df: snowpark.DataFrame, data_sources: List[data_source.DataSource], inplace: bool = False
21
- ) -> "DatasetDataFrame":
22
- """
23
- Create a new DatasetDataFrame instance from a snowpark.DataFrame instance with
24
- additional source information.
25
-
26
- Args:
27
- df (snowpark.DataFrame): The Snowpark DataFrame to be converted.
28
- data_sources (List[DataSource]): A list of data sources to associate with the DataFrame.
29
- inplace (bool): If True, modifies the DataFrame in place; otherwise, returns a new DatasetDataFrame.
30
-
31
- Returns:
32
- DatasetDataFrame: A new or modified DatasetDataFrame depending on the 'inplace' argument.
33
- """
34
- if not inplace:
35
- df = copy.deepcopy(df)
36
- df.__class__ = DatasetDataFrame
37
- df._data_sources = data_sources # type:ignore[attr-defined]
38
- return df # type: ignore[return-value]
39
-
40
- def _get_sources(self) -> List[data_source.DataSource]:
41
- """
42
- Returns the data sources associated with the DataFrame.
43
- """
44
- return self._data_sources # type: ignore[no-any-return]