sagemaker-core 1.0.62__py3-none-any.whl → 2.1.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 (362) hide show
  1. sagemaker/core/__init__.py +16 -0
  2. sagemaker/core/_studio.py +116 -0
  3. sagemaker/core/_version.py +11 -0
  4. sagemaker/core/accept_types.py +131 -0
  5. sagemaker/core/analytics.py +744 -0
  6. sagemaker/core/apiutils/__init__.py +13 -0
  7. sagemaker/core/apiutils/_base_types.py +228 -0
  8. sagemaker/core/apiutils/_boto_functions.py +130 -0
  9. sagemaker/core/apiutils/_utils.py +34 -0
  10. sagemaker/core/base_deserializers.py +35 -0
  11. sagemaker/core/base_serializers.py +35 -0
  12. sagemaker/core/clarify/__init__.py +2898 -0
  13. sagemaker/core/collection.py +467 -0
  14. sagemaker/core/common_utils.py +2281 -0
  15. sagemaker/core/compute_resource_requirements/__init__.py +18 -0
  16. sagemaker/core/compute_resource_requirements/resource_requirements.py +94 -0
  17. sagemaker/core/config/__init__.py +181 -0
  18. sagemaker/core/config/config.py +238 -0
  19. sagemaker/core/config/config_manager.py +595 -0
  20. sagemaker/core/config/config_schema.py +1220 -0
  21. sagemaker/core/config/config_utils.py +297 -0
  22. {sagemaker_core/main → sagemaker/core}/config_schema.py +408 -3
  23. sagemaker/core/constants.py +73 -0
  24. sagemaker/core/content_types.py +137 -0
  25. sagemaker/core/debugger/__init__.py +39 -0
  26. sagemaker/core/debugger/debugger.py +945 -0
  27. sagemaker/core/debugger/framework_profile.py +292 -0
  28. sagemaker/core/debugger/metrics_config.py +468 -0
  29. sagemaker/core/debugger/profiler.py +42 -0
  30. sagemaker/core/debugger/profiler_config.py +190 -0
  31. sagemaker/core/debugger/profiler_constants.py +40 -0
  32. sagemaker/core/debugger/utils.py +148 -0
  33. sagemaker/core/deprecations.py +254 -0
  34. sagemaker/core/deserializers/__init__.py +10 -0
  35. sagemaker/core/deserializers/base.py +424 -0
  36. sagemaker/core/deserializers/implementations.py +157 -0
  37. sagemaker/core/drift_check_baselines.py +106 -0
  38. sagemaker/core/enums.py +51 -0
  39. sagemaker/core/environment_variables.py +101 -0
  40. sagemaker/core/exceptions.py +108 -0
  41. sagemaker/core/experiments/__init__.py +53 -0
  42. sagemaker/core/experiments/_api_types.py +251 -0
  43. sagemaker/core/experiments/_environment.py +124 -0
  44. sagemaker/core/experiments/_helper.py +294 -0
  45. sagemaker/core/experiments/_metrics.py +333 -0
  46. sagemaker/core/experiments/_run_context.py +58 -0
  47. sagemaker/core/experiments/_utils.py +216 -0
  48. sagemaker/core/experiments/experiment.py +244 -0
  49. sagemaker/core/experiments/run.py +970 -0
  50. sagemaker/core/experiments/trial.py +296 -0
  51. sagemaker/core/experiments/trial_component.py +387 -0
  52. sagemaker/core/explainer/__init__.py +24 -0
  53. sagemaker/core/explainer/clarify_explainer_config.py +298 -0
  54. sagemaker/core/explainer/explainer_config.py +44 -0
  55. sagemaker/core/fw_utils.py +1176 -0
  56. sagemaker/core/git_utils.py +349 -0
  57. sagemaker/core/helper/pipeline_variable.py +82 -0
  58. sagemaker/core/helper/session_helper.py +2965 -0
  59. sagemaker/core/huggingface/__init__.py +29 -0
  60. sagemaker/core/huggingface/llm_utils.py +150 -0
  61. sagemaker/core/huggingface/processing.py +139 -0
  62. sagemaker/core/huggingface/training_compiler/config.py +167 -0
  63. sagemaker/core/hyperparameters.py +172 -0
  64. sagemaker/core/image_retriever/__init__.py +3 -0
  65. sagemaker/core/image_retriever/image_retriever.py +640 -0
  66. sagemaker/core/image_retriever/image_retriever_utils.py +511 -0
  67. sagemaker/core/image_retriever/test.py +7 -0
  68. sagemaker/core/image_uri_config/__init__.py +13 -0
  69. sagemaker/core/image_uri_config/autogluon.json +1335 -0
  70. sagemaker/core/image_uri_config/blazingtext.json +50 -0
  71. sagemaker/core/image_uri_config/chainer.json +104 -0
  72. sagemaker/core/image_uri_config/clarify.json +39 -0
  73. sagemaker/core/image_uri_config/coach-mxnet.json +70 -0
  74. sagemaker/core/image_uri_config/coach-tensorflow.json +186 -0
  75. sagemaker/core/image_uri_config/data-wrangler.json +91 -0
  76. sagemaker/core/image_uri_config/debugger.json +34 -0
  77. sagemaker/core/image_uri_config/detailed-profiler.json +18 -0
  78. sagemaker/core/image_uri_config/djl-deepspeed.json +385 -0
  79. sagemaker/core/image_uri_config/djl-fastertransformer.json +167 -0
  80. sagemaker/core/image_uri_config/djl-lmi.json +136 -0
  81. sagemaker/core/image_uri_config/djl-neuronx.json +258 -0
  82. sagemaker/core/image_uri_config/djl-tensorrtllm.json +262 -0
  83. sagemaker/core/image_uri_config/factorization-machines.json +50 -0
  84. sagemaker/core/image_uri_config/forecasting-deepar.json +50 -0
  85. sagemaker/core/image_uri_config/huggingface-llm-neuronx.json +660 -0
  86. sagemaker/core/image_uri_config/huggingface-llm.json +1158 -0
  87. sagemaker/core/image_uri_config/huggingface-neuron.json +52 -0
  88. sagemaker/core/image_uri_config/huggingface-neuronx.json +510 -0
  89. sagemaker/core/image_uri_config/huggingface-tei-cpu.json +298 -0
  90. sagemaker/core/image_uri_config/huggingface-tei.json +298 -0
  91. sagemaker/core/image_uri_config/huggingface-training-compiler.json +195 -0
  92. sagemaker/core/image_uri_config/huggingface.json +2138 -0
  93. sagemaker/core/image_uri_config/hyperpod-recipes-neuron.json +52 -0
  94. sagemaker/core/image_uri_config/image-classification-neo.json +43 -0
  95. sagemaker/core/image_uri_config/image-classification.json +50 -0
  96. sagemaker/core/image_uri_config/inferentia-mxnet.json +88 -0
  97. sagemaker/core/image_uri_config/inferentia-pytorch.json +127 -0
  98. sagemaker/core/image_uri_config/inferentia-tensorflow.json +88 -0
  99. sagemaker/core/image_uri_config/instance_gpu_info.json +782 -0
  100. sagemaker/core/image_uri_config/ipinsights.json +50 -0
  101. sagemaker/core/image_uri_config/kmeans.json +50 -0
  102. sagemaker/core/image_uri_config/knn.json +50 -0
  103. sagemaker/core/image_uri_config/lda.json +26 -0
  104. sagemaker/core/image_uri_config/linear-learner.json +50 -0
  105. sagemaker/core/image_uri_config/model-monitor.json +42 -0
  106. sagemaker/core/image_uri_config/mxnet.json +1154 -0
  107. sagemaker/core/image_uri_config/neo-mxnet.json +64 -0
  108. sagemaker/core/image_uri_config/neo-pytorch.json +341 -0
  109. sagemaker/core/image_uri_config/neo-tensorflow.json +109 -0
  110. sagemaker/core/image_uri_config/ntm.json +50 -0
  111. sagemaker/core/image_uri_config/object-detection.json +50 -0
  112. sagemaker/core/image_uri_config/object2vec.json +50 -0
  113. sagemaker/core/image_uri_config/pca.json +50 -0
  114. sagemaker/core/image_uri_config/pytorch-neuron.json +43 -0
  115. sagemaker/core/image_uri_config/pytorch-smp.json +218 -0
  116. sagemaker/core/image_uri_config/pytorch-training-compiler.json +80 -0
  117. sagemaker/core/image_uri_config/pytorch.json +3101 -0
  118. sagemaker/core/image_uri_config/randomcutforest.json +50 -0
  119. sagemaker/core/image_uri_config/ray-pytorch.json +46 -0
  120. sagemaker/core/image_uri_config/ray-tensorflow.json +194 -0
  121. sagemaker/core/image_uri_config/sagemaker-base-python.json +46 -0
  122. sagemaker/core/image_uri_config/sagemaker-distribution.json +37 -0
  123. sagemaker/core/image_uri_config/sagemaker-geospatial.json +13 -0
  124. sagemaker/core/image_uri_config/sagemaker-tritonserver.json +212 -0
  125. sagemaker/core/image_uri_config/semantic-segmentation.json +50 -0
  126. sagemaker/core/image_uri_config/seq2seq.json +50 -0
  127. sagemaker/core/image_uri_config/sklearn.json +446 -0
  128. sagemaker/core/image_uri_config/spark.json +280 -0
  129. sagemaker/core/image_uri_config/sparkml-serving.json +97 -0
  130. sagemaker/core/image_uri_config/stabilityai.json +53 -0
  131. sagemaker/core/image_uri_config/tensorflow.json +5086 -0
  132. sagemaker/core/image_uri_config/vw.json +25 -0
  133. sagemaker/core/image_uri_config/xgboost-neo.json +43 -0
  134. sagemaker/core/image_uri_config/xgboost.json +888 -0
  135. sagemaker/core/image_uris.py +810 -0
  136. sagemaker/core/inference_config.py +144 -0
  137. sagemaker/core/inference_recommender/__init__.py +18 -0
  138. sagemaker/core/inference_recommender/inference_recommender_mixin.py +622 -0
  139. sagemaker/core/inputs.py +366 -0
  140. sagemaker/core/instance_group.py +61 -0
  141. sagemaker/core/instance_types.py +164 -0
  142. sagemaker/core/instance_types_gpu_info.py +43 -0
  143. sagemaker/core/interactive_apps/__init__.py +41 -0
  144. sagemaker/core/interactive_apps/base_interactive_app.py +204 -0
  145. sagemaker/core/interactive_apps/detail_profiler_app.py +139 -0
  146. sagemaker/core/interactive_apps/tensorboard.py +149 -0
  147. sagemaker/core/iterators.py +186 -0
  148. sagemaker/core/job.py +380 -0
  149. sagemaker/core/jumpstart/__init__.py +156 -0
  150. sagemaker/core/jumpstart/accessors.py +390 -0
  151. sagemaker/core/jumpstart/artifacts/__init__.py +69 -0
  152. sagemaker/core/jumpstart/artifacts/environment_variables.py +252 -0
  153. sagemaker/core/jumpstart/artifacts/hyperparameters.py +120 -0
  154. sagemaker/core/jumpstart/artifacts/image_uris.py +139 -0
  155. sagemaker/core/jumpstart/artifacts/incremental_training.py +87 -0
  156. sagemaker/core/jumpstart/artifacts/instance_types.py +223 -0
  157. sagemaker/core/jumpstart/artifacts/kwargs.py +289 -0
  158. sagemaker/core/jumpstart/artifacts/metric_definitions.py +117 -0
  159. sagemaker/core/jumpstart/artifacts/model_packages.py +202 -0
  160. sagemaker/core/jumpstart/artifacts/model_uris.py +252 -0
  161. sagemaker/core/jumpstart/artifacts/payloads.py +96 -0
  162. sagemaker/core/jumpstart/artifacts/predictors.py +540 -0
  163. sagemaker/core/jumpstart/artifacts/resource_names.py +86 -0
  164. sagemaker/core/jumpstart/artifacts/resource_requirements.py +162 -0
  165. sagemaker/core/jumpstart/artifacts/script_uris.py +172 -0
  166. sagemaker/core/jumpstart/cache.py +663 -0
  167. sagemaker/core/jumpstart/configs.py +50 -0
  168. sagemaker/core/jumpstart/constants.py +198 -0
  169. sagemaker/core/jumpstart/deserializers.py +81 -0
  170. sagemaker/core/jumpstart/document.py +76 -0
  171. sagemaker/core/jumpstart/enums.py +168 -0
  172. sagemaker/core/jumpstart/exceptions.py +236 -0
  173. sagemaker/core/jumpstart/factory/utils.py +833 -0
  174. sagemaker/core/jumpstart/filters.py +597 -0
  175. sagemaker/core/jumpstart/hub/constants.py +16 -0
  176. sagemaker/core/jumpstart/hub/hub.py +291 -0
  177. sagemaker/core/jumpstart/hub/interfaces.py +936 -0
  178. sagemaker/core/jumpstart/hub/parser_utils.py +70 -0
  179. sagemaker/core/jumpstart/hub/parsers.py +288 -0
  180. sagemaker/core/jumpstart/hub/types.py +35 -0
  181. sagemaker/core/jumpstart/hub/utils.py +260 -0
  182. sagemaker/core/jumpstart/models.py +499 -0
  183. sagemaker/core/jumpstart/notebook_utils.py +575 -0
  184. sagemaker/core/jumpstart/parameters.py +20 -0
  185. sagemaker/core/jumpstart/payload_utils.py +239 -0
  186. sagemaker/core/jumpstart/region_config.json +163 -0
  187. sagemaker/core/jumpstart/search.py +171 -0
  188. sagemaker/core/jumpstart/serializers.py +81 -0
  189. sagemaker/core/jumpstart/session_utils.py +234 -0
  190. sagemaker/core/jumpstart/types.py +3044 -0
  191. sagemaker/core/jumpstart/utils.py +1731 -0
  192. sagemaker/core/jumpstart/validators.py +257 -0
  193. sagemaker/core/lambda_helper.py +312 -0
  194. sagemaker/core/lineage/__init__.py +42 -0
  195. sagemaker/core/lineage/_api_types.py +239 -0
  196. sagemaker/core/lineage/_utils.py +49 -0
  197. sagemaker/core/lineage/action.py +345 -0
  198. sagemaker/core/lineage/artifact.py +646 -0
  199. sagemaker/core/lineage/association.py +190 -0
  200. sagemaker/core/lineage/context.py +505 -0
  201. sagemaker/core/lineage/lineage_trial_component.py +191 -0
  202. sagemaker/core/lineage/query.py +732 -0
  203. sagemaker/core/lineage/visualizer.py +346 -0
  204. sagemaker/core/local/__init__.py +18 -0
  205. sagemaker/core/local/data.py +413 -0
  206. sagemaker/core/local/entities.py +678 -0
  207. sagemaker/core/local/exceptions.py +17 -0
  208. sagemaker/core/local/image.py +1243 -0
  209. sagemaker/core/local/local_session.py +739 -0
  210. sagemaker/core/local/utils.py +245 -0
  211. sagemaker/core/logs.py +181 -0
  212. sagemaker/core/metadata_properties.py +56 -0
  213. sagemaker/core/metric_definitions.py +91 -0
  214. sagemaker/core/mlflow/__init__.py +38 -0
  215. sagemaker/core/mlflow/forward_sagemaker_metrics.py +44 -0
  216. sagemaker/core/model_card/__init__.py +26 -0
  217. sagemaker/core/model_life_cycle.py +51 -0
  218. sagemaker/core/model_metrics.py +160 -0
  219. sagemaker/core/model_monitor/__init__.py +66 -0
  220. sagemaker/core/model_monitor/clarify_model_monitoring.py +1495 -0
  221. sagemaker/core/model_monitor/cron_expression_generator.py +82 -0
  222. sagemaker/core/model_monitor/data_capture_config.py +115 -0
  223. sagemaker/core/model_monitor/data_quality_monitoring_config.py +66 -0
  224. sagemaker/core/model_monitor/dataset_format.py +102 -0
  225. sagemaker/core/model_monitor/model_monitoring.py +4266 -0
  226. sagemaker/core/model_monitor/monitoring_alert.py +76 -0
  227. sagemaker/core/model_monitor/monitoring_files.py +506 -0
  228. sagemaker/core/model_monitor/utils.py +793 -0
  229. sagemaker/core/model_registry.py +480 -0
  230. sagemaker/core/model_uris.py +97 -0
  231. sagemaker/core/modules/__init__.py +19 -0
  232. sagemaker/core/modules/configs.py +226 -0
  233. sagemaker/core/modules/constants.py +37 -0
  234. sagemaker/core/modules/distributed.py +182 -0
  235. sagemaker/core/modules/local_core/__init__.py +0 -0
  236. sagemaker/core/modules/local_core/local_container.py +605 -0
  237. sagemaker/core/modules/templates.py +83 -0
  238. sagemaker/core/modules/train/__init__.py +14 -0
  239. sagemaker/core/modules/train/container_drivers/__init__.py +14 -0
  240. sagemaker/core/modules/train/container_drivers/common/__init__.py +14 -0
  241. sagemaker/core/modules/train/container_drivers/common/utils.py +213 -0
  242. sagemaker/core/modules/train/container_drivers/distributed_drivers/__init__.py +14 -0
  243. sagemaker/core/modules/train/container_drivers/distributed_drivers/basic_script_driver.py +81 -0
  244. sagemaker/core/modules/train/container_drivers/distributed_drivers/mpi_driver.py +123 -0
  245. sagemaker/core/modules/train/container_drivers/distributed_drivers/mpi_utils.py +302 -0
  246. sagemaker/core/modules/train/container_drivers/distributed_drivers/torchrun_driver.py +129 -0
  247. sagemaker/core/modules/train/container_drivers/scripts/__init__.py +14 -0
  248. sagemaker/core/modules/train/container_drivers/scripts/environment.py +305 -0
  249. sagemaker/core/modules/train/sm_recipes/__init__.py +0 -0
  250. sagemaker/core/modules/train/sm_recipes/utils.py +330 -0
  251. sagemaker/core/modules/types.py +19 -0
  252. sagemaker/core/modules/utils.py +194 -0
  253. sagemaker/core/network.py +185 -0
  254. sagemaker/core/parameter.py +173 -0
  255. sagemaker/core/payloads.py +185 -0
  256. sagemaker/core/processing.py +1597 -0
  257. sagemaker/core/remote_function/__init__.py +19 -0
  258. sagemaker/core/remote_function/checkpoint_location.py +47 -0
  259. sagemaker/core/remote_function/client.py +1285 -0
  260. sagemaker/core/remote_function/core/__init__.py +0 -0
  261. sagemaker/core/remote_function/core/_custom_dispatch_table.py +72 -0
  262. sagemaker/core/remote_function/core/pipeline_variables.py +353 -0
  263. sagemaker/core/remote_function/core/serialization.py +422 -0
  264. sagemaker/core/remote_function/core/stored_function.py +226 -0
  265. sagemaker/core/remote_function/custom_file_filter.py +128 -0
  266. sagemaker/core/remote_function/errors.py +104 -0
  267. sagemaker/core/remote_function/invoke_function.py +172 -0
  268. sagemaker/core/remote_function/job.py +2140 -0
  269. sagemaker/core/remote_function/logging_config.py +38 -0
  270. sagemaker/core/remote_function/runtime_environment/__init__.py +14 -0
  271. sagemaker/core/remote_function/runtime_environment/bootstrap_runtime_environment.py +605 -0
  272. sagemaker/core/remote_function/runtime_environment/mpi_utils_remote.py +252 -0
  273. sagemaker/core/remote_function/runtime_environment/runtime_environment_manager.py +554 -0
  274. sagemaker/core/remote_function/runtime_environment/spark_app.py +18 -0
  275. sagemaker/core/remote_function/spark_config.py +149 -0
  276. sagemaker/core/resource_requirements.py +168 -0
  277. {sagemaker_core/main → sagemaker/core}/resources.py +19098 -10895
  278. sagemaker/core/s3/__init__.py +41 -0
  279. sagemaker/core/s3/client.py +367 -0
  280. sagemaker/core/s3/utils.py +175 -0
  281. sagemaker/core/script_uris.py +93 -0
  282. sagemaker/core/serializers/__init__.py +11 -0
  283. sagemaker/core/serializers/base.py +510 -0
  284. sagemaker/core/serializers/implementations.py +159 -0
  285. sagemaker/core/serializers/utils.py +223 -0
  286. sagemaker/core/serverless_inference_config.py +63 -0
  287. sagemaker/core/session_settings.py +55 -0
  288. sagemaker/core/shapes/__init__.py +3 -0
  289. sagemaker/core/shapes/model_card_shapes.py +159 -0
  290. {sagemaker_core/main → sagemaker/core/shapes}/shapes.py +5810 -1806
  291. sagemaker/core/spark/__init__.py +16 -0
  292. sagemaker/core/spark/defaults.py +16 -0
  293. sagemaker/core/spark/processing.py +1380 -0
  294. sagemaker/core/telemetry/__init__.py +23 -0
  295. sagemaker/core/telemetry/constants.py +84 -0
  296. sagemaker/core/telemetry/telemetry_logging.py +284 -0
  297. sagemaker/core/tools/__init__.py +1 -0
  298. {sagemaker_core → sagemaker/core}/tools/codegen.py +4 -4
  299. {sagemaker_core → sagemaker/core}/tools/constants.py +23 -15
  300. {sagemaker_core → sagemaker/core}/tools/data_extractor.py +1 -1
  301. {sagemaker_core → sagemaker/core}/tools/method.py +1 -1
  302. sagemaker/core/tools/model_card/generate_model_card_from_schema.py +562 -0
  303. {sagemaker_core → sagemaker/core}/tools/resources_codegen.py +165 -98
  304. {sagemaker_core → sagemaker/core}/tools/resources_extractor.py +5 -13
  305. {sagemaker_core → sagemaker/core}/tools/shapes_codegen.py +16 -17
  306. {sagemaker_core → sagemaker/core}/tools/shapes_extractor.py +29 -67
  307. {sagemaker_core → sagemaker/core}/tools/templates.py +39 -17
  308. sagemaker/core/training/__init__.py +14 -0
  309. sagemaker/core/training/configs.py +333 -0
  310. sagemaker/core/training/constants.py +37 -0
  311. sagemaker/core/training/utils.py +77 -0
  312. sagemaker/core/training_compiler/__init__.py +16 -0
  313. sagemaker/core/training_compiler/config.py +197 -0
  314. sagemaker/core/training_compiler_config.py +197 -0
  315. sagemaker/core/transformer.py +793 -0
  316. sagemaker/core/user_agent.py +76 -0
  317. sagemaker/core/utilities/__init__.py +24 -0
  318. sagemaker/core/utilities/cache.py +169 -0
  319. sagemaker/core/utilities/search_expression.py +133 -0
  320. sagemaker/core/utils/__init__.py +48 -0
  321. sagemaker/core/utils/code_injection/__init__.py +0 -0
  322. {sagemaker_core/main → sagemaker/core/utils}/code_injection/codec.py +2 -2
  323. {sagemaker_core/main → sagemaker/core/utils}/code_injection/shape_dag.py +5979 -176
  324. {sagemaker_core/main → sagemaker/core/utils}/exceptions.py +8 -8
  325. sagemaker_core/main/default_configs_helper.py → sagemaker/core/utils/intelligent_defaults_helper.py +5 -6
  326. {sagemaker_core/main → sagemaker/core/utils}/logs.py +1 -2
  327. {sagemaker_core/main → sagemaker/core/utils}/utils.py +27 -22
  328. sagemaker/core/workflow/__init__.py +152 -0
  329. sagemaker/core/workflow/conditions.py +313 -0
  330. sagemaker/core/workflow/entities.py +58 -0
  331. sagemaker/core/workflow/execution_variables.py +89 -0
  332. sagemaker/core/workflow/functions.py +193 -0
  333. sagemaker/core/workflow/parameters.py +222 -0
  334. sagemaker/core/workflow/pipeline_context.py +394 -0
  335. sagemaker/core/workflow/pipeline_definition_config.py +31 -0
  336. sagemaker/core/workflow/properties.py +285 -0
  337. sagemaker/core/workflow/step_outputs.py +65 -0
  338. sagemaker/core/workflow/utilities.py +507 -0
  339. sagemaker/lineage/__init__.py +33 -0
  340. sagemaker/lineage/action.py +28 -0
  341. sagemaker/lineage/artifact.py +28 -0
  342. sagemaker/lineage/context.py +28 -0
  343. sagemaker/lineage/lineage_trial_component.py +28 -0
  344. {sagemaker_core-1.0.62.dist-info → sagemaker_core-2.1.1.dist-info}/METADATA +28 -9
  345. sagemaker_core-2.1.1.dist-info/RECORD +355 -0
  346. sagemaker_core-2.1.1.dist-info/top_level.txt +1 -0
  347. sagemaker_core/_version.py +0 -3
  348. sagemaker_core/helper/session_helper.py +0 -769
  349. sagemaker_core/resources/__init__.py +0 -1
  350. sagemaker_core/shapes/__init__.py +0 -1
  351. sagemaker_core/tools/__init__.py +0 -1
  352. sagemaker_core-1.0.62.dist-info/RECORD +0 -35
  353. sagemaker_core-1.0.62.dist-info/top_level.txt +0 -1
  354. {sagemaker_core → sagemaker/core/helper}/__init__.py +0 -0
  355. {sagemaker_core/helper → sagemaker/core/huggingface/training_compiler}/__init__.py +0 -0
  356. {sagemaker_core/main → sagemaker/core/jumpstart/factory}/__init__.py +0 -0
  357. {sagemaker_core/main/code_injection → sagemaker/core/jumpstart/hub}/__init__.py +0 -0
  358. {sagemaker_core/main → sagemaker/core/utils}/code_injection/base.py +0 -0
  359. {sagemaker_core/main → sagemaker/core/utils}/code_injection/constants.py +0 -0
  360. {sagemaker_core/main → sagemaker/core/utils}/user_agent.py +0 -0
  361. {sagemaker_core-1.0.62.dist-info → sagemaker_core-2.1.1.dist-info}/WHEEL +0 -0
  362. {sagemaker_core-1.0.62.dist-info → sagemaker_core-2.1.1.dist-info}/licenses/LICENSE +0 -0
