sagemaker-core 2.1.1__py3-none-any.whl → 2.3.1__py3-none-any.whl

{sagemaker_core-2.1.1.dist-info → sagemaker_core-2.3.1.dist-info}/RECORD

@@ -1,3 +1,4 @@
+sagemaker/__init__.py,sha256=wxIXz5aO38fCpEr0NNsvxyx-X8kv6trTFwDxV_knZ6E,104
 sagemaker/core/__init__.py,sha256=HFFMSe-3h2XYVO4VM_ltZzdHy3foEtnKHD01I7J7DTA,521
 sagemaker/core/_studio.py,sha256=ahBlOr6rayKxClGnpEyPJ0x8BoSKiA1nSM3cXvInxiE,3341
 sagemaker/core/_version.py,sha256=TbmhyNSaGE7fgbLKhx9olW4bV3oPxZZsxTz3wkYdsdU,327
@@ -6,7 +7,7 @@ sagemaker/core/analytics.py,sha256=sQcseHyGdazwozUANnm-gZot4lMW6keJSh9Oib-2ZEw,3
 sagemaker/core/base_deserializers.py,sha256=-90dIeUnCvZ2D1Dk7VApEuDjGiOZG-JGdGwrI5oymws,1247
 sagemaker/core/base_serializers.py,sha256=isYz_R__C74po-MAZ_0n6WlczD-DAi9darvzk7IpOJc,1231
 sagemaker/core/collection.py,sha256=xCpr1vVnONQtyGH2bQw3eFrIDn3i7FzIkH7ZsNk-yEc,18843
-sagemaker/core/common_utils.py,sha256=HARZuwlrK0VuWxb7AwzdINbvNU_aeidhyfWuhhwxrBQ,80610
+sagemaker/core/common_utils.py,sha256=oksEWtvKEesYvx70WwNJ2erZyN4a8mGnzkt8Xns4rZM,85042
 sagemaker/core/config_schema.py,sha256=1oT4_YbYyeXc2W_It6tlH4V5erqmAWpiEJaaXbuqoJc,84753
 sagemaker/core/constants.py,sha256=-wRY6RGDxoIaUTeMEr2nBfc--hptsIRvo46TTqOFvw8,2584
 sagemaker/core/content_types.py,sha256=C7Q-JGB25ozrQlnGw_mXjnA_mP4bSbzTCzJssCOuksc,6413
@@ -15,16 +16,16 @@ sagemaker/core/drift_check_baselines.py,sha256=p8PQkUQMFsZOEuPx_Q-TXM5xnDZ1QHqnr
 sagemaker/core/enums.py,sha256=V6wacThvAs2ZZh-0InTrAe4fejT8AsNqlhukLWNce5s,1687
 sagemaker/core/environment_variables.py,sha256=d3xrN16DPYpcN9BtGJxewFiolQ7tFLyXedNeKnc6DGM,4977
 sagemaker/core/exceptions.py,sha256=NGRkmdGGh2ZlpOZPprbCkJYJn4jMCvQw9UYZkFD3Vzk,3667
-sagemaker/core/fw_utils.py,sha256=y67CnGy9P4TZx1P4_MOA--KcbwmK1-e6x9fXWoSECJI,44965
-sagemaker/core/git_utils.py,sha256=W_gbzhUnq-S1_vc4Ae_WMIDqaiOBgPx8RSMDTa-GfRs,15551
+sagemaker/core/fw_utils.py,sha256=_dyCGwmQ4VlVgD5BCJL6aD0RcWH-_a_ep-4M-ygrSa0,46428
+sagemaker/core/git_utils.py,sha256=alQYwXkm69jkVC1aBTlqvX-z-P3of50HBvrJnrtA3HA,18184
 sagemaker/core/hyperparameters.py,sha256=qeaRPSL_IsAcpVw8idaM2W01D-gfx680IKsplthPPkU,8581
-sagemaker/core/image_uris.py,sha256=0PPI9mXXD8_Q6FFR_7Wm-3FmVZf7-Ei-GjVyg3pRhYg,34265
+sagemaker/core/image_uris.py,sha256=FeGlgMwYad6SupKvVTausaKt9VLevboUdkpkqpk0BWk,34470
 sagemaker/core/inference_config.py,sha256=LEJ5ei9ha30WT_tKId_BL6XyWiWFayO74oWwR2YX5Lw,7007
 sagemaker/core/inputs.py,sha256=3CPYESbuC5NwFUkv82fMG6k4V5xR6NQVmJhD4VGqoMQ,17976
 sagemaker/core/instance_group.py,sha256=faSzt3rHRgQl8w1PYKxz1hb31xG2KznvuukNpg3cWv8,2522
 sagemaker/core/instance_types.py,sha256=0J9vHGYVaiqCuVpohw1AnXlSc2O_PYp6u7Rzc9-iZFo,7666
 sagemaker/core/instance_types_gpu_info.py,sha256=h2DbsA7VkiWkhp4fEoaodsBamFxGwJXUHe6sof8oCfg,1572
-sagemaker/core/iterators.py,sha256=QdA1b5Ni1UwCJUZywGH6BiocgtrPt9Va-i_M33IjwaM,7269
+sagemaker/core/iterators.py,sha256=ch23c9fEaHQEjrt9SQ_vYwwiFOybR7MChRWSRRnG36A,7810
 sagemaker/core/job.py,sha256=uobdRe57YhzVxyXDR7ITGt_dG8hf9yGHO3wTLnxAmw4,14063
 sagemaker/core/lambda_helper.py,sha256=UtB1AlU2GFxfzBaNRB5vyAxbEv4oTIhvt9UBWSWFX5Q,12501
 sagemaker/core/logs.py,sha256=Xx4EEf8J0oS9it_nyJgFe4DsnSooWfnbSrvcF_Y0m8E,5917
@@ -32,12 +33,12 @@ sagemaker/core/metadata_properties.py,sha256=N9LIkaUwaqeyBTKRwGr0h06Xw282S_q5Tly
 sagemaker/core/metric_definitions.py,sha256=rzPVEKNwfWeZ2UvGq1NRmQtt2bCcTAr_vBkLcrs54o4,4393
 sagemaker/core/model_life_cycle.py,sha256=Mhjz-ETubuNaBnyw6-7l4XoJMfy4c3jw1GGYcjRruiw,2042
 sagemaker/core/model_metrics.py,sha256=eyPL67aGQJm2l8XNrSnivqKPoyWCOqxM1EpVkgtbdHE,7155