sagemaker/core/image_uri_config/huggingface-neuron.json ADDED
@@ -0,0 +1,52 @@
1
+ {
2
+ "inference": {
3
+ "processors": [
4
+ "inf"
5
+ ],
6
+ "version_aliases": {
7
+ "4.12": "4.12.3"
8
+ },
9
+ "versions": {
10
+ "4.12.3": {
11
+ "version_aliases": {
12
+ "pytorch1.9": "pytorch1.9.1"
13
+ },
14
+ "pytorch1.9.1": {
15
+ "py_versions": [
16
+ "py37"
17
+ ],
18
+ "repository": "huggingface-pytorch-inference-neuron",
19
+ "registries": {
20
+ "ap-east-2": "975050140332",
21
+ "ap-northeast-1": "763104351884",
22
+ "ap-south-1": "763104351884",
23
+ "ap-south-2": "772153158452",
24
+ "ap-southeast-1": "763104351884",
25
+ "ap-southeast-2": "763104351884",
26
+ "ap-southeast-4": "457447274322",
27
+ "ap-southeast-5": "550225433462",
28
+ "ap-southeast-7": "590183813437",
29
+ "eu-central-1": "763104351884",
30
+ "eu-central-2": "380420809688",
31
+ "eu-south-2": "503227376785",
32
+ "eu-west-1": "763104351884",
33
+ "eu-west-3": "763104351884",
34
+ "il-central-1": "780543022126",
35
+ "mx-central-1":"637423239942",
36
+ "sa-east-1": "763104351884",
37
+ "us-east-1": "763104351884",
38
+ "us-east-2": "763104351884",
39
+ "us-west-2": "763104351884",
40
+ "ca-west-1": "204538143572"
41
+ },
42
+ "container_version": {
43
+ "inf": "ubuntu18.04"
44
+ },
45
+ "sdk_versions": [
46
+ "sdk1.17.1"
47
+ ]
48
+ }
49
+ }
50
+ }
51
+ }
52
+ }
sagemaker/core/image_uri_config/huggingface-neuronx.json ADDED
@@ -0,0 +1,510 @@
1
+ {
2
+ "training": {
3
+ "processors": [
4
+ "trn"
5
+ ],
6
+ "version_aliases": {
7
+ "4.28": "4.28.1",
8
+ "4.34": "4.34.1",
9
+ "4.36": "4.36.2",
10
+ "4.43": "4.43.2",
11
+ "4.48": "4.48.1"
12
+ },
13
+ "versions": {
14
+ "4.28.1": {
15
+ "version_aliases": {
16
+ "pytorch1.13": "pytorch1.13.0"
17
+ },
18
+ "pytorch1.13.0": {
19
+ "py_versions": [
20
+ "py38"
21
+ ],
22
+ "repository": "huggingface-pytorch-training-neuronx",
23
+ "registries": {
24
+ "ap-east-2": "975050140332",
25
+ "ap-northeast-1": "763104351884",
26
+ "ap-south-1": "763104351884",
27
+ "ap-south-2": "772153158452",
28
+ "ap-southeast-1": "763104351884",
29
+ "ap-southeast-2": "763104351884",
30
+ "ap-southeast-4": "457447274322",
31
+ "ap-southeast-5": "550225433462",
32
+ "ap-southeast-7": "590183813437",
33
+ "cn-north-1": "727897471807",
34
+ "cn-northwest-1": "727897471807",
35
+ "eu-central-1": "763104351884",
36
+ "eu-central-2": "380420809688",
37
+ "eu-south-2": "503227376785",
38
+ "eu-west-1": "763104351884",
39
+ "eu-west-3": "763104351884",
40
+ "il-central-1": "780543022126",
41
+ "mx-central-1":"637423239942",
42
+ "sa-east-1": "763104351884",
43
+ "us-east-1": "763104351884",
44
+ "us-east-2": "763104351884",
45
+ "us-gov-east-1": "446045086412",
46
+ "us-gov-west-1": "442386744353",
47
+ "us-west-2": "763104351884",
48
+ "ca-west-1": "204538143572"
49
+ },
50
+ "container_version": {
51
+ "trn": "ubuntu20.04"
52
+ },
53
+ "sdk_versions": [
54
+ "sdk2.9.1"
55
+ ]
56
+ }
57
+ },
58
+ "4.34.1": {
59
+ "version_aliases": {
60
+ "pytorch1.13": "pytorch1.13.1"
61
+ },
62
+ "pytorch1.13.1": {
63
+ "py_versions": [
64
+ "py310"
65
+ ],
66
+ "repository": "huggingface-pytorch-inference-neuronx",
67
+ "registries": {
68
+ "ap-east-2": "975050140332",
69
+ "ap-northeast-1": "763104351884",
70
+ "ap-south-1": "763104351884",
71
+ "ap-south-2": "772153158452",
72
+ "ap-southeast-1": "763104351884",
73
+ "ap-southeast-2": "763104351884",
74
+ "ap-southeast-4": "457447274322",
75
+ "ap-southeast-5": "550225433462",
76
+ "ap-southeast-7": "590183813437",
77
+ "cn-north-1": "727897471807",
78
+ "cn-northwest-1": "727897471807",
79
+ "eu-central-1": "763104351884",
80
+ "eu-central-2": "380420809688",
81
+ "eu-south-2": "503227376785",
82
+ "eu-west-1": "763104351884",
83
+ "eu-west-3": "763104351884",
84
+ "il-central-1": "780543022126",
85
+ "mx-central-1":"637423239942",
86
+ "sa-east-1": "763104351884",
87
+ "us-east-1": "763104351884",
88
+ "us-east-2": "763104351884",
89
+ "us-gov-east-1": "446045086412",
90
+ "us-gov-west-1": "442386744353",
91
+ "us-west-2": "763104351884",
92
+ "ca-west-1": "204538143572"
93
+ },
94
+ "container_version": {
95
+ "inf": "ubuntu20.04"
96
+ },
97
+ "sdk_versions": [
98
+ "sdk2.15.0"
99
+ ]
100
+ }
101
+ },
102
+ "4.36.2": {
103
+ "version_aliases": {
104
+ "pytorch1.13": "pytorch1.13.1"
105
+ },
106
+ "pytorch1.13.1": {
107
+ "py_versions": [
108
+ "py310"
109
+ ],
110
+ "repository": "huggingface-pytorch-inference-neuronx",
111
+ "registries": {
112
+ "ap-east-2": "975050140332",
113
+ "ap-northeast-1": "763104351884",
114
+ "ap-south-1": "763104351884",
115
+ "ap-south-2": "772153158452",
116
+ "ap-southeast-1": "763104351884",
117
+ "ap-southeast-2": "763104351884",
118
+ "ap-southeast-4": "457447274322",
119
+ "ap-southeast-5": "550225433462",
120
+ "ap-southeast-7": "590183813437",
121
+ "cn-north-1": "727897471807",
122
+ "cn-northwest-1": "727897471807",
123
+ "eu-central-1": "763104351884",
124
+ "eu-central-2": "380420809688",
125
+ "eu-south-2": "503227376785",
126
+ "eu-west-1": "763104351884",
127
+ "eu-west-3": "763104351884",
128
+ "il-central-1": "780543022126",
129
+ "mx-central-1":"637423239942",
130
+ "sa-east-1": "763104351884",
131
+ "us-east-1": "763104351884",
132
+ "us-east-2": "763104351884",
133
+ "us-gov-east-1": "446045086412",
134
+ "us-gov-west-1": "442386744353",
135
+ "us-west-2": "763104351884",
136
+ "ca-west-1": "204538143572"
137
+ },
138
+ "container_version": {
139
+ "inf": "ubuntu20.04"
140
+ },
141
+ "sdk_versions": [
142
+ "sdk2.18.0"
143
+ ]
144
+ }
145
+ },
146
+ "4.43.2": {
147
+ "version_aliases": {
148
+ "pytorch2.1": "pytorch2.1.2"
149
+ },
150
+ "pytorch2.1.2": {
151
+ "py_versions": [
152
+ "py310"
153
+ ],
154
+ "repository": "huggingface-pytorch-inference-neuronx",
155
+ "registries": {
156
+ "ap-northeast-1": "763104351884",
157
+ "ap-south-1": "763104351884",
158
+ "ap-south-2": "772153158452",
159
+ "ap-southeast-1": "763104351884",
160
+ "ap-southeast-2": "763104351884",
161
+ "ap-southeast-4": "457447274322",
162
+ "ap-southeast-5": "550225433462",
163
+ "ap-southeast-7": "590183813437",
164
+ "cn-north-1": "727897471807",
165
+ "cn-northwest-1": "727897471807",
166
+ "eu-central-1": "763104351884",
167
+ "eu-central-2": "380420809688",
168
+ "eu-south-2": "503227376785",
169
+ "eu-west-1": "763104351884",
170
+ "eu-west-3": "763104351884",
171
+ "il-central-1": "780543022126",
172
+ "mx-central-1":"637423239942",
173
+ "sa-east-1": "763104351884",
174
+ "us-east-1": "763104351884",
175
+ "us-east-2": "763104351884",
176
+ "us-gov-east-1": "446045086412",
177
+ "us-gov-west-1": "442386744353",
178
+ "us-west-2": "763104351884",
179
+ "ca-west-1": "204538143572"
180
+ },
181
+ "container_version": {
182
+ "inf": "ubuntu20.04"
183
+ },
184
+ "sdk_versions": [
185
+ "sdk2.20.0"
186
+ ]
187
+ }
188
+ },
189
+ "4.48.1": {
190
+ "version_aliases": {
191
+ "pytorch2.1": "pytorch2.1.2"
192
+ },
193
+ "pytorch2.1.2": {
194
+ "py_versions": [
195
+ "py310"
196
+ ],
197
+ "repository": "huggingface-pytorch-inference-neuronx",
198
+ "registries": {
199
+ "ap-northeast-1": "763104351884",
200
+ "ap-south-1": "763104351884",
201
+ "ap-south-2": "772153158452",
202
+ "ap-southeast-1": "763104351884",
203
+ "ap-southeast-2": "763104351884",
204
+ "ap-southeast-4": "457447274322",
205
+ "ap-southeast-5": "550225433462",
206
+ "ap-southeast-7": "590183813437",
207
+ "cn-north-1": "727897471807",
208
+ "cn-northwest-1": "727897471807",
209
+ "eu-central-1": "763104351884",
210
+ "eu-central-2": "380420809688",
211
+ "eu-south-2": "503227376785",
212
+ "eu-west-1": "763104351884",
213
+ "eu-west-3": "763104351884",
214
+ "il-central-1": "780543022126",
215
+ "mx-central-1":"637423239942",
216
+ "sa-east-1": "763104351884",
217
+ "us-east-1": "763104351884",
218
+ "us-east-2": "763104351884",
219
+ "us-gov-east-1": "446045086412",
220
+ "us-gov-west-1": "442386744353",
221
+ "us-west-2": "763104351884",
222
+ "ca-west-1": "204538143572"
223
+ },
224
+ "container_version": {
225
+ "inf": "ubuntu20.04"
226
+ },
227
+ "sdk_versions": [
228
+ "sdk2.20.0"
229
+ ]
230
+ }
231
+ }
232
+ }
233
+ },
234
+ "inference": {
235
+ "processors": [
236
+ "inf"
237
+ ],
238
+ "version_aliases": {
239
+ "4.28": "4.28.1",
240
+ "4.34": "4.34.1",
241
+ "4.36": "4.36.2",
242
+ "4.43": "4.43.2"
243
+ },
244
+ "versions": {
245
+ "4.28.1": {
246
+ "version_aliases": {
247
+ "pytorch1.13": "pytorch1.13.0"
248
+ },
249
+ "pytorch1.13.0": {
250
+ "py_versions": [
251
+ "py38"
252
+ ],
253
+ "repository": "huggingface-pytorch-inference-neuronx",
254
+ "registries": {
255
+ "af-south-1": "626614931356",
256
+ "il-central-1": "780543022126",
257
+ "ap-east-1": "871362719292",
258
+ "ap-east-2": "975050140332",
259
+ "ap-northeast-1": "763104351884",
260
+ "ap-northeast-2": "763104351884",
261
+ "ap-northeast-3": "364406365360",
262
+ "ap-south-1": "763104351884",
263
+ "ap-south-2": "772153158452",
264
+ "ap-southeast-1": "763104351884",
265
+ "ap-southeast-2": "763104351884",
266
+ "ap-southeast-4": "457447274322",
267
+ "ap-southeast-5": "550225433462",
268
+ "ap-southeast-7": "590183813437",
269
+ "ca-central-1": "763104351884",
270
+ "cn-north-1": "727897471807",
271
+ "cn-northwest-1": "727897471807",
272
+ "eu-central-1": "763104351884",
273
+ "eu-central-2": "380420809688",
274
+ "eu-north-1": "763104351884",
275
+ "eu-west-1": "763104351884",
276
+ "eu-west-2": "763104351884",
277
+ "eu-west-3": "763104351884",
278
+ "eu-south-1": "692866216735",
279
+ "eu-south-2": "503227376785",
280
+ "me-south-1": "217643126080",
281
+ "mx-central-1":"637423239942",
282
+ "sa-east-1": "763104351884",
283
+ "us-east-1": "763104351884",
284
+ "us-east-2": "763104351884",
285
+ "us-gov-east-1": "446045086412",
286
+ "us-gov-west-1": "442386744353",
287
+ "us-iso-east-1": "886529160074",
288
+ "us-isob-east-1": "094389454867",
289
+ "us-west-1": "763104351884",
290
+ "us-west-2": "763104351884",
291
+ "ca-west-1": "204538143572"
292
+ },
293
+ "container_version": {
294
+ "inf": "ubuntu20.04"
295
+ },
296
+ "sdk_versions": [
297
+ "sdk2.9.1"
298
+ ]
299
+ }
300
+ },
301
+ "4.34.1": {
302
+ "version_aliases": {
303
+ "pytorch1.13": "pytorch1.13.1"
304
+ },
305
+ "pytorch1.13.1": {
306
+ "py_versions": [
307
+ "py310"
308
+ ],
309
+ "repository": "huggingface-pytorch-inference-neuronx",
310
+ "registries": {
311
+ "af-south-1": "626614931356",
312
+ "il-central-1": "780543022126",
313
+ "ap-east-1": "871362719292",
314
+ "ap-east-2": "975050140332",
315
+ "ap-northeast-1": "763104351884",
316
+ "ap-northeast-2": "763104351884",
317
+ "ap-northeast-3": "364406365360",
318
+ "ap-south-1": "763104351884",
319
+ "ap-south-2": "772153158452",
320
+ "ap-southeast-1": "763104351884",
321
+ "ap-southeast-2": "763104351884",
322
+ "ap-southeast-4": "457447274322",
323
+ "ap-southeast-5": "550225433462",
324
+ "ap-southeast-7": "590183813437",
325
+ "ca-central-1": "763104351884",
326
+ "cn-north-1": "727897471807",
327
+ "cn-northwest-1": "727897471807",
328
+ "eu-central-1": "763104351884",
329
+ "eu-central-2": "380420809688",
330
+ "eu-north-1": "763104351884",
331
+ "eu-west-1": "763104351884",
332
+ "eu-west-2": "763104351884",
333
+ "eu-west-3": "763104351884",
334
+ "eu-south-1": "692866216735",
335
+ "eu-south-2": "503227376785",
336
+ "me-south-1": "217643126080",
337
+ "mx-central-1":"637423239942",
338
+ "sa-east-1": "763104351884",
339
+ "us-east-1": "763104351884",
340
+ "us-east-2": "763104351884",
341
+ "us-gov-east-1": "446045086412",
342
+ "us-gov-west-1": "442386744353",
343
+ "us-iso-east-1": "886529160074",
344
+ "us-isob-east-1": "094389454867",
345
+ "us-west-1": "763104351884",
346
+ "us-west-2": "763104351884",
347
+ "ca-west-1": "204538143572"
348
+ },
349
+ "container_version": {
350
+ "inf": "ubuntu20.04"
351
+ },
352
+ "sdk_versions": [
353
+ "sdk2.15.0"
354
+ ]
355
+ }
356
+ },
357
+ "4.36.2": {
358
+ "version_aliases": {
359
+ "pytorch1.13": "pytorch1.13.1",
360
+ "pytorch2.1": "pytorch2.1.2"
361
+ },
362
+ "pytorch1.13.1": {
363
+ "py_versions": [
364
+ "py310"
365
+ ],
366
+ "repository": "huggingface-pytorch-inference-neuronx",
367
+ "registries": {
368
+ "af-south-1": "626614931356",
369
+ "il-central-1": "780543022126",
370
+ "ap-east-1": "871362719292",
371
+ "ap-east-2": "975050140332",
372
+ "ap-northeast-1": "763104351884",
373
+ "ap-northeast-2": "763104351884",
374
+ "ap-northeast-3": "364406365360",
375
+ "ap-south-1": "763104351884",
376
+ "ap-south-2": "772153158452",
377
+ "ap-southeast-1": "763104351884",
378
+ "ap-southeast-2": "763104351884",
379
+ "ap-southeast-4": "457447274322",
380
+ "ap-southeast-5": "550225433462",
381
+ "ap-southeast-7": "590183813437",
382
+ "ca-central-1": "763104351884",
383
+ "cn-north-1": "727897471807",
384
+ "cn-northwest-1": "727897471807",
385
+ "eu-central-1": "763104351884",
386
+ "eu-central-2": "380420809688",
387
+ "eu-north-1": "763104351884",
388
+ "eu-west-1": "763104351884",
389
+ "eu-west-2": "763104351884",
390
+ "eu-west-3": "763104351884",
391
+ "eu-south-1": "692866216735",
392
+ "eu-south-2": "503227376785",
393
+ "me-south-1": "217643126080",
394
+ "mx-central-1":"637423239942",
395
+ "sa-east-1": "763104351884",
396
+ "us-east-1": "763104351884",
397
+ "us-east-2": "763104351884",
398
+ "us-gov-east-1": "446045086412",
399
+ "us-gov-west-1": "442386744353",
400
+ "us-iso-east-1": "886529160074",
401
+ "us-isob-east-1": "094389454867",
402
+ "us-west-1": "763104351884",
403
+ "us-west-2": "763104351884",
404
+ "ca-west-1": "204538143572"
405
+ },
406
+ "container_version": {
407
+ "inf": "ubuntu20.04"
408
+ },
409
+ "sdk_versions": [
410
+ "sdk2.16.1"
411
+ ]
412
+ },
413
+ "pytorch2.1.2": {
414
+ "py_versions": [
415
+ "py310"
416
+ ],
417
+ "repository": "huggingface-pytorch-inference-neuronx",
418
+ "registries": {
419
+ "af-south-1": "626614931356",
420
+ "il-central-1": "780543022126",
421
+ "ap-east-1": "871362719292",
422
+ "ap-east-2": "975050140332",
423
+ "ap-northeast-1": "763104351884",
424
+ "ap-northeast-2": "763104351884",
425
+ "ap-northeast-3": "364406365360",
426
+ "ap-south-1": "763104351884",
427
+ "ap-south-2": "772153158452",
428
+ "ap-southeast-1": "763104351884",
429
+ "ap-southeast-2": "763104351884",
430
+ "ap-southeast-4": "457447274322",
431
+ "ap-southeast-5": "550225433462",
432
+ "ap-southeast-7": "590183813437",
433
+ "ca-central-1": "763104351884",
434
+ "cn-north-1": "727897471807",
435
+ "cn-northwest-1": "727897471807",
436
+ "eu-central-1": "763104351884",
437
+ "eu-central-2": "380420809688",
438
+ "eu-north-1": "763104351884",
439
+ "eu-west-1": "763104351884",
440
+ "eu-west-2": "763104351884",
441
+ "eu-west-3": "763104351884",
442
+ "eu-south-1": "692866216735",
443
+ "eu-south-2": "503227376785",
444
+ "me-south-1": "217643126080",
445
+ "mx-central-1":"637423239942",
446
+ "sa-east-1": "763104351884",
447
+ "us-east-1": "763104351884",
448
+ "us-east-2": "763104351884",
449
+ "us-gov-east-1": "446045086412",
450
+ "us-gov-west-1": "442386744353",
451
+ "us-iso-east-1": "886529160074",
452
+ "us-isob-east-1": "094389454867",
453
+ "us-west-1": "763104351884",
454
+ "us-west-2": "763104351884",
455
+ "ca-west-1": "204538143572"
456
+ },
457
+ "container_version": {
458
+ "inf": "ubuntu20.04"
459
+ },
460
+ "sdk_versions": [
461
+ "sdk2.18.0"
462
+ ]
463
+ }
464
+ },
465
+ "4.43.2": {
466
+ "version_aliases": {
467
+ "pytorch2.1": "pytorch2.1.2"
468
+ },
469
+ "pytorch2.1.2": {
470
+ "py_versions": [
471
+ "py310"
472
+ ],
473
+ "repository": "huggingface-pytorch-inference-neuronx",
474
+ "registries": {
475
+ "ap-northeast-1": "763104351884",
476
+ "ap-south-1": "763104351884",
477
+ "ap-south-2": "772153158452",
478
+ "ap-southeast-1": "763104351884",
479
+ "ap-southeast-2": "763104351884",
480
+ "ap-southeast-4": "457447274322",
481
+ "ap-southeast-5": "550225433462",
482
+ "ap-southeast-7": "590183813437",
483
+ "cn-north-1": "727897471807",
484
+ "cn-northwest-1": "727897471807",
485
+ "eu-central-1": "763104351884",
486
+ "eu-central-2": "380420809688",
487
+ "eu-south-2": "503227376785",
488
+ "eu-west-1": "763104351884",
489
+ "eu-west-3": "763104351884",
490
+ "il-central-1": "780543022126",
491
+ "mx-central-1":"637423239942",
492
+ "sa-east-1": "763104351884",
493
+ "us-east-1": "763104351884",
494
+ "us-east-2": "763104351884",
495
+ "us-gov-east-1": "446045086412",
496
+ "us-gov-west-1": "442386744353",
497
+ "us-west-2": "763104351884",
498
+ "ca-west-1": "204538143572"
499
+ },
500
+ "container_version": {
501
+ "inf": "ubuntu20.04"
502
+ },
503
+ "sdk_versions": [
504
+ "sdk2.20.0"
505
+ ]
506
+ }
507
+ }
508
+ }
509
+ }
510
+ }