-sagemaker/core/model_registry.py,sha256=JNwErFyo_GshlRuE6llIctxMnWwu7pYmEAtimDTvzAA,20879
+sagemaker/core/model_registry.py,sha256=O6qepWaXPlwJvVnYCBWLDRlYj6EzE0RTKexmTaZ485A,20898
 sagemaker/core/model_uris.py,sha256=1SdcZE1Hhd12dMpJhfNuDRq9qsTcALLfklI28lCtLm4,4695
 sagemaker/core/network.py,sha256=f31RvflBvGBvMLqm55d99qFG8gdmA4UQrm0FJGQ2KwQ,7418
 sagemaker/core/parameter.py,sha256=5cq8bvJymrAW-WebRGE76GaH7LaYKyvb_vKwHOGaUL8,5748
 sagemaker/core/payloads.py,sha256=FFSwfPoRdJRZWZa1YCEf75kv6Go5uYlHSQj_Ie-pzgE,8720
-sagemaker/core/processing.py,sha256=RkzA9Bj3Cf0QN7D1yO08PEiblDwvvv1y1THdnwCpVaA,70908
+sagemaker/core/processing.py,sha256=Dm6EF7LIUJWCp7I5RMN7RagC3of098f7d4_Rd_bquQc,71034
 sagemaker/core/resource_requirements.py,sha256=2UOYq6TWzNMrE2g8YsHomrWkHGdk02IS5JtMWrQwn50,7039
 sagemaker/core/resources.py,sha256=iiI5XtEjH0VjDstH3RsSISD0oUWmFXB8ZRjAAKVgY0Q,1775863
 sagemaker/core/script_uris.py,sha256=HcnlX2OLEXaBkdv9PdoDD2_sfWbdbrYVCOozQBAGeYg,4439
@@ -76,7 +77,7 @@ sagemaker/core/experiments/_helper.py,sha256=IEwuCAXJWB3wtwrj_W74W0zOg4gVVrZ79RK
 sagemaker/core/experiments/_metrics.py,sha256=Frw_7pUqW_LG35O0OwZQwfeCzgnc1057KllebVxVbwY,11767
 sagemaker/core/experiments/_run_context.py,sha256=8VJkbV50TpzJ8XU9kyhkfD0EtMTJ4jXwi7WgEBAOZGg,1757
 sagemaker/core/experiments/_utils.py,sha256=orfWE-nweLLOaVKvvFypTn-iq7ObckZRpjAc4weEJ0g,7934
-sagemaker/core/experiments/experiment.py,sha256=1u2nY1X2Hojn-2AooV-P8yaYtYkT-7rq-3Af0LGwpJg,10223
+sagemaker/core/experiments/experiment.py,sha256=9hhlR5qCnB9WLfMswEwB7Y-zlED7qrI-SGyWsUKlNVs,10430
 sagemaker/core/experiments/run.py,sha256=Wl3mcD_4plglSprcpdecscQ0YaUZlliZOGvcL0vsYyM,38814
 sagemaker/core/experiments/trial.py,sha256=a24qakHEfbcnbhvReGDyj1s7g4cj_ilM2U0oBTsKljc,12372
 sagemaker/core/experiments/trial_component.py,sha256=02yv0YFUgtW4bt38Mc0-0ajkFyGCLgPMOO1wu2x6xYU,17114
@@ -85,17 +86,11 @@ sagemaker/core/explainer/clarify_explainer_config.py,sha256=10fMeqVVX6fWEDVUVt3G
 sagemaker/core/explainer/explainer_config.py,sha256=h5IzOnmvCErFk8mlGf1GT9ojK2ZpQ7KScuz5jCuT-Y8,1763
 sagemaker/core/helper/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sagemaker/core/helper/pipeline_variable.py,sha256=K1sd7-tx8QfKFy6okox62KgSoLqdQn4XLtTGI8WzxzU,3048
-sagemaker/core/helper/session_helper.py,sha256=0Wb0gDMfDejJ-CtX6mpZ1zmr7MY8ld7NJIo1rOwU89k,124019
-sagemaker/core/huggingface/__init__.py,sha256=zyRugEr9PJEfOkgLj_tE9_YSjnI9Njk3B46QNqspQgg,1284
-sagemaker/core/huggingface/llm_utils.py,sha256=7STooczHw80_0180_EH2E8m4LBG3zwHErZXGvZMQ2tA,5588
-sagemaker/core/huggingface/processing.py,sha256=LzdE-f3loGj49zjQJJ4P-eDIAZO9bUPrv2z5eCIgQps,6033
-sagemaker/core/huggingface/training_compiler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sagemaker/core/huggingface/training_compiler/config.py,sha256=67cJSB01-ztcl-zfaD1wfLm_IHceQ-KyVi0Cllq-xzw,8019
+sagemaker/core/helper/session_helper.py,sha256=PpF5csI2a6A1G1u4A4PxEDvyv0n3-pCQqrtLlsEN-HM,124527
 sagemaker/core/image_retriever/__init__.py,sha256=Jn3TWcihenajRyEbKATTBRGjTJi039CnrswsSKcLbwI,128
 sagemaker/core/image_retriever/image_retriever.py,sha256=EPMi7rj2boMPcjnMoQs-TXCd7I2oKTGyALVb1eGJC0A,30205
-sagemaker/core/image_retriever/image_retriever_utils.py,sha256=RLkbNFibqEOonz1pwQn3n1eUOMSXfY0s59AXAuJxXRg,20148
+sagemaker/core/image_retriever/image_retriever_utils.py,sha256=l6X9boAcczEIaPl88i87Ey328kwL7Q5ZvYw1cDdVlxM,20091
 sagemaker/core/image_retriever/test.py,sha256=ZFYrI7kR7Q4tSxb8W7Lya39wjkvdDKmEAiTBelnk9Ug,89
-sagemaker/core/image_uri_config/__init__.py,sha256=GAgqnhmt2Swv6XnSeXMh6CHqgHRUsBvZ8o7vF1g38O0,596
 sagemaker/core/image_uri_config/autogluon.json,sha256=xwUB37qGyClWYwv5Y0q2WAi5-ZRKdC6Iw-3VoIvoRwI,58839
 sagemaker/core/image_uri_config/blazingtext.json,sha256=Q7zida81k5uv066x_PCYrb-YEsZc3g8oExp_4WMAt_E,1952
 sagemaker/core/image_uri_config/chainer.json,sha256=zaxwUPCQa8zZ6O28Uohjc81dssxdDz8etGXQi_76tNw,4212
@@ -112,14 +107,15 @@ sagemaker/core/image_uri_config/djl-neuronx.json,sha256=QF-b0RJjzl9A-ieHJjjk46e8
 sagemaker/core/image_uri_config/djl-tensorrtllm.json,sha256=enITTkw0jMpvOjxv7QC1bV7U7rzkwIDdaspUK1Xs_hs,11397
 sagemaker/core/image_uri_config/factorization-machines.json,sha256=kAZpjPj5jkOsGU8bPWllAhovmDXTltWSgCvaAJvXV6s,1963
 sagemaker/core/image_uri_config/forecasting-deepar.json,sha256=0EfKpdqRSJKuLfHKSUiQ2CBN8tXDUjeErchHmDN95F4,1959
-sagemaker/core/image_uri_config/huggingface-llm-neuronx.json,sha256=6Zmxu1BCmS4FVknqwSvJNPJBh4djhPzrjbsuroKywp4,30847
-sagemaker/core/image_uri_config/huggingface-llm.json,sha256=EmrxZl2zhAZV29_huB-_78aYtOya0z3WkCG4AjIJ7dw,54159
+sagemaker/core/image_uri_config/huggingface-llm-neuronx.json,sha256=m2qTC2wOe7qXbRK7Giy4VZRY0UdxyoGYlLrlZ1xECXo,36009
+sagemaker/core/image_uri_config/huggingface-llm.json,sha256=nQwCZkBgbG-MH1BXFEdLsv8Dj1KOlo1B3Sq9Ut5fy3o,59295
 sagemaker/core/image_uri_config/huggingface-neuron.json,sha256=B4JqfOSugs9HW2T22eNPjcuzqi7PVxc52UZ3bOB59vY,1964
-sagemaker/core/image_uri_config/huggingface-neuronx.json,sha256=GEyOb--IEUztV1dVvzDoIPwfHeuwDrI_LEp36f5oFvY,23088
+sagemaker/core/image_uri_config/huggingface-neuronx.json,sha256=vIP1djtJDzjmDcwT028oKon0hydYXFPHiOrR-QMRxME,31034
 sagemaker/core/image_uri_config/huggingface-tei-cpu.json,sha256=pd7POB4Gk_Dq_YTMhuQMdTukbeydLKXV11zUxovW_Pc,13187
 sagemaker/core/image_uri_config/huggingface-tei.json,sha256=6aBUBOBQn-0G_MBps_cpAGJNMT-pRUvubSuMEvnWrQc,13204
 sagemaker/core/image_uri_config/huggingface-training-compiler.json,sha256=DnCPhKPlQtEiUg6V2RNUzkbLe-gAr_YKJq7USQVSW8I,8738
-sagemaker/core/image_uri_config/huggingface.json,sha256=0-CVtLHsywJNJihVFWQZPz2dduX62_fVfnefkgxH1HM,104220
+sagemaker/core/image_uri_config/huggingface-vllm-neuronx.json,sha256=gy6eqJkKyfl2YZtb2JUo5qA39aMTdqRnobwDf10YLF8,1198
+sagemaker/core/image_uri_config/huggingface.json,sha256=tus0nYBzwY-alS0sF2F1fnVjHEZ5Ursv5dykOKJwVso,111302
 sagemaker/core/image_uri_config/hyperpod-recipes-neuron.json,sha256=PTAIWLrTJ3DTJyODjf2T6i_QkX7-nhA2-k3rRd8g3h4,1924
 sagemaker/core/image_uri_config/image-classification-neo.json,sha256=XYf76nHUA42dXUQWJJJy4O_8-EIR_d0J5Igj_79PXo8,1665
 sagemaker/core/image_uri_config/image-classification.json,sha256=PDK6ISr9dzWwVVPz7XeTZCs5T2ztVB1z4fmgQoCX3-g,1961
@@ -151,17 +147,17 @@ sagemaker/core/image_uri_config/ray-tensorflow.json,sha256=WnidbAzGY9Xz5pijqKrcn
 sagemaker/core/image_uri_config/sagemaker-base-python.json,sha256=4S4fx97mtrKgY1kui70MKGcHjZoW0Tt1DRmbWWBBHRs,1900
 sagemaker/core/image_uri_config/sagemaker-distribution.json,sha256=D15rjPfuXQ_1A0q0jPi7Lbryri_6Wd0jp1941JW3aaw,1359
 sagemaker/core/image_uri_config/sagemaker-geospatial.json,sha256=onitbqKrsxP9FiLYv5-YHNxWw57lKJJdfBviekLtbow,251
-sagemaker/core/image_uri_config/sagemaker-tritonserver.json,sha256=qES0aX22zuaodEd75H4A44F-zNcLADnimB4jGnmyNWM,8848
+sagemaker/core/image_uri_config/sagemaker-tritonserver.json,sha256=YI7ielrEek9YcTpCsDQMDjESd35NeSDF0y6DWeedEuA,10616
 sagemaker/core/image_uri_config/semantic-segmentation.json,sha256=F7CmcedwppBwAt8bJesP1MCFfkQo9ISFlloxpz3Bso8,1962
 sagemaker/core/image_uri_config/seq2seq.json,sha256=CJOrRAImaXXSGtxkh_vFQqYD93QkqTTLJYUt1DO06R4,1948
-sagemaker/core/image_uri_config/sklearn.json,sha256=WyJWAATV8A7PxvjsXvzi5TOpcbkh225nqR2mEceyRTU,20003
+sagemaker/core/image_uri_config/sklearn.json,sha256=aMNGYNDxQPV1rE9HZB1xKmbC12LcIfJG2cQn74wtUpA,22205
 sagemaker/core/image_uri_config/spark.json,sha256=-Pxku5AiFvmUqgOpqWFCRqMk4tVyZOTE5jOEnbZ3ET4,12868
 sagemaker/core/image_uri_config/sparkml-serving.json,sha256=HzMRrp1mbyF_VNcnJ9T83FBQvicWWgBsLDoixVCkNNA,4052
 sagemaker/core/image_uri_config/stabilityai.json,sha256=MBzBi0bg3IxVgv__x08NBvNdKgR69n17OYm9iCwsDV4,2050
 sagemaker/core/image_uri_config/tensorflow.json,sha256=_KBm-dRO7Xzry9oDKmNRuEqFi8wj_Ga7997vcyOdGJE,240040
 sagemaker/core/image_uri_config/vw.json,sha256=-MfHhVlpEDxPWxStlL-CmpcBNHcMBieU_NEts09BDKI,857
 sagemaker/core/image_uri_config/xgboost-neo.json,sha256=HYlq52lyItrApOvtx6Ew4J-XBDS0YusIfvAwinrRwsA,1652
-sagemaker/core/image_uri_config/xgboost.json,sha256=PcEHwLtCbyBq0P26NEWAece7y9nm3sRBdzn7PKUa06w,41639
+sagemaker/core/image_uri_config/xgboost.json,sha256=3eR3TzerFVhkH2h6XFhLIr7Emvumg-vsdOj7t7l--fQ,45723
 sagemaker/core/inference_recommender/__init__.py,sha256=nty_FWYBcHR8aodym2YtiiyVtyujQxMSmfaoewAbUa0,801
 sagemaker/core/inference_recommender/inference_recommender_mixin.py,sha256=BTA7lUCAPxCxdZFtivb34DeoR6orP9pS4cdKH5Errxk,28741
 sagemaker/core/interactive_apps/__init__.py,sha256=KrgA4M5RnLip8ClNHwC7KxSnHu6D9oe1aSOWh8YR_Is,1272
@@ -178,11 +174,11 @@ sagemaker/core/jumpstart/document.py,sha256=ek3fJ0p2l9zTCmpnmghi4Yi852j0oCGj6Usg
 sagemaker/core/jumpstart/enums.py,sha256=2BkIRRRNnemLzdmG3XMUuzVr8FWhI33ZI3FXtgz8iZ4,4872
 sagemaker/core/jumpstart/exceptions.py,sha256=f7HXiSYdfQwpqHRVCIjRUDFCRTliJWx0cBFjDYRhURU,9262
 sagemaker/core/jumpstart/filters.py,sha256=xe53UlzSVCFt-FLKWbaq09XTYmpV4MMHn-eF_wTR2es,20982
-sagemaker/core/jumpstart/models.py,sha256=BvvytN1Kvcu1Dqle4LscUDa71PwAocMGP5t5pG8bee0,14636
+sagemaker/core/jumpstart/models.py,sha256=bYOnTz_dfSisCnLNAk7YfNp9JvaHpt9g7OTKHtpI_pk,14766
 sagemaker/core/jumpstart/notebook_utils.py,sha256=i2dIlISgzFUoYeILN5Tpyc_jyKH9PpqbRiol6dTMeQA,23295
 sagemaker/core/jumpstart/parameters.py,sha256=opIIW_oU7O72vq_SlBkKoMExqIYHnUbHRXJ07-ovJZo,948
 sagemaker/core/jumpstart/payload_utils.py,sha256=bqLUYBLxaJ6sD7n4OREBgq7MhEa8f61oVWxI7Z8F5kc,9380
-sagemaker/core/jumpstart/region_config.json,sha256=sAiSeU09T8JsQlkVuiKdHxeI1uDagpsQKjTjZL8rfaE,7167
+sagemaker/core/jumpstart/region_config.json,sha256=yi51K_wj6-GM0dZ23TglJl7lU0Xmn0cDFF3-_XSEW9w,7500
 sagemaker/core/jumpstart/search.py,sha256=KwduhuER3_Ul7roLewBVBEYw8YZSlXW7K49VFQyx-Mw,6135
 sagemaker/core/jumpstart/serializers.py,sha256=2jOUfK4_msGeQqSuQCR-sXQx5UHc1U1Z8DBnfsVT7OM,3935
 sagemaker/core/jumpstart/session_utils.py,sha256=TARN-sAyubafGpj_KX43N6MPrKU4NDFLbeACS07B88M,9169
@@ -225,17 +221,17 @@ sagemaker/core/lineage/lineage_trial_component.py,sha256=ou_SkNTTRDo9BoVmsVfPxJD
 sagemaker/core/lineage/query.py,sha256=aH0aeEyF8doBM0zQ-_UV-PjyU3Pf8N3fQ4ZOsPcDdVM,27066
 sagemaker/core/lineage/visualizer.py,sha256=MY3tCTZgeXNxUZ_m-aZhJqV6Dw12fpkk2qbQXA2lQ_g,13246
 sagemaker/core/local/__init__.py,sha256=Dx1VYzVS--FYew8A-CwWsGvB3_18qXgwG9cet7U85jc,744
-sagemaker/core/local/data.py,sha256=uIpjWyPL3MPTFoanYczhcy0eM6oXi0rM2xvb0wdHudA,13205
+sagemaker/core/local/data.py,sha256=RpEBEkuKpVbQLRJ5VMoO_sMA8OFjceo3fZls0002fiw,13675
 sagemaker/core/local/entities.py,sha256=ARxpUteG-7Nla1nG98ALbStVD2yNK92GkNtItSwKIFY,25904
 sagemaker/core/local/exceptions.py,sha256=hjcRgJk9TtBBWVpSrByurrnERwrvJ3e5NOrMcPnIdpk,779
 sagemaker/core/local/image.py,sha256=0jiA5HqfbQWo9BqufnEODCBgG1yLjb37u9DxjfkHoYM,47218
 sagemaker/core/local/local_session.py,sha256=hAAzjGhyWhoP5gPMqwKnTECYt4e-r14gjOchppgxHAM,26829
-sagemaker/core/local/utils.py,sha256=HgWjDp_TI2YMp13q7xrLf3TTkbVsM3Tmo24fJvrSv_A,8177
+sagemaker/core/local/utils.py,sha256=trDakVtOpAqnZyAni-R0WgeXlXng564X9dFBtwmIzrc,8196
 sagemaker/core/mlflow/__init__.py,sha256=fe-GZTujt4kzM6qqkc-7gEOxL6a_6M-VnUy3VfRGCUg,1372
 sagemaker/core/mlflow/forward_sagemaker_metrics.py,sha256=3_kpOWCof6igCHpXg43cqTBIuYh3r644FESM5k4nB8E,1648
 sagemaker/core/model_card/__init__.py,sha256=A6zkr52JZWKDdIoH-tF5-0ni3iQ0gYRrVmW4pqz2c3M,986
 sagemaker/core/model_monitor/__init__.py,sha256=3ls5dUkIapv4AStR7-AjWqpIWU-_-IjztOoNxH_kiYo,3433
-sagemaker/core/model_monitor/clarify_model_monitoring.py,sha256=m2asOJkwqwVNCOGFXkoJygznhh8Oi5M3faEO4IzmaTI,69520
+sagemaker/core/model_monitor/clarify_model_monitoring.py,sha256=jUlaBeRJVc4Ox8TP5hl67QOrNcG0IjqsIevKZuDTm2o,69650
 sagemaker/core/model_monitor/cron_expression_generator.py,sha256=4A05akhzppenJQHPt1xY2pJsbJ1lcOK3G3fuY5m6sTg,2930
 sagemaker/core/model_monitor/data_capture_config.py,sha256=f7lbWdEwlQwvzoQEDh6BCmVgznDlc2pWT7re0cjkrPQ,5176
 sagemaker/core/model_monitor/data_quality_monitoring_config.py,sha256=Jg5Pd06ZlYUtIQ3Qg8Sgz9k6kcip2NVNchsMhGsJnz8,2486
@@ -245,7 +241,7 @@ sagemaker/core/model_monitor/monitoring_alert.py,sha256=r6AFJi7dt_lFLWsicKv9KlUM
 sagemaker/core/model_monitor/monitoring_files.py,sha256=Bpo9pEtVL7e_osQcYnbuziX9HOaI2uBkCWiQ6jYF3us,20524
 sagemaker/core/model_monitor/utils.py,sha256=jCBfB-Jj12W4JsMjS8jD04etBijR6v-1kIgAsWUZbYY,34089
 sagemaker/core/modules/__init__.py,sha256=SSZBrimc63iqSPjSjR4Lx-iTluu8znM2QQZioudE1Zs,829
-sagemaker/core/modules/configs.py,sha256=zuv5lDhbyd1rjf8cytlCKrIKBEOrT_xwpOeqO-ogd4Y,9136
+sagemaker/core/modules/configs.py,sha256=xq4wv0ln_mew4lI1DkUQQ6XBvoEdaS7hWMJ7da-XUUI,9636
 sagemaker/core/modules/constants.py,sha256=hiViW1H2tzmiLrsrhe4UvLJEKGCJ0WqiG3vb1t1E-wE,1254
 sagemaker/core/modules/distributed.py,sha256=ZCm3RF3UgnrtoE-Zcjxa_X1RgR12-1ba4f2-SSIMvSk,6833
 sagemaker/core/modules/templates.py,sha256=7VrkG6axyqBr-Xx0fuwFbyLfrxU9snQpxyUL46BjLp8,2320
@@ -256,7 +252,7 @@ sagemaker/core/modules/local_core/local_container.py,sha256=ZWF9-WuCS5izqA4UXRR3
 sagemaker/core/modules/train/__init__.py,sha256=04hUL2wiHGXWHpHD9vtlpTA1YHEJvN3Ja3aaf1YSLXY,640
 sagemaker/core/modules/train/container_drivers/__init__.py,sha256=hFRmz5Ch3wx0a9vpgVqyXozipjZ2TSKvb20-X1sP26o,652
 sagemaker/core/modules/train/container_drivers/common/__init__.py,sha256=W_hcSzzOvV8oNNWzivU6vZqv3jYd7cSkKDaQXjlf8es,661
-sagemaker/core/modules/train/container_drivers/common/utils.py,sha256=Z2heyOBmLhE3rqsy3L6qolEBdLXxmSqXHtk-j98AC_Y,6981
+sagemaker/core/modules/train/container_drivers/common/utils.py,sha256=rsx1VD-ixzeKCf1ZVDFUFE8ZwMDpOJx0_hi9XncHAqY,6698
 sagemaker/core/modules/train/container_drivers/distributed_drivers/__init__.py,sha256=P_9zjCEBEpSJ_hmU0661joL8t3ARn4qovOENqCZczj4,662
 sagemaker/core/modules/train/container_drivers/distributed_drivers/basic_script_driver.py,sha256=gcSKwb_laf8_bZ3s1eFjEujLlL-ujtjan3OLd6LrJmY,2427
 sagemaker/core/modules/train/container_drivers/distributed_drivers/mpi_driver.py,sha256=ikQO52Co16nxxXPHQ7NkoH28MHwsMjpYvQ9Rd8tmawE,4266
@@ -265,21 +261,21 @@ sagemaker/core/modules/train/container_drivers/distributed_drivers/torchrun_driv
 sagemaker/core/modules/train/container_drivers/scripts/__init__.py,sha256=j-mHREHwHYpuAr9egOazGgT-a6MPfxnCFmH4rtqHwTY,662
 sagemaker/core/modules/train/container_drivers/scripts/environment.py,sha256=IseQtc0qnqvdlRKiostC4OOMVmLgJpO1ai9MW6E6ulw,11070
 sagemaker/core/modules/train/sm_recipes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sagemaker/core/modules/train/sm_recipes/utils.py,sha256=ec9SbavQIcidJ5zmWEJk640-5oATNJ_WMlNY9_-8r1o,11725
+sagemaker/core/modules/train/sm_recipes/utils.py,sha256=E-ToK3koYTGuJZ7zueodB4TUASOx-mLfbyxmH2F3c5w,11722
 sagemaker/core/remote_function/__init__.py,sha256=A1vDi5kiaPuwrGA6yS4VapWRWMeorDtSYYtwH1KM7G0,1035
 sagemaker/core/remote_function/checkpoint_location.py,sha256=6DMXQ6TOnYcMjox__YXUB3kMtmoJ9l6fDxdnPtucwkE,1894
-sagemaker/core/remote_function/client.py,sha256=aS-OKoyIaok2XNBFcBIYXxT3-D49QTwBfQSEgWyE83g,60934
+sagemaker/core/remote_function/client.py,sha256=D-doUzTS25eelxL-OhP6QJzTRgd2Hl7JO3HgnXK7Z9o,61650
 sagemaker/core/remote_function/custom_file_filter.py,sha256=01YUOG1lrFsD6Tti2I7YyJPabF8Bw4181y1Ep1Kb6EA,4549
-sagemaker/core/remote_function/errors.py,sha256=fzhFFItk7NWDYlvjmZg1wl_fbVx5In-_-ODzKXSL5H8,3616
-sagemaker/core/remote_function/invoke_function.py,sha256=qevwl_G33VGPtVDN81-0QAxKv8q0W_rzbqUjNWbITco,5670
-sagemaker/core/remote_function/job.py,sha256=u5O5okPR5-2zmohfT7kt347fH5nU_mKrzBcSoHiH0TE,87986
+sagemaker/core/remote_function/errors.py,sha256=8Zgyp5EWI-J8L6_ValeGAtOKV4YxXs3w-voqeh-f5N4,3494
+sagemaker/core/remote_function/invoke_function.py,sha256=HGL9xTnXaVeL2e5ALeVagQDeuakBLRSavc7UebqhmOk,5511
+sagemaker/core/remote_function/job.py,sha256=0enllsj62GN_pl-oSXHpxyA6A_IWfD2npUc2L-_8OGM,86992
 sagemaker/core/remote_function/logging_config.py,sha256=e6C8w_CR2iJh5QLazeYZAmTjOgF7zAz1IGAPk1MGIy8,1387
 sagemaker/core/remote_function/spark_config.py,sha256=3jhyNOmeEvS9K7-trneFOpHzdq6yHEU_p4yufTZnL90,6298
 sagemaker/core/remote_function/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sagemaker/core/remote_function/core/_custom_dispatch_table.py,sha256=KuJL6gJxFgx4OqPDo0LWK97vQZ6EUdS_nOi3P-Y-3NE,2659
-sagemaker/core/remote_function/core/pipeline_variables.py,sha256=NAidYz2p6p7dA8VMRIdjKadSptHNmtFFp3OuiPehx6E,11791
-sagemaker/core/remote_function/core/serialization.py,sha256=Ia4hX-hLROoU1nukusXqNZxj7ASuhY73M3uTLXQ4vCc,16275
-sagemaker/core/remote_function/core/stored_function.py,sha256=ceXA2C28fooWI8xJPfmUOGwkzfUgTvUvJUTQ9vgvrgA,8284
+sagemaker/core/remote_function/core/pipeline_variables.py,sha256=P_7Nzg0c9r02g5dsOnDXRgj7jfC2O8m39YW_bwl0QYU,11503
+sagemaker/core/remote_function/core/serialization.py,sha256=9GKGHUsZ0uwKVis3SyqjPYhTBCyoS71qryZERkP4ooc,15243
+sagemaker/core/remote_function/core/stored_function.py,sha256=7q0eV2Q0pF83woSh-QP82Xdr_so-NA-5b7XfaLos3fA,7950
 sagemaker/core/remote_function/runtime_environment/__init__.py,sha256=I3CZULFeDwHwjPUzq3xk_ZrWh6x8d-kO6kz-rX6_sO4,652
 sagemaker/core/remote_function/runtime_environment/bootstrap_runtime_environment.py,sha256=eSUDdN1U9leS4Gv-i-ZJ8TKtnkfqBL6iaEB4E6Xgv7M,22122
 sagemaker/core/remote_function/runtime_environment/mpi_utils_remote.py,sha256=wqNH5Zuc5WB0rPNy3sf3szszolkbGJElEveh4FHkSMw,8726
@@ -299,8 +295,8 @@ sagemaker/core/spark/__init__.py,sha256=RISFq3KL40KRGkWsk813EqauI0FcjeCrsIpn1lS1
 sagemaker/core/spark/defaults.py,sha256=Ga7OzEuG29qT-78eCeW_kmI2gP__746Ew6xoRfNPoRM,671
 sagemaker/core/spark/processing.py,sha256=Yf1ueSAJ57kY8rGrPEDU39cQAdPxcztV-Oi4YciKif8,61569
 sagemaker/core/telemetry/__init__.py,sha256=fgtTWBIe2a3zZ3lXnVbFN-5Uxsy2LvMyEAKUcj-eak0,912
-sagemaker/core/telemetry/constants.py,sha256=TZgwEhDO3-LxIGAO4RBIPZEk6cFUXffL58rgjxB3PNg,2744
-sagemaker/core/telemetry/telemetry_logging.py,sha256=TIVf4X6ZhTPJ1rUQx8MJwbUIAgMb3zL5TH3LdFaKY84,10379
+sagemaker/core/telemetry/constants.py,sha256=dSKj_fFqbR_vD7xHRSCU2ahiFS3esL_PQ0qrwrKZxsg,2596
+sagemaker/core/telemetry/telemetry_logging.py,sha256=IMS_jfY6GLpk1FiYyWPGhjO1ZUgK3dRX5Sx_G1WgRFM,10412
 sagemaker/core/tools/__init__.py,sha256=Wv7J6XW66GOYhUytyQ32XZHW0TbfS11mLH8hl7L4Fuo,70
 sagemaker/core/tools/codegen.py,sha256=MMSmzm1jXg09lPfcrxDo2VhHUTUQbhdjlthhfubDGx8,2006
 sagemaker/core/tools/constants.py,sha256=Li4DCfEto3_CWM9LV-_PlweEKU42GiYbKcDtlilriYQ,3728
@@ -313,7 +309,7 @@ sagemaker/core/tools/shapes_extractor.py,sha256=vxGxdvstoJ8xoNKX8npT6aixGkHntt_P
 sagemaker/core/tools/templates.py,sha256=lzckRzlX3Ht8j3H4SYp79P2B4q5M9jX1swSuKC_t8EU,24676
 sagemaker/core/tools/model_card/generate_model_card_from_schema.py,sha256=wHZGKMWttL-lYSdMjyuh0YO569hkemNsB2ICq6wscR8,22622
 sagemaker/core/training/__init__.py,sha256=mnnemiOpx61sdFlJNKBlRcRAaMoKIImCdM67eqeNhB8,643
-sagemaker/core/training/configs.py,sha256=ZUjem52iJj-OSZCMnimdHARMNFNLjW8mDUzQeFOBVrM,13932
+sagemaker/core/training/configs.py,sha256=E35rVbDSFw3UGKBV2t_FSe-n2Y0p0eNmuybsRsQzF_w,14507
 sagemaker/core/training/constants.py,sha256=neypgR0hRAzJ7vvY0mxoww0H9NTXgv1qq6DboY6M2vo,1248
 sagemaker/core/training/utils.py,sha256=EiGf0Xv9OKCi8JrzetEkCE2fQszAm8s6FrUnMxYhQbw,2632
 sagemaker/core/training_compiler/__init__.py,sha256=0DNGNiUXdmvJ6cywPSyE4Nxrd2hsL8IATSrbYrfzqkU,753
@@ -342,14 +338,14 @@ sagemaker/core/workflow/pipeline_context.py,sha256=PPDXxp0TYhd_UmShwKRwIauMDiUhm
 sagemaker/core/workflow/pipeline_definition_config.py,sha256=h7peb05tolLC6YK0dZ3ucr3J7UnA4loGNZU-FyyFLCo,1241
 sagemaker/core/workflow/properties.py,sha256=Sn1n53hF3RFhFMY-mtn6Md4nyEqbX--kI7v-Bbeji_o,10538
 sagemaker/core/workflow/step_outputs.py,sha256=iaM-iQPa-SjwrF1_EYJb0W1oj1eCTVj4RyyW4sgLGhw,1891
-sagemaker/core/workflow/utilities.py,sha256=vlgh1w3E9B25p3n0MFzun1Jrq9ofv-xovW4tcEvIWb0,19211
+sagemaker/core/workflow/utilities.py,sha256=oGsYI4Y5D_Lz2Lkqb1zUif3ygizlR1l09sZzQfHUVDk,19386
 sagemaker/lineage/__init__.py,sha256=Mvz5zBHiXnPWFsP73xS02mnUWuI69ZsiMo5MYKu4QZE,1211
 sagemaker/lineage/action.py,sha256=riSefnRx-Tq0FgOqTaa2rMzYnCgPFLAvU3YIeTN6H_U,976
 sagemaker/lineage/artifact.py,sha256=HhKq36HWaEFDRiwGtJVhgMaz8ARIYCVY9cs0hnGerqc,986
 sagemaker/lineage/context.py,sha256=eNgpEhvsWZgf6VMCJVbAn9LFMyz_GumYQkDi2kVHnoQ,981
 sagemaker/lineage/lineage_trial_component.py,sha256=x9vzNn5hNb9FWyn0EipZyUKTclv-8_X9Q86bUXWNL_Q,1061
-sagemaker_core-2.1.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-2.1.1.dist-info/METADATA,sha256=iIg62XgekXjIvAapnfprAOY_XhRbMuXdKRa2e_p1cLo,5521
-sagemaker_core-2.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sagemaker_core-2.1.1.dist-info/top_level.txt,sha256=dzCY2cZ4aU_S6NM-0PxIEkCsiZXnzUM2UQbHsuyIU1c,10
-sagemaker_core-2.1.1.dist-info/RECORD,,
+sagemaker_core-2.3.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sagemaker_core-2.3.1.dist-info/METADATA,sha256=TzSISKAA5mAyxtU99UZ8AG0Kh-xfDsZnNacmuzI7YuU,5521
+sagemaker_core-2.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sagemaker_core-2.3.1.dist-info/top_level.txt,sha256=dzCY2cZ4aU_S6NM-0PxIEkCsiZXnzUM2UQbHsuyIU1c,10
+sagemaker_core-2.3.1.dist-info/RECORD,,

sagemaker/core/huggingface/__init__.py

@@ -1,29 +0,0 @@
-# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License"). You
-# may not use this file except in compliance with the License. A copy of
-# the License is located at
-#
-#     http://aws.amazon.com/apache2.0/
-#
-# or in the "license" file accompanying this file. This file is
-# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
-# ANY KIND, either express or implied. See the License for the specific
-# language governing permissions and limitations under the License.
-"""HuggingFace framework support for SageMaker."""
-from __future__ import absolute_import
-
-from sagemaker.core.huggingface.estimator import HuggingFace  # noqa: F401
-from sagemaker.core.huggingface.llm_utils import get_huggingface_llm_image_uri  # noqa: F401
-from sagemaker.core.huggingface.model import HuggingFaceModel, HuggingFacePredictor  # noqa: F401
-from sagemaker.core.huggingface.processing import HuggingFaceProcessor  # noqa: F401
-from sagemaker.core.huggingface.training_compiler.config import TrainingCompilerConfig  # noqa: F401
-
-__all__ = [
-    "HuggingFace",
-    "HuggingFaceModel",
-    "HuggingFacePredictor",
-    "HuggingFaceProcessor",
-    "TrainingCompilerConfig",
-    "get_huggingface_llm_image_uri",
-]

sagemaker/core/huggingface/llm_utils.py

@@ -1,150 +0,0 @@
-# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License"). You
-# may not use this file except in compliance with the License. A copy of
-# the License is located at
-#
-#     http://aws.amazon.com/apache2.0/
-#
-# or in the "license" file accompanying this file. This file is
-# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
-# ANY KIND, either express or implied. See the License for the specific
-# language governing permissions and limitations under the License.
-"""Functions for generating ECR image URIs for pre-built SageMaker Docker images."""
-from __future__ import absolute_import
-
-import os
-from typing import Optional
-import importlib.util
-
-import urllib.request
-from urllib.error import HTTPError, URLError
-import json
-from json import JSONDecodeError
-import logging
-from sagemaker.core import image_uris
-from sagemaker.core.helper.session_helper import Session
-
-logger = logging.getLogger(__name__)
-
-
-def get_huggingface_llm_image_uri(
-    backend: str,
-    session: Optional[Session] = None,
-    region: Optional[str] = None,
-    version: Optional[str] = None,
-) -> str:
-    """Retrieves the image URI for inference.
-
-    Args:
-        backend (str): The backend to use. Valid values include "huggingface" and "lmi".
-        session (Session): The SageMaker Session to use. (Default: None).
-        region (str): The AWS region to use for image URI. (default: None).
-        version (str): The framework version for which to retrieve an
-            image URI. If no version is set, defaults to latest version. (default: None).
-
-    Returns:
-        str: The image URI string.
-    """
-
-    if region is None:
-        if session is None:
-            region = Session().boto_session.region_name
-        else:
-            region = session.boto_session.region_name
-    if backend == "huggingface":
-        return image_uris.retrieve(
-            "huggingface-llm",
-            region=region,
-            version=version,
-            image_scope="inference",
-        )
-    if backend == "huggingface-neuronx":
-        return image_uris.retrieve(
-            "huggingface-llm-neuronx",
-            region=region,
-            version=version,
-            image_scope="inference",
-            inference_tool="neuronx",
-        )
-    if backend == "huggingface-tei":
-        return image_uris.retrieve(
-            "huggingface-tei",
-            region=region,
-            version=version,
-            image_scope="inference",
-        )
-    if backend == "huggingface-tei-cpu":
-        return image_uris.retrieve(
-            "huggingface-tei-cpu",
-            region=region,
-            version=version,
-            image_scope="inference",
-        )
-    if backend == "lmi":
-        version = version or "0.24.0"
-        return image_uris.retrieve(framework="djl-deepspeed", region=region, version=version)
-    raise ValueError("Unsupported backend: %s" % backend)
-
-
-def get_huggingface_model_metadata(model_id: str, hf_hub_token: Optional[str] = None) -> dict:
-    """Retrieves the json metadata of the HuggingFace Model via HuggingFace API.
-
-    Args:
-        model_id (str): The HuggingFace Model ID
-        hf_hub_token (str): The HuggingFace Hub Token needed for Private/Gated HuggingFace Models
-
-    Returns:
-        dict: The model metadata retrieved with the HuggingFace API
-    """
-    if not model_id:
-        raise ValueError("Model ID is empty. Please provide a valid Model ID.")
-    hf_model_metadata_url = f"https://huggingface.co/api/models/{model_id}"
-    hf_model_metadata_json = None
-    try:
-        if hf_hub_token:
-            hf_model_metadata_url = urllib.request.Request(
-                hf_model_metadata_url, None, {"Authorization": "Bearer " + hf_hub_token}
-            )
-        with urllib.request.urlopen(hf_model_metadata_url) as response:
-            hf_model_metadata_json = json.load(response)
-    except (HTTPError, URLError, TimeoutError, JSONDecodeError) as e:
-        if "HTTP Error 401: Unauthorized" in str(e):
-            raise ValueError(
-                "Trying to access a gated/private HuggingFace model without valid credentials. "
-                "Please provide a HUGGING_FACE_HUB_TOKEN in env_vars"
-            )
-        logger.warning(
-            "Exception encountered while trying to retrieve HuggingFace model metadata %s. "
-            "Details: %s",
-            hf_model_metadata_url,
-            e,
-        )
-    if not hf_model_metadata_json:
-        raise ValueError(
-            "Did not find model metadata for the following HuggingFace Model ID %s" % model_id
-        )
-    return hf_model_metadata_json
-
-
-def download_huggingface_model_metadata(
-    model_id: str, model_local_path: str, hf_hub_token: Optional[str] = None
-) -> None:
-    """Downloads the HuggingFace Model snapshot via HuggingFace API.
-
-    Args:
-        model_id (str): The HuggingFace Model ID
-        model_local_path (str): The local path to save the HuggingFace Model snapshot.
-        hf_hub_token (str): The HuggingFace Hub Token
-
-    Raises:
-        ImportError: If huggingface_hub is not installed.
-    """
-    if not importlib.util.find_spec("huggingface_hub"):
-        raise ImportError("Unable to import huggingface_hub, check if huggingface_hub is installed")
-
-    from huggingface_hub import snapshot_download
-
-    os.makedirs(model_local_path, exist_ok=True)
-    logger.info("Downloading model %s from Hugging Face Hub to %s", model_id, model_local_path)
-    snapshot_download(repo_id=model_id, local_dir=model_local_path, token=hf_hub_token)

sagemaker/core/huggingface/processing.py

@@ -1,139 +0,0 @@
-# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License"). You
-# may not use this file except in compliance with the License. A copy of
-# the License is located at
-#
-#     http://aws.amazon.com/apache2.0/
-#
-# or in the "license" file accompanying this file. This file is
-# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
-# ANY KIND, either express or implied. See the License for the specific
-# language governing permissions and limitations under the License.
-"""This module contains code related to HuggingFace Processors which are used for Processing jobs.
-
-These jobs let customers perform data pre-processing, post-processing, feature engineering,
-data validation, and model evaluation and interpretation on SageMaker.
-"""
-from __future__ import absolute_import
-
-from typing import Union, Optional, List, Dict
-
-from sagemaker.core.helper.session_helper import Session
-from sagemaker.core.network import NetworkConfig
-from sagemaker.core.processing import FrameworkProcessor
-from sagemaker.core.huggingface.estimator import HuggingFace
-
-from sagemaker.core.helper.pipeline_variable import PipelineVariable
-from sagemaker.core.common_utils import format_tags, Tags
-
-
-class HuggingFaceProcessor(FrameworkProcessor):
-    """Handles Amazon SageMaker processing tasks for jobs using HuggingFace containers."""
-
-    estimator_cls = HuggingFace
-
-    def __init__(
-        self,
-        role: Optional[Union[str, PipelineVariable]] = None,
-        instance_count: Union[int, PipelineVariable] = None,
-        instance_type: Union[str, PipelineVariable] = None,
-        transformers_version: Optional[str] = None,
-        tensorflow_version: Optional[str] = None,
-        pytorch_version: Optional[str] = None,
-        py_version: str = "py36",
-        image_uri: Optional[Union[str, PipelineVariable]] = None,
-        command: Optional[List[str]] = None,
-        volume_size_in_gb: Union[int, PipelineVariable] = 30,
-        volume_kms_key: Optional[Union[str, PipelineVariable]] = None,
-        output_kms_key: Optional[Union[str, PipelineVariable]] = None,
-        code_location: Optional[str] = None,
-        max_runtime_in_seconds: Optional[Union[int, PipelineVariable]] = None,
-        base_job_name: Optional[str] = None,
-        sagemaker_session: Optional[Session] = None,
-        env: Optional[Dict[str, Union[str, PipelineVariable]]] = None,
-        tags: Optional[Tags] = None,
-        network_config: Optional[NetworkConfig] = None,
-    ):
-        """This processor executes a Python script in a HuggingFace execution environment.
-
-        Unless ``image_uri`` is specified, the environment is an Amazon-built Docker container
-        that executes functions defined in the supplied ``code`` Python script.
-
-        The arguments have the same meaning as in ``FrameworkProcessor``, with the following
-        exceptions.
-
-        Args:
-            transformers_version (str): Transformers version you want to use for
-                executing your model training code. Defaults to ``None``. Required unless
-                ``image_uri`` is provided. The current supported version is ``4.4.2``.
-            tensorflow_version (str): TensorFlow version you want to use for
-                executing your model training code. Defaults to ``None``. Required unless
-                ``pytorch_version`` is provided. The current supported version is ``2.4.1``.
-            pytorch_version (str): PyTorch version you want to use for
-                executing your model training code. Defaults to ``None``. Required unless
-                ``tensorflow_version`` is provided. The current supported version is ``1.6.0``.
-            py_version (str): Python version you want to use for executing your model training
-                code. Defaults to ``None``. Required unless ``image_uri`` is provided.  If
-                using PyTorch, the current supported version is ``py36``. If using TensorFlow,
-                the current supported version is ``py37``.
-
-        .. tip::
-
-            You can find additional parameters for initializing this class at
-            :class:`~sagemaker.processing.FrameworkProcessor`.
-        """
-        self.pytorch_version = pytorch_version
-        self.tensorflow_version = tensorflow_version
-        super().__init__(
-            self.estimator_cls,
-            transformers_version,
-            role,
-            instance_count,
-            instance_type,
-            py_version,
-            image_uri,
-            command,
-            volume_size_in_gb,
-            volume_kms_key,
-            output_kms_key,
-            code_location,
-            max_runtime_in_seconds,
-            base_job_name,
-            sagemaker_session,
-            env,
-            format_tags(tags),
-            network_config,
-        )
-
-    def _create_estimator(
-        self,
-        entry_point="",
-        source_dir=None,
-        dependencies=None,
-        git_config=None,
-    ):
-        """Override default estimator factory function for HuggingFace's different parameters
-
-        HuggingFace estimators have 3 framework version parameters instead of one: The version for
-        Transformers, PyTorch, and TensorFlow.
-        """
-        return self.estimator_cls(
-            transformers_version=self.framework_version,
-            tensorflow_version=self.tensorflow_version,
-            pytorch_version=self.pytorch_version,
-            py_version=self.py_version,
-            entry_point=entry_point,
-            source_dir=source_dir,
-            dependencies=dependencies,
-            git_config=git_config,
-            code_location=self.code_location,
-            enable_network_isolation=False,
-            image_uri=self.image_uri,
-            role=self.role,
-            instance_count=self.instance_count,
-            instance_type=self.instance_type,
-            sagemaker_session=self.sagemaker_session,
-            debugger_hook_config=False,
-            disable_profiler=True,
-        )