├── .github └── workflows │ ├── ci-health.yml │ ├── codeql.yml │ ├── pr-checks-master-v2.yml │ ├── pr-checks-master.yml │ └── security-monitoring.yml ├── .gitignore ├── .pylintrc ├── CHANGELOG.md ├── CONTRIBUTING.md ├── LICENSE ├── README.rst ├── VERSION ├── pyproject.toml ├── requirements ├── extras │ └── test_requirements.txt └── tox │ ├── black_requirements.txt │ ├── doc8_requirements.txt │ ├── docstyle_requirements.txt │ ├── flake8_requirements.txt │ ├── mypy_requirements.txt │ ├── pydocstyle_requirements.txt │ ├── pylint_requirements.txt │ ├── spelling_requirements.txt │ └── twine_requirements.txt ├── sagemaker-core ├── .env ├── .pylintrc ├── .readthedocs.yaml ├── CHANGELOG.md ├── CODEOWNERS ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── README.rst ├── VERSION ├── branding │ └── icon │ │ └── sagemaker-banner.png ├── docs │ ├── Makefile │ ├── conf.py │ ├── index.rst │ ├── make.bat │ └── requirements.txt ├── example_notebooks │ ├── get_started.ipynb │ ├── images │ │ ├── experiment_created.png │ │ ├── experiment_run_analyze_plot.png │ │ ├── experiment_run_metrics.png │ │ ├── experiment_run_parameters.png │ │ └── experiment_runs_comparison.png │ ├── inference_and_resource_chaining.ipynb │ ├── intelligent_defaults_and_logging.ipynb │ ├── sagemaker-core-llama-3-8B-speculative-decoding.ipynb │ ├── sagemaker-core-llama-3-8B.ipynb │ ├── sagemaker_core_overview.ipynb │ └── track_local_pytorch_experiment.ipynb ├── pyproject.toml ├── resource_plan.csv ├── sample │ ├── sagemaker-a2i-runtime │ │ └── 2019-11-07 │ │ │ ├── endpoint-rule-set-1.json │ │ │ ├── examples-1.json │ │ │ ├── paginators-1.json │ │ │ └── service-2.json │ ├── sagemaker-edge │ │ └── 2020-09-23 │ │ │ ├── endpoint-rule-set-1.json │ │ │ ├── examples-1.json │ │ │ ├── paginators-1.json │ │ │ └── service-2.json │ ├── sagemaker-featurestore-runtime │ │ └── 2020-07-01 │ │ │ ├── endpoint-rule-set-1.json │ │ │ ├── examples-1.json │ │ │ ├── paginators-1.json │ │ │ └── service-2.json │ ├── sagemaker-geospatial │ │ └── 2020-05-27 │ │ │ ├── endpoint-rule-set-1.json │ │ │ ├── paginators-1.json │ │ │ └── service-2.json │ ├── sagemaker-metrics │ │ └── 2022-09-30 │ │ │ ├── endpoint-rule-set-1.json │ │ │ ├── paginators-1.json │ │ │ └── service-2.json │ ├── sagemaker-runtime │ │ └── 2017-05-13 │ │ │ ├── endpoint-rule-set-1.json │ │ │ ├── examples-1.json │ │ │ ├── paginators-1.json │ │ │ └── service-2.json │ └── sagemaker │ │ └── 2017-07-24 │ │ ├── default-configs.json │ │ ├── endpoint-rule-set-1.json │ │ ├── examples-1.json │ │ ├── paginators-1.json │ │ ├── service-2.json │ │ └── waiters-2.json ├── src │ └── sagemaker │ │ ├── core │ │ ├── __init__.py │ │ ├── _studio.py │ │ ├── _version.py │ │ ├── accept_types.py │ │ ├── analytics.py │ │ ├── apiutils │ │ │ ├── __init__.py │ │ │ ├── _base_types.py │ │ │ ├── _boto_functions.py │ │ │ └── _utils.py │ │ ├── base_deserializers.py │ │ ├── base_serializers.py │ │ ├── clarify │ │ │ └── __init__.py │ │ ├── collection.py │ │ ├── common_utils.py │ │ ├── compute_resource_requirements │ │ │ ├── __init__.py │ │ │ └── resource_requirements.py │ │ ├── config │ │ │ ├── __init__.py │ │ │ ├── config.py │ │ │ ├── config_manager.py │ │ │ ├── config_schema.py │ │ │ └── config_utils.py │ │ ├── config_schema.py │ │ ├── constants.py │ │ ├── content_types.py │ │ ├── debugger │ │ │ ├── __init__.py │ │ │ ├── debugger.py │ │ │ ├── framework_profile.py │ │ │ ├── metrics_config.py │ │ │ ├── profiler.py │ │ │ ├── profiler_config.py │ │ │ ├── profiler_constants.py │ │ │ └── utils.py │ │ ├── deprecations.py │ │ ├── deserializers │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ └── implementations.py │ │ ├── drift_check_baselines.py │ │ ├── enums.py │ │ ├── environment_variables.py │ │ ├── exceptions.py │ │ ├── experiments │ │ │ ├── __init__.py │ │ │ ├── _api_types.py │ │ │ ├── _environment.py │ │ │ ├── _helper.py │ │ │ ├── _metrics.py │ │ │ ├── _run_context.py │ │ │ ├── _utils.py │ │ │ ├── experiment.py │ │ │ ├── run.py │ │ │ ├── trial.py │ │ │ └── trial_component.py │ │ ├── explainer │ │ │ ├── __init__.py │ │ │ ├── clarify_explainer_config.py │ │ │ └── explainer_config.py │ │ ├── fw_utils.py │ │ ├── git_utils.py │ │ ├── helper │ │ │ ├── __init__.py │ │ │ ├── pipeline_variable.py │ │ │ └── session_helper.py │ │ ├── huggingface │ │ │ ├── __init__.py │ │ │ ├── llm_utils.py │ │ │ ├── processing.py │ │ │ └── training_compiler │ │ │ │ ├── __init__.py │ │ │ │ └── config.py │ │ ├── hyperparameters.py │ │ ├── image_retriever │ │ │ ├── __init__.py │ │ │ ├── image_retriever.py │ │ │ ├── image_retriever_utils.py │ │ │ ├── image_uri_config │ │ │ │ ├── autogluon.json │ │ │ │ ├── blazingtext.json │ │ │ │ ├── chainer.json │ │ │ │ ├── clarify.json │ │ │ │ ├── coach-mxnet.json │ │ │ │ ├── coach-tensorflow.json │ │ │ │ ├── data-wrangler.json │ │ │ │ ├── debugger.json │ │ │ │ ├── detailed-profiler.json │ │ │ │ ├── djl-deepspeed.json │ │ │ │ ├── djl-fastertransformer.json │ │ │ │ ├── djl-lmi.json │ │ │ │ ├── djl-neuronx.json │ │ │ │ ├── djl-tensorrtllm.json │ │ │ │ ├── factorization-machines.json │ │ │ │ ├── forecasting-deepar.json │ │ │ │ ├── huggingface-llm-neuronx.json │ │ │ │ ├── huggingface-llm.json │ │ │ │ ├── huggingface-neuron.json │ │ │ │ ├── huggingface-neuronx.json │ │ │ │ ├── huggingface-tei-cpu.json │ │ │ │ ├── huggingface-tei.json │ │ │ │ ├── huggingface-training-compiler.json │ │ │ │ ├── huggingface.json │ │ │ │ ├── hyperpod-recipes-neuron.json │ │ │ │ ├── image-classification-neo.json │ │ │ │ ├── image-classification.json │ │ │ │ ├── inferentia-mxnet.json │ │ │ │ ├── inferentia-pytorch.json │ │ │ │ ├── inferentia-tensorflow.json │ │ │ │ ├── instance_gpu_info.json │ │ │ │ ├── ipinsights.json │ │ │ │ ├── kmeans.json │ │ │ │ ├── knn.json │ │ │ │ ├── lda.json │ │ │ │ ├── linear-learner.json │ │ │ │ ├── model-monitor.json │ │ │ │ ├── mxnet.json │ │ │ │ ├── neo-mxnet.json │ │ │ │ ├── neo-pytorch.json │ │ │ │ ├── neo-tensorflow.json │ │ │ │ ├── ntm.json │ │ │ │ ├── object-detection.json │ │ │ │ ├── object2vec.json │ │ │ │ ├── pca.json │ │ │ │ ├── pytorch-neuron.json │ │ │ │ ├── pytorch-smp.json │ │ │ │ ├── pytorch-training-compiler.json │ │ │ │ ├── pytorch.json │ │ │ │ ├── randomcutforest.json │ │ │ │ ├── ray-pytorch.json │ │ │ │ ├── ray-tensorflow.json │ │ │ │ ├── sagemaker-base-python.json │ │ │ │ ├── sagemaker-distribution.json │ │ │ │ ├── sagemaker-geospatial.json │ │ │ │ ├── sagemaker-tritonserver.json │ │ │ │ ├── semantic-segmentation.json │ │ │ │ ├── seq2seq.json │ │ │ │ ├── sklearn.json │ │ │ │ ├── spark.json │ │ │ │ ├── sparkml-serving.json │ │ │ │ ├── stabilityai.json │ │ │ │ ├── tensorflow.json │ │ │ │ ├── vw.json │ │ │ │ ├── xgboost-neo.json │ │ │ │ └── xgboost.json │ │ │ └── test.py │ │ ├── image_uri_config │ │ │ ├── __init__.py │ │ │ ├── autogluon.json │ │ │ ├── blazingtext.json │ │ │ ├── chainer.json │ │ │ ├── clarify.json │ │ │ ├── coach-mxnet.json │ │ │ ├── coach-tensorflow.json │ │ │ ├── data-wrangler.json │ │ │ ├── debugger.json │ │ │ ├── detailed-profiler.json │ │ │ ├── djl-deepspeed.json │ │ │ ├── djl-fastertransformer.json │ │ │ ├── djl-lmi.json │ │ │ ├── djl-neuronx.json │ │ │ ├── djl-tensorrtllm.json │ │ │ ├── factorization-machines.json │ │ │ ├── forecasting-deepar.json │ │ │ ├── huggingface-llm-neuronx.json │ │ │ ├── huggingface-llm.json │ │ │ ├── huggingface-neuron.json │ │ │ ├── huggingface-neuronx.json │ │ │ ├── huggingface-tei-cpu.json │ │ │ ├── huggingface-tei.json │ │ │ ├── huggingface-training-compiler.json │ │ │ ├── huggingface.json │ │ │ ├── hyperpod-recipes-neuron.json │ │ │ ├── image-classification-neo.json │ │ │ ├── image-classification.json │ │ │ ├── inferentia-mxnet.json │ │ │ ├── inferentia-pytorch.json │ │ │ ├── inferentia-tensorflow.json │ │ │ ├── instance_gpu_info.json │ │ │ ├── ipinsights.json │ │ │ ├── kmeans.json │ │ │ ├── knn.json │ │ │ ├── lda.json │ │ │ ├── linear-learner.json │ │ │ ├── model-monitor.json │ │ │ ├── mxnet.json │ │ │ ├── neo-mxnet.json │ │ │ ├── neo-pytorch.json │ │ │ ├── neo-tensorflow.json │ │ │ ├── ntm.json │ │ │ ├── object-detection.json │ │ │ ├── object2vec.json │ │ │ ├── pca.json │ │ │ ├── pytorch-neuron.json │ │ │ ├── pytorch-smp.json │ │ │ ├── pytorch-training-compiler.json │ │ │ ├── pytorch.json │ │ │ ├── randomcutforest.json │ │ │ ├── ray-pytorch.json │ │ │ ├── ray-tensorflow.json │ │ │ ├── sagemaker-base-python.json │ │ │ ├── sagemaker-distribution.json │ │ │ ├── sagemaker-geospatial.json │ │ │ ├── sagemaker-tritonserver.json │ │ │ ├── semantic-segmentation.json │ │ │ ├── seq2seq.json │ │ │ ├── sklearn.json │ │ │ ├── spark.json │ │ │ ├── sparkml-serving.json │ │ │ ├── stabilityai.json │ │ │ ├── tensorflow.json │ │ │ ├── vw.json │ │ │ ├── xgboost-neo.json │ │ │ └── xgboost.json │ │ ├── image_uri_config_updated │ │ │ ├── autogluon.json │ │ │ ├── blazingtext.json │ │ │ ├── chainer.json │ │ │ ├── clarify.json │ │ │ ├── coach-mxnet.json │ │ │ ├── coach-tensorflow.json │ │ │ ├── data-wrangler.json │ │ │ ├── debugger.json │ │ │ ├── detailed-profiler.json │ │ │ ├── djl-deepspeed.json │ │ │ ├── djl-fastertransformer.json │ │ │ ├── djl-lmi.json │ │ │ ├── djl-neuronx.json │ │ │ ├── djl-tensorrtllm.json │ │ │ ├── factorization-machines.json │ │ │ ├── forecasting-deepar.json │ │ │ ├── huggingface-llm-neuronx.json │ │ │ ├── huggingface-llm.json │ │ │ ├── huggingface-neuron.json │ │ │ ├── huggingface-neuronx.json │ │ │ ├── huggingface-tei-cpu.json │ │ │ ├── huggingface-tei.json │ │ │ ├── huggingface-training-compiler.json │ │ │ ├── huggingface.json │ │ │ ├── hyperpod-recipes-neuron.json │ │ │ ├── image-classification-neo.json │ │ │ ├── image-classification.json │ │ │ ├── inferentia-mxnet.json │ │ │ ├── inferentia-pytorch.json │ │ │ ├── inferentia-tensorflow.json │ │ │ ├── instance_gpu_info.json │ │ │ ├── ipinsights.json │ │ │ ├── kmeans.json │ │ │ ├── knn.json │ │ │ ├── lda.json │ │ │ ├── linear-learner.json │ │ │ ├── model-monitor.json │ │ │ ├── mxnet.json │ │ │ ├── neo-mxnet.json │ │ │ ├── neo-pytorch.json │ │ │ ├── neo-tensorflow.json │ │ │ ├── ntm.json │ │ │ ├── object-detection.json │ │ │ ├── object2vec.json │ │ │ ├── pca.json │ │ │ ├── pytorch-neuron.json │ │ │ ├── pytorch-smp.json │ │ │ ├── pytorch-training-compiler.json │ │ │ ├── pytorch.json │ │ │ ├── randomcutforest.json │ │ │ ├── ray-pytorch.json │ │ │ ├── ray-tensorflow.json │ │ │ ├── sagemaker-base-python.json │ │ │ ├── sagemaker-distribution.json │ │ │ ├── sagemaker-geospatial.json │ │ │ ├── sagemaker-tritonserver.json │ │ │ ├── semantic-segmentation.json │ │ │ ├── seq2seq.json │ │ │ ├── sklearn.json │ │ │ ├── spark.json │ │ │ ├── sparkml-serving.json │ │ │ ├── stabilityai.json │ │ │ ├── tensorflow.json │ │ │ ├── vw.json │ │ │ ├── xgboost-neo.json │ │ │ └── xgboost.json │ │ ├── image_uris.py │ │ ├── inference_config.py │ │ ├── inference_recommender │ │ │ ├── __init__.py │ │ │ └── inference_recommender_mixin.py │ │ ├── inputs.py │ │ ├── instance_group.py │ │ ├── instance_types.py │ │ ├── instance_types_gpu_info.py │ │ ├── interactive_apps │ │ │ ├── __init__.py │ │ │ ├── base_interactive_app.py │ │ │ ├── detail_profiler_app.py │ │ │ └── tensorboard.py │ │ ├── iterators.py │ │ ├── job.py │ │ ├── jumpstart │ │ │ ├── __init__.py │ │ │ ├── accessors.py │ │ │ ├── artifacts │ │ │ │ ├── __init__.py │ │ │ │ ├── environment_variables.py │ │ │ │ ├── hyperparameters.py │ │ │ │ ├── image_uris.py │ │ │ │ ├── incremental_training.py │ │ │ │ ├── instance_types.py │ │ │ │ ├── kwargs.py │ │ │ │ ├── metric_definitions.py │ │ │ │ ├── model_packages.py │ │ │ │ ├── model_uris.py │ │ │ │ ├── payloads.py │ │ │ │ ├── predictors.py │ │ │ │ ├── resource_names.py │ │ │ │ ├── resource_requirements.py │ │ │ │ └── script_uris.py │ │ │ ├── cache.py │ │ │ ├── configs.py │ │ │ ├── constants.py │ │ │ ├── deserializers.py │ │ │ ├── document.py │ │ │ ├── enums.py │ │ │ ├── exceptions.py │ │ │ ├── factory │ │ │ │ ├── __init__.py │ │ │ │ └── utils.py │ │ │ ├── filters.py │ │ │ ├── hub │ │ │ │ ├── __init__.py │ │ │ │ ├── constants.py │ │ │ │ ├── hub.py │ │ │ │ ├── interfaces.py │ │ │ │ ├── parser_utils.py │ │ │ │ ├── parsers.py │ │ │ │ ├── types.py │ │ │ │ └── utils.py │ │ │ ├── models.py │ │ │ ├── notebook_utils.py │ │ │ ├── parameters.py │ │ │ ├── payload_utils.py │ │ │ ├── region_config.json │ │ │ ├── search.py │ │ │ ├── serializers.py │ │ │ ├── session_utils.py │ │ │ ├── types.py │ │ │ ├── utils.py │ │ │ └── validators.py │ │ ├── lambda_helper.py │ │ ├── lineage │ │ │ ├── __init__.py │ │ │ ├── _api_types.py │ │ │ ├── _utils.py │ │ │ ├── action.py │ │ │ ├── artifact.py │ │ │ ├── association.py │ │ │ ├── context.py │ │ │ ├── lineage_trial_component.py │ │ │ ├── query.py │ │ │ └── visualizer.py │ │ ├── local │ │ │ ├── __init__.py │ │ │ ├── data.py │ │ │ ├── entities.py │ │ │ ├── exceptions.py │ │ │ ├── image.py │ │ │ ├── local_session.py │ │ │ └── utils.py │ │ ├── logs.py │ │ ├── metadata_properties.py │ │ ├── metric_definitions.py │ │ ├── mlflow │ │ │ ├── __init__.py │ │ │ └── forward_sagemaker_metrics.py │ │ ├── model_card │ │ │ └── __init__.py │ │ ├── model_life_cycle.py │ │ ├── model_metrics.py │ │ ├── model_monitor │ │ │ ├── __init__.py │ │ │ ├── clarify_model_monitoring.py │ │ │ ├── cron_expression_generator.py │ │ │ ├── data_capture_config.py │ │ │ ├── data_quality_monitoring_config.py │ │ │ ├── dataset_format.py │ │ │ ├── model_monitoring.py │ │ │ ├── monitoring_alert.py │ │ │ ├── monitoring_files.py │ │ │ └── utils.py │ │ ├── model_registry.py │ │ ├── model_uris.py │ │ ├── modules │ │ │ ├── __init__.py │ │ │ ├── configs.py │ │ │ ├── constants.py │ │ │ ├── distributed.py │ │ │ ├── local_core │ │ │ │ ├── __init__.py │ │ │ │ └── local_container.py │ │ │ ├── templates.py │ │ │ ├── train │ │ │ │ ├── __init__.py │ │ │ │ ├── container_drivers │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── common │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── utils.py │ │ │ │ │ ├── distributed_drivers │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── basic_script_driver.py │ │ │ │ │ │ ├── mpi_driver.py │ │ │ │ │ │ ├── mpi_utils.py │ │ │ │ │ │ └── torchrun_driver.py │ │ │ │ │ └── scripts │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── environment.py │ │ │ │ └── sm_recipes │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── training_recipes.json │ │ │ │ │ └── utils.py │ │ │ ├── types.py │ │ │ └── utils.py │ │ ├── network.py │ │ ├── parameter.py │ │ ├── payloads.py │ │ ├── processing.py │ │ ├── remote_function │ │ │ ├── __init__.py │ │ │ ├── checkpoint_location.py │ │ │ ├── client.py │ │ │ ├── core │ │ │ │ ├── __init__.py │ │ │ │ ├── _custom_dispatch_table.py │ │ │ │ ├── pipeline_variables.py │ │ │ │ ├── serialization.py │ │ │ │ └── stored_function.py │ │ │ ├── custom_file_filter.py │ │ │ ├── errors.py │ │ │ ├── invoke_function.py │ │ │ ├── job.py │ │ │ ├── logging_config.py │ │ │ ├── runtime_environment │ │ │ │ ├── __init__.py │ │ │ │ ├── bootstrap_runtime_environment.py │ │ │ │ ├── mpi_utils_remote.py │ │ │ │ ├── runtime_environment_manager.py │ │ │ │ └── spark_app.py │ │ │ └── spark_config.py │ │ ├── resource_requirements.py │ │ ├── resources.py │ │ ├── s3 │ │ │ ├── __init__.py │ │ │ ├── client.py │ │ │ └── utils.py │ │ ├── script_uris.py │ │ ├── serializers │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── implementations.py │ │ │ └── utils.py │ │ ├── serverless_inference_config.py │ │ ├── session_settings.py │ │ ├── shapes │ │ │ ├── __init__.py │ │ │ ├── model_card_shapes.py │ │ │ └── shapes.py │ │ ├── spark │ │ │ ├── __init__.py │ │ │ ├── defaults.py │ │ │ └── processing.py │ │ ├── telemetry │ │ │ ├── __init__.py │ │ │ ├── constants.py │ │ │ └── telemetry_logging.py │ │ ├── tools │ │ │ ├── __init__.py │ │ │ ├── additional_operations.json │ │ │ ├── api_coverage.json │ │ │ ├── codegen.py │ │ │ ├── constants.py │ │ │ ├── data_extractor.py │ │ │ ├── method.py │ │ │ ├── model_card │ │ │ │ ├── generate_model_card_from_schema.py │ │ │ │ └── model_card_schema.json │ │ │ ├── resource_plan.csv │ │ │ ├── resources_codegen.py │ │ │ ├── resources_extractor.py │ │ │ ├── shapes_codegen.py │ │ │ ├── shapes_extractor.py │ │ │ └── templates.py │ │ ├── training │ │ │ ├── __init__.py │ │ │ ├── configs.py │ │ │ ├── constants.py │ │ │ └── utils.py │ │ ├── training_compiler │ │ │ ├── __init__.py │ │ │ └── config.py │ │ ├── training_compiler_config.py │ │ ├── transformer.py │ │ ├── user_agent.py │ │ ├── utilities │ │ │ ├── __init__.py │ │ │ ├── cache.py │ │ │ └── search_expression.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── code_injection │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── codec.py │ │ │ │ ├── constants.py │ │ │ │ └── shape_dag.py │ │ │ ├── exceptions.py │ │ │ ├── intelligent_defaults_helper.py │ │ │ ├── logs.py │ │ │ ├── user_agent.py │ │ │ └── utils.py │ │ └── workflow │ │ │ ├── __init__.py │ │ │ ├── conditions.py │ │ │ ├── entities.py │ │ │ ├── execution_variables.py │ │ │ ├── functions.py │ │ │ ├── parameters.py │ │ │ ├── pipeline_context.py │ │ │ ├── pipeline_definition_config.py │ │ │ ├── properties.py │ │ │ ├── step_outputs.py │ │ │ └── utilities.py │ │ └── lineage │ │ ├── __init__.py │ │ ├── action.py │ │ ├── artifact.py │ │ ├── context.py │ │ └── lineage_trial_component.py ├── tests │ ├── data │ │ ├── config │ │ │ ├── config.yaml │ │ │ ├── config_file_with_code.yaml │ │ │ ├── expected_output_config_after_merge.yaml │ │ │ ├── invalid_config_file.yaml │ │ │ ├── sample_additional_config_for_merge.yaml │ │ │ └── sample_config_for_merge.yaml │ │ ├── cuteCat.jpg │ │ ├── cuteCat.raw │ │ ├── with_integers.csv │ │ └── xgboost_abalone │ │ │ └── abalone │ ├── integ │ │ ├── image_retriever │ │ │ └── test_image_retriever.py │ │ └── jumpstart │ │ │ ├── __init__.py │ │ │ ├── test_model.py │ │ │ └── test_search_integ.py │ └── unit │ │ ├── __init__.py │ │ ├── config │ │ ├── conftest.py │ │ ├── test_config_manager.py │ │ └── test_config_schema.py │ │ ├── conftest.py │ │ ├── deserializers │ │ ├── __init__.py │ │ └── test_base_deserializers.py │ │ ├── generated │ │ ├── __init__.py │ │ ├── resource_test_config.json │ │ ├── test_config_schema.py │ │ ├── test_logs.py │ │ ├── test_model_card.py │ │ ├── test_resources.py │ │ ├── test_shapes.py │ │ ├── test_user_agent.py │ │ └── test_utils.py │ │ ├── helper │ │ └── test_session_helper.py │ │ ├── interactive_apps │ │ ├── __init__.py │ │ ├── test_profiler_app.py │ │ └── test_tensorboard.py │ │ ├── jumpstart │ │ ├── __init__.py │ │ ├── hub │ │ │ ├── test_interfaces.py │ │ │ └── test_parsers.py │ │ ├── hub_content_document.json │ │ ├── test_cache.py │ │ ├── test_document.py │ │ ├── test_factory_utils.py │ │ ├── test_filters.py │ │ ├── test_models.py │ │ ├── test_notebook_utils.py │ │ ├── test_search_unit.py │ │ └── test_utils_extended.py │ │ ├── lineage │ │ └── test_query.py │ │ ├── local │ │ ├── __init__.py │ │ ├── test_entities.py │ │ ├── test_image.py │ │ ├── test_local_session.py │ │ └── test_local_utils.py │ │ ├── model_monitor │ │ ├── __init__.py │ │ ├── test_clarify_model_monitoring.py │ │ └── test_utils.py │ │ ├── modules │ │ ├── __init__.py │ │ ├── local_core │ │ │ ├── __init__.py │ │ │ └── test_local_container.py │ │ └── train │ │ │ ├── __init__.py │ │ │ ├── container_drivers │ │ │ ├── __init__.py │ │ │ └── distributed_drivers │ │ │ │ ├── __init__.py │ │ │ │ └── test_mpi_utils.py │ │ │ ├── test_environment.py │ │ │ └── test_sm_recipes_utils.py │ │ ├── remote_function │ │ ├── __init__.py │ │ ├── runtime_environment │ │ │ ├── __init__.py │ │ │ ├── test_bootstrap_runtime_environment.py │ │ │ ├── test_mpi_utils_remote.py │ │ │ └── test_runtime_environment_manager.py │ │ ├── test_client.py │ │ ├── test_job.py │ │ └── test_job_comprehensive.py │ │ ├── s3 │ │ └── test_client.py │ │ ├── serializers │ │ ├── __init__.py │ │ ├── test_base_serializers.py │ │ └── test_utils.py │ │ ├── session │ │ ├── __init__.py │ │ ├── conftest.py │ │ ├── test_session_bucket_operations.py │ │ ├── test_session_helper.py │ │ └── test_session_identity.py │ │ ├── spark │ │ └── __init__.py │ │ ├── telemetry │ │ ├── __init__.py │ │ └── test_telemetry_logging.py │ │ ├── test_accept_types.py │ │ ├── test_analytics.py │ │ ├── test_base_deserializers.py │ │ ├── test_clarify.py │ │ ├── test_codec.py │ │ ├── test_collection.py │ │ ├── test_common_utils.py │ │ ├── test_constants.py │ │ ├── test_content_types.py │ │ ├── test_deserializer_implementations.py │ │ ├── test_drift_check_baselines.py │ │ ├── test_fw_utils.py │ │ ├── test_git_utils.py │ │ ├── test_hyperparameters.py │ │ ├── test_image_retriever.py │ │ ├── test_image_retriever_utils.py │ │ ├── test_image_uris.py │ │ ├── test_inference_recommender_mixin.py │ │ ├── test_instance_group.py │ │ ├── test_iterators.py │ │ ├── test_job.py │ │ ├── test_jumpstart_types.py │ │ ├── test_jumpstart_types_coverage.py │ │ ├── test_jumpstart_types_extended.py │ │ ├── test_jumpstart_utils.py │ │ ├── test_lambda_helper.py │ │ ├── test_metadata_properties.py │ │ ├── test_metric_definitions.py │ │ ├── test_model_life_cycle.py │ │ ├── test_model_monitoring.py │ │ ├── test_model_registry.py │ │ ├── test_model_uris.py │ │ ├── test_modules_constants.py │ │ ├── test_payloads.py │ │ ├── test_processing.py │ │ ├── test_profiler_constants.py │ │ ├── test_resource_requirements.py │ │ ├── test_script_uris.py │ │ ├── test_serializer_implementations.py │ │ ├── test_serverless_inference_config.py │ │ ├── test_training_constants.py │ │ ├── test_transformer.py │ │ ├── test_version.py │ │ ├── tools │ │ ├── __init__.py │ │ ├── test_api_coverage.py │ │ ├── test_resources_extractor.py │ │ ├── test_shapes_codegen.py │ │ └── test_shapes_extractor.py │ │ ├── utils │ │ ├── __init__.py │ │ └── test_intelligent_defaults_helper.py │ │ └── workflow │ │ ├── __init__.py │ │ └── test_utilities.py ├── tox.ini └── workflow_helper │ ├── __init__.py │ ├── compute_boto_api_coverage.py │ └── compute_resource_coverage.py ├── sagemaker-mlops ├── CHANGELOG.md ├── README.md ├── VERSION ├── pyproject.toml ├── src │ └── sagemaker │ │ ├── __init__.py │ │ └── mlops │ │ ├── __init__.py │ │ ├── local │ │ ├── __init__.py │ │ ├── exceptions.py │ │ ├── local_pipeline_session.py │ │ ├── pipeline.py │ │ └── pipeline_entities.py │ │ └── workflow │ │ ├── __init__.py │ │ ├── _event_bridge_client_helper.py │ │ ├── _repack_model.py │ │ ├── _steps_compiler.py │ │ ├── _utils.py │ │ ├── automl_step.py │ │ ├── callback_step.py │ │ ├── check_job_config.py │ │ ├── clarify_check_step.py │ │ ├── condition_step.py │ │ ├── emr_step.py │ │ ├── fail_step.py │ │ ├── function_step.py │ │ ├── lambda_step.py │ │ ├── model_step.py │ │ ├── monitor_batch_transform_step.py │ │ ├── notebook_job_step.py │ │ ├── parallelism_config.py │ │ ├── pipeline.py │ │ ├── pipeline_experiment_config.py │ │ ├── quality_check_step.py │ │ ├── retry.py │ │ ├── selective_execution_config.py │ │ ├── step_collections.py │ │ ├── steps.py │ │ └── triggers.py ├── tests │ ├── integ │ │ ├── __init__.py │ │ ├── code │ │ │ ├── mnist.py │ │ │ ├── pipeline │ │ │ │ └── preprocess.py │ │ │ ├── preprocess.py │ │ │ └── pytorch_processing │ │ │ │ ├── preprocessing.py │ │ │ │ └── requirements.txt │ │ ├── conftest.py │ │ ├── data │ │ │ ├── abalone.csv │ │ │ ├── mnist │ │ │ │ └── MNIST │ │ │ │ │ └── raw │ │ │ │ │ ├── t10k-images-idx3-ubyte.gz │ │ │ │ │ ├── t10k-labels-idx1-ubyte.gz │ │ │ │ │ ├── train-images-idx3-ubyte.gz │ │ │ │ │ └── train-labels-idx1-ubyte.gz │ │ │ ├── model │ │ │ │ ├── registry-test-model.tar.gz │ │ │ │ └── transform_job │ │ │ │ │ └── xgb-churn-prediction-model.tar.gz │ │ │ ├── pipeline │ │ │ │ └── abalone.csv │ │ │ └── validation.csv │ │ ├── test_clarify.py │ │ ├── test_hyperparameter_tuning.py │ │ ├── test_model_registry.py │ │ ├── test_pipeline_train_registry.py │ │ ├── test_processing_job_sklearn.py │ │ ├── test_pytorch_processing.py │ │ └── test_transform_job.py │ └── unit │ │ ├── local │ │ ├── __init__.py │ │ ├── test_exceptions.py │ │ ├── test_local_pipeline_session.py │ │ ├── test_pipeline.py │ │ ├── test_pipeline_entities.py │ │ └── test_pipeline_executor.py │ │ └── workflow │ │ ├── __init__.py │ │ ├── test_callback_step.py │ │ ├── test_check_job_config.py │ │ ├── test_clarify_check_step.py │ │ ├── test_condition_step.py │ │ ├── test_emr_step.py │ │ ├── test_fail_step.py │ │ ├── test_function_step.py │ │ ├── test_lambda_step.py │ │ ├── test_model_step.py │ │ ├── test_monitor_batch_transform_step.py │ │ ├── test_notebook_job_step.py │ │ ├── test_parallelism_config.py │ │ ├── test_pipeline.py │ │ ├── test_pipeline_class.py │ │ ├── test_pipeline_experiment_config.py │ │ ├── test_quality_check_step.py │ │ ├── test_repack_model.py │ │ ├── test_retry.py │ │ ├── test_selective_execution_config.py │ │ ├── test_step_collections.py │ │ ├── test_steps.py │ │ ├── test_steps_compiler.py │ │ ├── test_triggers.py │ │ ├── test_tuning_step.py │ │ └── test_utils.py └── tox.ini ├── sagemaker-serve ├── CHANGELOG.md ├── README.md ├── VERSION ├── pyproject.toml ├── src │ └── sagemaker │ │ └── serve │ │ ├── __init__.py │ │ ├── async_inference │ │ ├── __init__.py │ │ ├── async_inference_config.py │ │ ├── async_inference_response.py │ │ └── waiter_config.py │ │ ├── batch_inference │ │ ├── __init__.py │ │ └── batch_transform_inference_config.py │ │ ├── bedrock_model_builder.py │ │ ├── builder │ │ ├── __init__.py │ │ ├── requirements_manager.py │ │ ├── schema_builder.py │ │ ├── serve_settings.py │ │ └── triton_schema_builder.py │ │ ├── compute_resource_requirements │ │ ├── __init__.py │ │ └── resource_requirements.py │ │ ├── configs.py │ │ ├── constants.py │ │ ├── deployment_progress.py │ │ ├── detector │ │ ├── __init__.py │ │ ├── dependency_manager.py │ │ ├── image_detector.py │ │ ├── pickle_dependencies.py │ │ └── pickler.py │ │ ├── inference_recommendation_mixin.py │ │ ├── local_resources.py │ │ ├── marshalling │ │ ├── __init__.py │ │ ├── custom_payload_translator.py │ │ └── triton_translator.py │ │ ├── mode │ │ ├── __init__.py │ │ ├── function_pointers.py │ │ ├── in_process_mode.py │ │ ├── local_container_mode.py │ │ └── sagemaker_endpoint_mode.py │ │ ├── model_builder.py │ │ ├── model_builder_servers.py │ │ ├── model_builder_utils.py │ │ ├── model_format │ │ ├── __init__.py │ │ └── mlflow │ │ │ ├── __init__.py │ │ │ ├── constants.py │ │ │ └── utils.py │ │ ├── model_server │ │ ├── __init__.py │ │ ├── djl_serving │ │ │ ├── __init__.py │ │ │ ├── prepare.py │ │ │ ├── server.py │ │ │ └── utils.py │ │ ├── in_process_model_server │ │ │ ├── __init__.py │ │ │ ├── app.py │ │ │ └── in_process_server.py │ │ ├── multi_model_server │ │ │ ├── __init__.py │ │ │ ├── inference.py │ │ │ ├── prepare.py │ │ │ └── server.py │ │ ├── smd │ │ │ ├── custom_execution_inference.py │ │ │ ├── prepare.py │ │ │ └── server.py │ │ ├── tei │ │ │ ├── __init__.py │ │ │ └── server.py │ │ ├── tensorflow_serving │ │ │ ├── __init__.py │ │ │ ├── inference.py │ │ │ ├── prepare.py │ │ │ └── server.py │ │ ├── tgi │ │ │ ├── __init__.py │ │ │ ├── prepare.py │ │ │ ├── server.py │ │ │ └── utils.py │ │ ├── torchserve │ │ │ ├── __init__.py │ │ │ ├── inference.py │ │ │ ├── prepare.py │ │ │ ├── server.py │ │ │ └── xgboost_inference.py │ │ └── triton │ │ │ ├── __init__.py │ │ │ ├── config_template.py │ │ │ ├── model.py │ │ │ ├── pack_conda_env.sh │ │ │ └── server.py │ │ ├── predictor_async.py │ │ ├── schema │ │ └── task.json │ │ ├── serverless │ │ ├── __init__.py │ │ ├── model.py │ │ └── serverless_inference_config.py │ │ ├── spec │ │ ├── __init__.py │ │ ├── inference_base.py │ │ └── inference_spec.py │ │ ├── utils │ │ ├── __init__.py │ │ ├── conda_in_process.yml │ │ ├── exceptions.py │ │ ├── hardware_detector.py │ │ ├── hf_utils.py │ │ ├── in_process_requirements.txt │ │ ├── lineage_constants.py │ │ ├── lineage_utils.py │ │ ├── local_hardware.py │ │ ├── logging_agent.py │ │ ├── packaging.py │ │ ├── task.py │ │ ├── telemetry_logger.py │ │ ├── types.py │ │ └── uploader.py │ │ └── validations │ │ ├── __init__.py │ │ ├── check_image_and_hardware_type.py │ │ ├── check_image_uri.py │ │ ├── check_integrity.py │ │ ├── optimization.py │ │ └── parse_registry_accounts.py ├── tests │ ├── integ │ │ ├── test_huggingface_integration.py │ │ ├── test_in_process_integration.py │ │ ├── test_jumpstart_integration.py │ │ ├── test_model_customization_deployment.py │ │ ├── test_optimize_integration.py │ │ ├── test_tei_integration.py │ │ ├── test_tgi_integration.py │ │ ├── test_train_inference_e2e_integration.py │ │ └── test_triton_integration.py │ └── unit │ │ ├── __init__.py │ │ ├── async_inference │ │ ├── __init__.py │ │ ├── test_async_inference_response.py │ │ └── test_async_inference_response_additional.py │ │ ├── batch_inference │ │ ├── __init__.py │ │ └── test_batch_transform_inference_config.py │ │ ├── builder │ │ ├── __init__.py │ │ ├── test_requirements_manager.py │ │ ├── test_schema_builder.py │ │ ├── test_serve_settings.py │ │ └── test_triton_schema_builder.py │ │ ├── compute_resource_requirements │ │ ├── __init__.py │ │ └── test_resource_requirements.py │ │ ├── detector │ │ ├── __init__.py │ │ ├── test_dependency_manager.py │ │ ├── test_image_detector.py │ │ ├── test_pickle_dependencies.py │ │ ├── test_pickle_dependencies_additional.py │ │ └── test_pickler.py │ │ ├── marshalling │ │ ├── __init__.py │ │ ├── test_custom_payload_translator.py │ │ └── test_triton_translator.py │ │ ├── mb_user_test.py │ │ ├── mode │ │ ├── __init__.py │ │ └── test_function_pointers.py │ │ ├── model_format │ │ ├── __init__.py │ │ ├── test_mlflow_constants.py │ │ └── test_mlflow_utils.py │ │ ├── model_server │ │ ├── __init__.py │ │ ├── test_djl_utils.py │ │ ├── test_in_process_model_server_app.py │ │ ├── test_multi_model_server_inference.py │ │ ├── test_multi_model_server_prepare.py │ │ ├── test_multi_model_server_server.py │ │ ├── test_smd_prepare.py │ │ ├── test_smd_server.py │ │ ├── test_tei_server.py │ │ ├── test_tensorflow_serving_inference.py │ │ ├── test_tensorflow_serving_prepare.py │ │ ├── test_tensorflow_serving_server.py │ │ ├── test_tgi_prepare.py │ │ ├── test_tgi_server.py │ │ ├── test_tgi_utils.py │ │ ├── test_torchserve_inference.py │ │ ├── test_torchserve_prepare.py │ │ ├── test_torchserve_server.py │ │ └── test_torchserve_xgboost_inference.py │ │ ├── run_all_tests.py │ │ ├── serverless │ │ ├── __init__.py │ │ └── test_serverless_inference_config.py │ │ ├── servers │ │ └── test_model_builder_servers.py │ │ ├── spec │ │ ├── __init__.py │ │ ├── test_inference_base.py │ │ ├── test_inference_base_additional.py │ │ └── test_inference_spec.py │ │ ├── test_bedrock_model_builder.py │ │ ├── test_configs.py │ │ ├── test_constants.py │ │ ├── test_deployment_progress.py │ │ ├── test_deployment_progress_additional.py │ │ ├── test_fixtures.py │ │ ├── test_inference_recommendation_mixin.py │ │ ├── test_local_resources.py │ │ ├── test_model_builder.py │ │ ├── test_model_builder_additional.py │ │ ├── test_model_builder_advanced.py │ │ ├── test_model_builder_build.py │ │ ├── test_model_builder_core.py │ │ ├── test_model_builder_coverage_boost.py │ │ ├── test_model_builder_deploy.py │ │ ├── test_model_builder_integration.py │ │ ├── test_model_builder_methods.py │ │ ├── test_model_builder_missing_coverage.py │ │ ├── test_model_builder_servers.py │ │ ├── test_model_builder_servers_coverage.py │ │ ├── test_model_builder_utils.py │ │ ├── test_model_builder_utils_additional.py │ │ ├── test_model_builder_utils_additional_gaps.py │ │ ├── test_model_builder_utils_coverage.py │ │ ├── test_model_builder_utils_extended_coverage.py │ │ ├── test_model_builder_utils_final_gaps.py │ │ ├── test_model_builder_utils_methods.py │ │ ├── test_model_builder_utils_new.py │ │ ├── test_model_builder_utils_optimization.py │ │ ├── test_model_builder_utils_triton.py │ │ ├── test_model_builder_v3.py │ │ ├── test_model_builder_v3_extended.py │ │ ├── test_model_builder_workflows.py │ │ ├── test_parse_registry_accounts.py │ │ ├── test_predictor_async.py │ │ ├── test_telemetry_logger.py │ │ ├── utils │ │ ├── __init__.py │ │ ├── test_exceptions.py │ │ ├── test_hardware_detector.py │ │ ├── test_hf_utils.py │ │ ├── test_lineage_constants.py │ │ ├── test_lineage_utils.py │ │ ├── test_local_hardware.py │ │ ├── test_local_hardware_additional.py │ │ ├── test_logging_agent.py │ │ ├── test_packaging.py │ │ ├── test_task.py │ │ ├── test_telemetry_logger_additional.py │ │ ├── test_types.py │ │ └── test_uploader.py │ │ └── validations │ │ ├── __init__.py │ │ ├── test_check_image_and_hardware_type.py │ │ ├── test_check_image_uri.py │ │ ├── test_check_integrity.py │ │ ├── test_optimization.py │ │ └── test_parse_registry_accounts.py └── tox.ini ├── sagemaker-train ├── CHANGELOG.md ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── README.rst ├── VERSION ├── example_notebooks │ └── evaluate │ │ ├── benchmark_demo.ipynb │ │ ├── custom_scorer_demo.ipynb │ │ └── llm_as_judge_demo.ipynb ├── pyproject.toml ├── src │ └── sagemaker │ │ ├── ai_registry │ │ ├── __init__.py │ │ ├── air_constants.py │ │ ├── air_hub.py │ │ ├── air_hub_entity.py │ │ ├── air_utils.py │ │ ├── dataset.py │ │ ├── dataset_utils.py │ │ ├── dataset_validation.py │ │ ├── evaluator.py │ │ └── utils.py │ │ └── train │ │ ├── __init__.py │ │ ├── base_trainer.py │ │ ├── common.py │ │ ├── common_utils │ │ ├── constants.py │ │ ├── finetune_utils.py │ │ ├── get_mlflow_endpoint.py │ │ ├── mlflow_metrics_util.py │ │ ├── model_resolution.py │ │ ├── recipe_utils.py │ │ ├── show_results_utils.py │ │ └── trainer_wait.py │ │ ├── configs.py │ │ ├── constants.py │ │ ├── container_drivers │ │ ├── __init__.py │ │ ├── common │ │ │ ├── __init__.py │ │ │ └── utils.py │ │ ├── distributed_drivers │ │ │ ├── __init__.py │ │ │ ├── basic_script_driver.py │ │ │ ├── mpi_driver.py │ │ │ ├── mpi_utils.py │ │ │ └── torchrun_driver.py │ │ └── scripts │ │ │ ├── __init__.py │ │ │ └── environment.py │ │ ├── defaults.py │ │ ├── distributed.py │ │ ├── dpo_trainer.py │ │ ├── evaluate │ │ ├── Resolved template parameters: {'role_arn.md │ │ ├── __init__.py │ │ ├── base_evaluator.py │ │ ├── benchmark_evaluator.py │ │ ├── constants.py │ │ ├── custom_scorer_evaluator.py │ │ ├── execution.py │ │ ├── llm_as_judge_evaluator.py │ │ └── pipeline_templates.py │ │ ├── local │ │ ├── __init__.py │ │ ├── data.py │ │ ├── entities.py │ │ └── local_container.py │ │ ├── model_trainer.py │ │ ├── modules │ │ └── model_trainer.py │ │ ├── remote_function │ │ ├── __init__.py │ │ ├── checkpoint_location.py │ │ ├── client.py │ │ ├── core │ │ │ ├── __init__.py │ │ │ ├── _custom_dispatch_table.py │ │ │ ├── pipeline_variables.py │ │ │ ├── serialization.py │ │ │ └── stored_function.py │ │ ├── custom_file_filter.py │ │ ├── errors.py │ │ ├── invoke_function.py │ │ ├── job.py │ │ ├── logging_config.py │ │ ├── runtime_environment │ │ │ ├── __init__.py │ │ │ ├── bootstrap_runtime_environment.py │ │ │ ├── mpi_utils_remote.py │ │ │ ├── runtime_environment_manager.py │ │ │ └── spark_app.py │ │ └── spark_config.py │ │ ├── rlaif_trainer.py │ │ ├── rlvr_trainer.py │ │ ├── sft_trainer.py │ │ ├── sm_recipes │ │ ├── __init__.py │ │ ├── training_recipes.json │ │ └── utils.py │ │ ├── templates.py │ │ ├── tuner.py │ │ ├── types.py │ │ └── utils.py ├── tests │ ├── __init__.py │ ├── data │ │ ├── __init__.py │ │ ├── _repack_model.py │ │ ├── custom_drivers │ │ │ ├── __init__.py │ │ │ └── driver.py │ │ ├── local_script │ │ │ ├── __init__.py │ │ │ ├── data │ │ │ │ ├── test │ │ │ │ │ ├── x_test.npy │ │ │ │ │ └── y_test.npy │ │ │ │ └── train │ │ │ │ │ ├── x_train.npy │ │ │ │ │ └── y_train.npy │ │ │ ├── local_training_script.py │ │ │ └── pytorch_model_def.py │ │ ├── params_script │ │ │ ├── __init__.py │ │ │ ├── hyperparameters.json │ │ │ ├── hyperparameters.yaml │ │ │ ├── requirements.txt │ │ │ ├── train.py │ │ │ └── train.sh │ │ ├── remote_function │ │ │ ├── __init__.py │ │ │ ├── config.yaml │ │ │ ├── non_existent_requirements.txt │ │ │ ├── old_deps_requirements.txt │ │ │ ├── pre_exec_commands │ │ │ ├── pre_exec_commands_bad_cmd │ │ │ ├── requirements.txt │ │ │ └── workdir │ │ │ │ ├── __init__.py │ │ │ │ ├── data │ │ │ │ └── data.csv │ │ │ │ ├── module │ │ │ │ ├── __init__.py │ │ │ │ └── train.py │ │ │ │ └── test │ │ │ │ ├── __init__.py │ │ │ │ └── test.py │ │ ├── script_mode │ │ │ ├── __init__.py │ │ │ ├── code.tar.gz │ │ │ ├── custom_script.py │ │ │ ├── data │ │ │ │ ├── test │ │ │ │ │ ├── x_test.npy │ │ │ │ │ └── y_test.npy │ │ │ │ └── train │ │ │ │ │ ├── x_train.npy │ │ │ │ │ └── y_train.npy │ │ │ ├── pytorch_model_def.py │ │ │ └── requirements.txt │ │ └── scripts │ │ │ ├── __init__.py │ │ │ └── entry_script.py │ ├── integ │ │ ├── __init__.py │ │ ├── ai_registry │ │ │ ├── __init__.py │ │ │ ├── conftest.py │ │ │ ├── test_air_hub.py │ │ │ ├── test_dataset.py │ │ │ └── test_evaluator.py │ │ ├── jumpstart │ │ │ ├── __init__.py │ │ │ └── test_jumpstart_train.py │ │ ├── lock.py │ │ └── train │ │ │ ├── __init__.py │ │ │ ├── conftest.py │ │ │ ├── test_benchmark_evaluator.py │ │ │ ├── test_custom_scorer_evaluator.py │ │ │ ├── test_dpo_trainer_integration.py │ │ │ ├── test_llm_as_judge_evaluator.py │ │ │ ├── test_local_model_trainer.py │ │ │ ├── test_model_trainer.py │ │ │ ├── test_rlaif_trainer_integration.py │ │ │ ├── test_rlvr_trainer_integration.py │ │ │ └── test_sft_trainer_integration.py │ └── unit │ │ ├── __init__.py │ │ ├── ai_registry │ │ ├── __init__.py │ │ ├── test_air_hub.py │ │ ├── test_air_hub_entity.py │ │ ├── test_dataset.py │ │ ├── test_dataset_domain_id.py │ │ ├── test_dataset_utils.py │ │ ├── test_dataset_validation.py │ │ ├── test_evaluator.py │ │ └── test_evaluator_domain_id.py │ │ └── train │ │ ├── __init__.py │ │ ├── common_utils │ │ ├── __init__.py │ │ ├── test_finetune_utils.py │ │ ├── test_get_mlflow_endpoint.py │ │ ├── test_mlflow_metrics_util.py │ │ ├── test_model_resolution.py │ │ ├── test_recipe_utils.py │ │ ├── test_show_results_utils.py │ │ └── test_trainer_wait.py │ │ ├── container_drivers │ │ ├── __init__.py │ │ ├── scripts │ │ │ └── test_enviornment.py │ │ ├── test_basic_script_driver.py │ │ ├── test_mpi_driver.py │ │ ├── test_mpi_utils.py │ │ ├── test_torchrun_driver.py │ │ └── test_utils.py │ │ ├── evaluate │ │ ├── __init__.py │ │ ├── test_base_evaluator.py │ │ ├── test_benchmark_evaluator.py │ │ ├── test_constants.py │ │ ├── test_custom_scorer_evaluator.py │ │ ├── test_execution.py │ │ ├── test_init.py │ │ ├── test_llm_as_judge_evaluator.py │ │ └── test_pipeline_templates.py │ │ ├── local │ │ ├── __init__.py │ │ ├── test_data.py │ │ └── test_entities.py │ │ ├── remote_function │ │ ├── __init__.py │ │ ├── test_bootstrap_runtime_environment.py │ │ ├── test_checkpoint_location.py │ │ ├── test_custom_file_filter.py │ │ ├── test_invoke_function.py │ │ ├── test_logging_config.py │ │ ├── test_mpi_utils_remote.py │ │ └── test_runtime_environment_manager.py │ │ ├── sm_recipes │ │ ├── __init__.py │ │ └── test_utils.py │ │ ├── test_defaults.py │ │ ├── test_dpo_trainer.py │ │ ├── test_model_trainer.py │ │ ├── test_rlaif_trainer.py │ │ ├── test_rlvr_trainer.py │ │ ├── test_sft_trainer.py │ │ ├── test_tuner.py │ │ └── test_tuner_phase5.py └── tox.ini └── v3-examples ├── inference-examples ├── huggingface-example.ipynb ├── in-process-mode-example.ipynb ├── inference-spec-example.ipynb ├── jumpstart-e2e-training-example.ipynb ├── jumpstart-example.ipynb ├── local-mode-example.ipynb ├── optimize-example.ipynb └── train-inference-e2e-example.ipynb ├── ml-ops-examples ├── data │ ├── validation.csv │ └── xgb-churn-prediction-model.tar.gz ├── v3-hyperparameter-tuning-example │ ├── mnist.py │ ├── v3-hyperparameter-tuning-example.ipynb │ └── v3-hyperparameter-tuning-pipeline.ipynb ├── v3-model-registry-example │ ├── model │ │ └── model.tar.gz │ └── v3-model-registry-example.ipynb ├── v3-pipeline-train-create-registry.ipynb ├── v3-processing-job-pytorch │ ├── scripts │ │ └── preprocess │ │ │ ├── preprocessing.py │ │ │ └── requirements.txt │ └── v3-pytorch-processing-example.ipynb ├── v3-processing-job-sklearn.ipynb ├── v3-sagemaker-clarify.ipynb └── v3-transform-job-example.ipynb ├── model-customization-examples ├── ai_registry_example.ipynb ├── bedrock-modelbuilder-deployment.ipynb ├── benchmark_demo.ipynb ├── custom_scorer_demo.ipynb ├── dpo_trainer_example_notebook_v3_prod.ipynb ├── llm_as_judge_demo.ipynb ├── model_builder_deployment_notebook.ipynb ├── rlaif_finetuning_example_notebook_v3_prod.ipynb ├── rlvr_finetuning_example_notebook_v3_prod.ipynb └── sft_finetuning_example_notebook_pysdk_prod_v3.ipynb └── training-examples ├── custom-distributed-training-example.ipynb ├── distributed-local-training-example.ipynb ├── docker-compose.yaml ├── hyperparameter-training-example.ipynb ├── jumpstart-training-example.ipynb └── local-training-example.ipynb /.github/workflows/ci-health.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/.github/workflows/ci-health.yml -------------------------------------------------------------------------------- /.github/workflows/codeql.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/.github/workflows/codeql.yml -------------------------------------------------------------------------------- /.github/workflows/pr-checks-master-v2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/.github/workflows/pr-checks-master-v2.yml -------------------------------------------------------------------------------- /.github/workflows/pr-checks-master.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/.github/workflows/pr-checks-master.yml -------------------------------------------------------------------------------- /.github/workflows/security-monitoring.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/.github/workflows/security-monitoring.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/.pylintrc -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/LICENSE -------------------------------------------------------------------------------- /README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/README.rst -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 3.1.1 2 | -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements/extras/test_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/requirements/extras/test_requirements.txt -------------------------------------------------------------------------------- /requirements/tox/black_requirements.txt: -------------------------------------------------------------------------------- 1 | black==24.3.0 2 | -------------------------------------------------------------------------------- /requirements/tox/doc8_requirements.txt: -------------------------------------------------------------------------------- 1 | doc8==1.1.2 2 | Pygments==2.18.0 3 | -------------------------------------------------------------------------------- /requirements/tox/docstyle_requirements.txt: -------------------------------------------------------------------------------- 1 | pydocstyle==6.1.1 2 | -------------------------------------------------------------------------------- /requirements/tox/flake8_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/requirements/tox/flake8_requirements.txt -------------------------------------------------------------------------------- /requirements/tox/mypy_requirements.txt: -------------------------------------------------------------------------------- 1 | mypy==0.942 2 | -------------------------------------------------------------------------------- /requirements/tox/pydocstyle_requirements.txt: -------------------------------------------------------------------------------- 1 | pydocstyle==6.1.1 2 | -------------------------------------------------------------------------------- /requirements/tox/pylint_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/requirements/tox/pylint_requirements.txt -------------------------------------------------------------------------------- /requirements/tox/spelling_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/requirements/tox/spelling_requirements.txt -------------------------------------------------------------------------------- /requirements/tox/twine_requirements.txt: -------------------------------------------------------------------------------- 1 | build==1.2.1 2 | twine==5.0.0 3 | -------------------------------------------------------------------------------- /sagemaker-core/.env: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/.env -------------------------------------------------------------------------------- /sagemaker-core/.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/.pylintrc -------------------------------------------------------------------------------- /sagemaker-core/.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/.readthedocs.yaml -------------------------------------------------------------------------------- /sagemaker-core/CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/CHANGELOG.md -------------------------------------------------------------------------------- /sagemaker-core/CODEOWNERS: -------------------------------------------------------------------------------- 1 | * @aws/sagemaker-ml-frameworks -------------------------------------------------------------------------------- /sagemaker-core/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/CONTRIBUTING.md -------------------------------------------------------------------------------- /sagemaker-core/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/LICENSE -------------------------------------------------------------------------------- /sagemaker-core/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/MANIFEST.in -------------------------------------------------------------------------------- /sagemaker-core/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/README.rst -------------------------------------------------------------------------------- /sagemaker-core/VERSION: -------------------------------------------------------------------------------- 1 | 2.1.1 2 | 3 | -------------------------------------------------------------------------------- /sagemaker-core/branding/icon/sagemaker-banner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/branding/icon/sagemaker-banner.png -------------------------------------------------------------------------------- /sagemaker-core/docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/docs/Makefile -------------------------------------------------------------------------------- /sagemaker-core/docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/docs/conf.py -------------------------------------------------------------------------------- /sagemaker-core/docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/docs/index.rst -------------------------------------------------------------------------------- /sagemaker-core/docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/docs/make.bat -------------------------------------------------------------------------------- /sagemaker-core/docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/docs/requirements.txt -------------------------------------------------------------------------------- /sagemaker-core/example_notebooks/get_started.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/example_notebooks/get_started.ipynb -------------------------------------------------------------------------------- /sagemaker-core/example_notebooks/images/experiment_created.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/example_notebooks/images/experiment_created.png -------------------------------------------------------------------------------- /sagemaker-core/example_notebooks/sagemaker-core-llama-3-8B.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/example_notebooks/sagemaker-core-llama-3-8B.ipynb -------------------------------------------------------------------------------- /sagemaker-core/example_notebooks/sagemaker_core_overview.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/example_notebooks/sagemaker_core_overview.ipynb -------------------------------------------------------------------------------- /sagemaker-core/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/pyproject.toml -------------------------------------------------------------------------------- /sagemaker-core/resource_plan.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/resource_plan.csv -------------------------------------------------------------------------------- /sagemaker-core/sample/sagemaker-edge/2020-09-23/examples-1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/sample/sagemaker-edge/2020-09-23/examples-1.json -------------------------------------------------------------------------------- /sagemaker-core/sample/sagemaker-edge/2020-09-23/paginators-1.json: -------------------------------------------------------------------------------- 1 | { 2 | "pagination": {} 3 | } 4 | -------------------------------------------------------------------------------- /sagemaker-core/sample/sagemaker-edge/2020-09-23/service-2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/sample/sagemaker-edge/2020-09-23/service-2.json -------------------------------------------------------------------------------- /sagemaker-core/sample/sagemaker-featurestore-runtime/2020-07-01/paginators-1.json: -------------------------------------------------------------------------------- 1 | { 2 | "pagination": {} 3 | } 4 | -------------------------------------------------------------------------------- /sagemaker-core/sample/sagemaker-metrics/2022-09-30/paginators-1.json: -------------------------------------------------------------------------------- 1 | { 2 | "pagination": {} 3 | } 4 | -------------------------------------------------------------------------------- /sagemaker-core/sample/sagemaker-runtime/2017-05-13/paginators-1.json: -------------------------------------------------------------------------------- 1 | { 2 | "pagination": {} 3 | } 4 | -------------------------------------------------------------------------------- /sagemaker-core/sample/sagemaker/2017-07-24/default-configs.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/sample/sagemaker/2017-07-24/default-configs.json -------------------------------------------------------------------------------- /sagemaker-core/sample/sagemaker/2017-07-24/examples-1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/sample/sagemaker/2017-07-24/examples-1.json -------------------------------------------------------------------------------- /sagemaker-core/sample/sagemaker/2017-07-24/paginators-1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/sample/sagemaker/2017-07-24/paginators-1.json -------------------------------------------------------------------------------- /sagemaker-core/sample/sagemaker/2017-07-24/service-2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/sample/sagemaker/2017-07-24/service-2.json -------------------------------------------------------------------------------- /sagemaker-core/sample/sagemaker/2017-07-24/waiters-2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/sample/sagemaker/2017-07-24/waiters-2.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/_studio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/_studio.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/_version.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/accept_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/accept_types.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/analytics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/analytics.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/apiutils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/apiutils/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/apiutils/_base_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/apiutils/_base_types.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/apiutils/_boto_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/apiutils/_boto_functions.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/apiutils/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/apiutils/_utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/base_deserializers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/base_deserializers.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/base_serializers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/base_serializers.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/clarify/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/clarify/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/collection.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/common_utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/config/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/config/config.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/config/config_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/config/config_manager.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/config/config_schema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/config/config_schema.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/config/config_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/config/config_utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/config_schema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/config_schema.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/constants.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/content_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/content_types.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/debugger/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/debugger/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/debugger/debugger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/debugger/debugger.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/debugger/framework_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/debugger/framework_profile.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/debugger/metrics_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/debugger/metrics_config.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/debugger/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/debugger/profiler.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/debugger/profiler_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/debugger/profiler_config.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/debugger/profiler_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/debugger/profiler_constants.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/debugger/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/debugger/utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/deprecations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/deprecations.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/deserializers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/deserializers/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/deserializers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/deserializers/base.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/drift_check_baselines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/drift_check_baselines.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/enums.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/enums.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/environment_variables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/environment_variables.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/exceptions.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/experiments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/experiments/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/experiments/_api_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/experiments/_api_types.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/experiments/_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/experiments/_environment.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/experiments/_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/experiments/_helper.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/experiments/_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/experiments/_metrics.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/experiments/_run_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/experiments/_run_context.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/experiments/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/experiments/_utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/experiments/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/experiments/experiment.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/experiments/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/experiments/run.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/experiments/trial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/experiments/trial.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/experiments/trial_component.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/experiments/trial_component.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/explainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/explainer/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/explainer/explainer_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/explainer/explainer_config.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/fw_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/fw_utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/git_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/git_utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/helper/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/helper/pipeline_variable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/helper/pipeline_variable.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/helper/session_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/helper/session_helper.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/huggingface/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/huggingface/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/huggingface/llm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/huggingface/llm_utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/huggingface/processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/huggingface/processing.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/huggingface/training_compiler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/hyperparameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/hyperparameters.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_retriever/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_retriever/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_retriever/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_retriever/test.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/chainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/chainer.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/clarify.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/clarify.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/debugger.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/debugger.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/djl-lmi.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/djl-lmi.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/kmeans.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/kmeans.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/knn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/knn.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/lda.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/lda.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/mxnet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/mxnet.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/ntm.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/ntm.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/pca.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/pca.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/pytorch.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/pytorch.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/seq2seq.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/seq2seq.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/sklearn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/sklearn.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/spark.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/spark.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/vw.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/vw.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uri_config/xgboost.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uri_config/xgboost.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/image_uris.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/image_uris.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/inference_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/inference_config.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/inputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/inputs.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/instance_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/instance_group.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/instance_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/instance_types.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/instance_types_gpu_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/instance_types_gpu_info.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/interactive_apps/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/interactive_apps/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/iterators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/iterators.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/job.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/accessors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/accessors.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/artifacts/kwargs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/artifacts/kwargs.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/cache.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/configs.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/constants.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/deserializers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/deserializers.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/document.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/document.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/enums.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/enums.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/exceptions.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/factory/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/factory/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/factory/utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/filters.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/hub/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/hub/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/hub/constants.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/hub/hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/hub/hub.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/hub/interfaces.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/hub/interfaces.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/hub/parser_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/hub/parser_utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/hub/parsers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/hub/parsers.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/hub/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/hub/types.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/hub/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/hub/utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/models.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/notebook_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/notebook_utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/parameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/parameters.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/payload_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/payload_utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/region_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/region_config.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/search.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/serializers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/serializers.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/session_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/session_utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/types.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/jumpstart/validators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/jumpstart/validators.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/lambda_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/lambda_helper.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/lineage/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/lineage/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/lineage/_api_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/lineage/_api_types.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/lineage/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/lineage/_utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/lineage/action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/lineage/action.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/lineage/artifact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/lineage/artifact.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/lineage/association.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/lineage/association.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/lineage/context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/lineage/context.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/lineage/query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/lineage/query.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/lineage/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/lineage/visualizer.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/local/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/local/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/local/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/local/data.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/local/entities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/local/entities.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/local/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/local/exceptions.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/local/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/local/image.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/local/local_session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/local/local_session.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/local/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/local/utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/logs.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/metadata_properties.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/metadata_properties.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/metric_definitions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/metric_definitions.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/mlflow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/mlflow/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/model_card/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/model_card/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/model_life_cycle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/model_life_cycle.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/model_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/model_metrics.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/model_monitor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/model_monitor/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/model_monitor/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/model_monitor/utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/model_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/model_registry.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/model_uris.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/model_uris.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/modules/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/modules/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/modules/configs.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/modules/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/modules/constants.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/modules/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/modules/distributed.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/modules/local_core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/modules/templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/modules/templates.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/modules/train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/modules/train/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/modules/train/sm_recipes/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/modules/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/modules/types.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/modules/utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/network.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/parameter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/parameter.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/payloads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/payloads.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/processing.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/remote_function/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/remote_function/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/remote_function/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/remote_function/client.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/remote_function/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/remote_function/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/remote_function/errors.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/remote_function/job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/remote_function/job.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/resource_requirements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/resource_requirements.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/resources.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/s3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/s3/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/s3/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/s3/client.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/s3/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/s3/utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/script_uris.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/script_uris.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/serializers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/serializers/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/serializers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/serializers/base.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/serializers/implementations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/serializers/implementations.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/serializers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/serializers/utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/serverless_inference_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/serverless_inference_config.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/session_settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/session_settings.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/shapes/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/shapes/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/shapes/model_card_shapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/shapes/model_card_shapes.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/shapes/shapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/shapes/shapes.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/spark/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/spark/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/spark/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/spark/defaults.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/spark/processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/spark/processing.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/telemetry/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/telemetry/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/telemetry/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/telemetry/constants.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/telemetry/telemetry_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/telemetry/telemetry_logging.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/tools/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/tools/api_coverage.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/tools/api_coverage.json -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/tools/codegen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/tools/codegen.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/tools/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/tools/constants.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/tools/data_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/tools/data_extractor.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/tools/method.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/tools/method.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/tools/resource_plan.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/tools/resource_plan.csv -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/tools/resources_codegen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/tools/resources_codegen.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/tools/resources_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/tools/resources_extractor.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/tools/shapes_codegen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/tools/shapes_codegen.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/tools/shapes_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/tools/shapes_extractor.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/tools/templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/tools/templates.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/training/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/training/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/training/configs.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/training/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/training/constants.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/training/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/training/utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/training_compiler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/training_compiler/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/training_compiler/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/training_compiler/config.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/training_compiler_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/training_compiler_config.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/transformer.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/user_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/user_agent.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/utilities/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/utilities/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/utilities/cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/utilities/cache.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/utilities/search_expression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/utilities/search_expression.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/utils/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/utils/code_injection/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/utils/code_injection/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/utils/code_injection/base.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/utils/code_injection/codec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/utils/code_injection/codec.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/utils/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/utils/exceptions.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/utils/logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/utils/logs.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/utils/user_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/utils/user_agent.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/utils/utils.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/workflow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/workflow/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/workflow/conditions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/workflow/conditions.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/workflow/entities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/workflow/entities.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/workflow/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/workflow/functions.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/workflow/parameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/workflow/parameters.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/workflow/pipeline_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/workflow/pipeline_context.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/workflow/properties.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/workflow/properties.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/workflow/step_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/workflow/step_outputs.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/core/workflow/utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/core/workflow/utilities.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/lineage/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/lineage/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/lineage/action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/lineage/action.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/lineage/artifact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/lineage/artifact.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/lineage/context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/lineage/context.py -------------------------------------------------------------------------------- /sagemaker-core/src/sagemaker/lineage/lineage_trial_component.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/src/sagemaker/lineage/lineage_trial_component.py -------------------------------------------------------------------------------- /sagemaker-core/tests/data/config/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/data/config/config.yaml -------------------------------------------------------------------------------- /sagemaker-core/tests/data/config/config_file_with_code.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/data/config/config_file_with_code.yaml -------------------------------------------------------------------------------- /sagemaker-core/tests/data/config/invalid_config_file.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/data/config/invalid_config_file.yaml -------------------------------------------------------------------------------- /sagemaker-core/tests/data/config/sample_config_for_merge.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/data/config/sample_config_for_merge.yaml -------------------------------------------------------------------------------- /sagemaker-core/tests/data/cuteCat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/data/cuteCat.jpg -------------------------------------------------------------------------------- /sagemaker-core/tests/data/cuteCat.raw: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/data/cuteCat.raw -------------------------------------------------------------------------------- /sagemaker-core/tests/data/with_integers.csv: -------------------------------------------------------------------------------- 1 | 1,2,3 2 | 4,5,6 3 | 7,8,9 4 | -------------------------------------------------------------------------------- /sagemaker-core/tests/data/xgboost_abalone/abalone: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/data/xgboost_abalone/abalone -------------------------------------------------------------------------------- /sagemaker-core/tests/integ/jumpstart/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/tests/integ/jumpstart/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/integ/jumpstart/test_model.py -------------------------------------------------------------------------------- /sagemaker-core/tests/integ/jumpstart/test_search_integ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/integ/jumpstart/test_search_integ.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/config/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/config/conftest.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/config/test_config_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/config/test_config_manager.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/config/test_config_schema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/config/test_config_schema.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/conftest.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/deserializers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/generated/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/generated/resource_test_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/generated/resource_test_config.json -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/generated/test_config_schema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/generated/test_config_schema.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/generated/test_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/generated/test_logs.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/generated/test_model_card.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/generated/test_model_card.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/generated/test_resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/generated/test_resources.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/generated/test_shapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/generated/test_shapes.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/generated/test_user_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/generated/test_user_agent.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/generated/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/generated/test_utils.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/helper/test_session_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/helper/test_session_helper.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/interactive_apps/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/interactive_apps/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/interactive_apps/test_profiler_app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/interactive_apps/test_profiler_app.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/interactive_apps/test_tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/interactive_apps/test_tensorboard.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/jumpstart/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/jumpstart/hub/test_interfaces.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/jumpstart/hub/test_interfaces.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/jumpstart/hub/test_parsers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/jumpstart/hub/test_parsers.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/jumpstart/hub_content_document.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/jumpstart/hub_content_document.json -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/jumpstart/test_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/jumpstart/test_cache.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/jumpstart/test_document.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/jumpstart/test_document.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/jumpstart/test_factory_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/jumpstart/test_factory_utils.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/jumpstart/test_filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/jumpstart/test_filters.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/jumpstart/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/jumpstart/test_models.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/jumpstart/test_notebook_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/jumpstart/test_notebook_utils.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/jumpstart/test_search_unit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/jumpstart/test_search_unit.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/jumpstart/test_utils_extended.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/jumpstart/test_utils_extended.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/lineage/test_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/lineage/test_query.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/local/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/local/test_entities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/local/test_entities.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/local/test_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/local/test_image.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/local/test_local_session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/local/test_local_session.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/local/test_local_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/local/test_local_utils.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/model_monitor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/model_monitor/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/model_monitor/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/model_monitor/test_utils.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/modules/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/modules/local_core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/modules/local_core/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/modules/train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/modules/train/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/modules/train/container_drivers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/modules/train/container_drivers/distributed_drivers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/modules/train/test_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/modules/train/test_environment.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/modules/train/test_sm_recipes_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/modules/train/test_sm_recipes_utils.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/remote_function/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/remote_function/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/remote_function/test_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/remote_function/test_client.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/remote_function/test_job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/remote_function/test_job.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/s3/test_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/s3/test_client.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/serializers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/serializers/test_base_serializers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/serializers/test_base_serializers.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/serializers/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/serializers/test_utils.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/session/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/session/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/session/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/session/conftest.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/session/test_session_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/session/test_session_helper.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/session/test_session_identity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/session/test_session_identity.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/spark/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/spark/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/telemetry/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/telemetry/test_telemetry_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/telemetry/test_telemetry_logging.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_accept_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_accept_types.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_analytics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_analytics.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_base_deserializers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_base_deserializers.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_clarify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_clarify.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_codec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_codec.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_collection.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_common_utils.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_constants.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_content_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_content_types.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_deserializer_implementations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_deserializer_implementations.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_drift_check_baselines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_drift_check_baselines.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_fw_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_fw_utils.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_git_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_git_utils.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_hyperparameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_hyperparameters.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_image_retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_image_retriever.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_image_retriever_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_image_retriever_utils.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_image_uris.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_image_uris.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_inference_recommender_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_inference_recommender_mixin.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_instance_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_instance_group.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_iterators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_iterators.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_job.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_jumpstart_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_jumpstart_types.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_jumpstart_types_coverage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_jumpstart_types_coverage.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_jumpstart_types_extended.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_jumpstart_types_extended.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_jumpstart_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_jumpstart_utils.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_lambda_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_lambda_helper.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_metadata_properties.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_metadata_properties.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_metric_definitions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_metric_definitions.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_model_life_cycle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_model_life_cycle.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_model_monitoring.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_model_monitoring.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_model_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_model_registry.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_model_uris.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_model_uris.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_modules_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_modules_constants.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_payloads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_payloads.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_processing.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_profiler_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_profiler_constants.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_resource_requirements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_resource_requirements.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_script_uris.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_script_uris.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_serializer_implementations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_serializer_implementations.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_serverless_inference_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_serverless_inference_config.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_training_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_training_constants.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_transformer.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/test_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/test_version.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/tools/test_api_coverage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/tools/test_api_coverage.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/tools/test_resources_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/tools/test_resources_extractor.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/tools/test_shapes_codegen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/tools/test_shapes_codegen.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/tools/test_shapes_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/tools/test_shapes_extractor.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/workflow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/workflow/__init__.py -------------------------------------------------------------------------------- /sagemaker-core/tests/unit/workflow/test_utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tests/unit/workflow/test_utilities.py -------------------------------------------------------------------------------- /sagemaker-core/tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/tox.ini -------------------------------------------------------------------------------- /sagemaker-core/workflow_helper/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-core/workflow_helper/compute_boto_api_coverage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/workflow_helper/compute_boto_api_coverage.py -------------------------------------------------------------------------------- /sagemaker-core/workflow_helper/compute_resource_coverage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-core/workflow_helper/compute_resource_coverage.py -------------------------------------------------------------------------------- /sagemaker-mlops/CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/CHANGELOG.md -------------------------------------------------------------------------------- /sagemaker-mlops/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/README.md -------------------------------------------------------------------------------- /sagemaker-mlops/VERSION: -------------------------------------------------------------------------------- 1 | 1.1.1 2 | -------------------------------------------------------------------------------- /sagemaker-mlops/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/pyproject.toml -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/__init__.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/__init__.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/local/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/local/__init__.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/local/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/local/exceptions.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/local/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/local/pipeline.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/local/pipeline_entities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/local/pipeline_entities.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/__init__.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/_repack_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/_repack_model.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/_steps_compiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/_steps_compiler.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/_utils.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/automl_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/automl_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/callback_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/callback_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/check_job_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/check_job_config.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/condition_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/condition_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/emr_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/emr_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/fail_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/fail_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/function_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/function_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/lambda_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/lambda_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/model_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/model_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/pipeline.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/retry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/retry.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/step_collections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/step_collections.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/steps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/steps.py -------------------------------------------------------------------------------- /sagemaker-mlops/src/sagemaker/mlops/workflow/triggers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/src/sagemaker/mlops/workflow/triggers.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/__init__.py: -------------------------------------------------------------------------------- 1 | """Integration tests for SageMaker V3 pipeline examples.""" 2 | -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/code/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/code/mnist.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/code/pipeline/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/code/pipeline/preprocess.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/code/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/code/preprocess.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/code/pytorch_processing/requirements.txt: -------------------------------------------------------------------------------- 1 | datasets 2 | transformers 3 | torch 4 | -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/conftest.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/data/abalone.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/data/abalone.csv -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/data/pipeline/abalone.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/data/pipeline/abalone.csv -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/data/validation.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/data/validation.csv -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/test_clarify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/test_clarify.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/test_hyperparameter_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/test_hyperparameter_tuning.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/test_model_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/test_model_registry.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/test_pipeline_train_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/test_pipeline_train_registry.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/test_processing_job_sklearn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/test_processing_job_sklearn.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/test_pytorch_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/test_pytorch_processing.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/integ/test_transform_job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/integ/test_transform_job.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/local/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/local/__init__.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/local/test_exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/local/test_exceptions.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/local/test_local_pipeline_session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/local/test_local_pipeline_session.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/local/test_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/local/test_pipeline.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/local/test_pipeline_entities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/local/test_pipeline_entities.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/local/test_pipeline_executor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/local/test_pipeline_executor.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/__init__.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_callback_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_callback_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_check_job_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_check_job_config.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_clarify_check_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_clarify_check_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_condition_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_condition_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_emr_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_emr_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_fail_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_fail_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_function_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_function_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_lambda_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_lambda_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_model_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_model_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_notebook_job_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_notebook_job_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_parallelism_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_parallelism_config.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_pipeline.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_pipeline_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_pipeline_class.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_quality_check_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_quality_check_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_repack_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_repack_model.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_retry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_retry.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_step_collections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_step_collections.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_steps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_steps.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_steps_compiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_steps_compiler.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_triggers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_triggers.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_tuning_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_tuning_step.py -------------------------------------------------------------------------------- /sagemaker-mlops/tests/unit/workflow/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tests/unit/workflow/test_utils.py -------------------------------------------------------------------------------- /sagemaker-mlops/tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-mlops/tox.ini -------------------------------------------------------------------------------- /sagemaker-serve/CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/CHANGELOG.md -------------------------------------------------------------------------------- /sagemaker-serve/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/README.md -------------------------------------------------------------------------------- /sagemaker-serve/VERSION: -------------------------------------------------------------------------------- 1 | 1.1.1 2 | -------------------------------------------------------------------------------- /sagemaker-serve/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/pyproject.toml -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/__init__.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/async_inference/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/async_inference/__init__.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/batch_inference/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/bedrock_model_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/bedrock_model_builder.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/builder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/builder/schema_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/builder/schema_builder.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/builder/serve_settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/builder/serve_settings.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/configs.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/constants.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/deployment_progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/deployment_progress.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/detector/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/detector/image_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/detector/image_detector.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/detector/pickler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/detector/pickler.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/local_resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/local_resources.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/marshalling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/mode/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/mode/function_pointers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/mode/function_pointers.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/mode/in_process_mode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/mode/in_process_mode.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/mode/local_container_mode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/mode/local_container_mode.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/model_builder.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_builder_servers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/model_builder_servers.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_builder_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/model_builder_utils.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_format/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_format/mlflow/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_format/mlflow/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/model_format/mlflow/utils.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_server/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_server/djl_serving/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_server/in_process_model_server/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_server/multi_model_server/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_server/smd/prepare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/model_server/smd/prepare.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_server/smd/server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/model_server/smd/server.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_server/tei/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_server/tensorflow_serving/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_server/tgi/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_server/tgi/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/model_server/tgi/utils.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_server/torchserve/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/model_server/triton/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/predictor_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/predictor_async.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/schema/task.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/schema/task.json -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/serverless/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/serverless/__init__.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/serverless/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/serverless/model.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/spec/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/spec/inference_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/spec/inference_base.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/spec/inference_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/spec/inference_spec.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/utils/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/utils/exceptions.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/utils/hf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/utils/hf_utils.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/utils/lineage_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/utils/lineage_utils.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/utils/local_hardware.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/utils/local_hardware.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/utils/logging_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/utils/logging_agent.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/utils/packaging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/utils/packaging.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/utils/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/utils/task.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/utils/telemetry_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/utils/telemetry_logger.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/utils/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/utils/types.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/utils/uploader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/src/sagemaker/serve/utils/uploader.py -------------------------------------------------------------------------------- /sagemaker-serve/src/sagemaker/serve/validations/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/integ/test_huggingface_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/integ/test_huggingface_integration.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/integ/test_in_process_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/integ/test_in_process_integration.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/integ/test_jumpstart_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/integ/test_jumpstart_integration.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/integ/test_optimize_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/integ/test_optimize_integration.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/integ/test_tei_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/integ/test_tei_integration.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/integ/test_tgi_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/integ/test_tgi_integration.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/integ/test_triton_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/integ/test_triton_integration.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/__init__.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/async_inference/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/batch_inference/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/builder/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/builder/test_schema_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/builder/test_schema_builder.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/builder/test_serve_settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/builder/test_serve_settings.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/compute_resource_requirements/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/detector/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/detector/test_image_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/detector/test_image_detector.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/detector/test_pickler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/detector/test_pickler.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/marshalling/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/mb_user_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/mb_user_test.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/mode/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/mode/test_function_pointers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/mode/test_function_pointers.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/model_format/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/model_format/test_mlflow_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/model_format/test_mlflow_utils.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/model_server/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/model_server/test_djl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/model_server/test_djl_utils.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/model_server/test_smd_prepare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/model_server/test_smd_prepare.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/model_server/test_smd_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/model_server/test_smd_server.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/model_server/test_tei_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/model_server/test_tei_server.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/model_server/test_tgi_prepare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/model_server/test_tgi_prepare.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/model_server/test_tgi_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/model_server/test_tgi_server.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/model_server/test_tgi_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/model_server/test_tgi_utils.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/run_all_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/run_all_tests.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/serverless/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/spec/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/spec/test_inference_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/spec/test_inference_base.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/spec/test_inference_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/spec/test_inference_spec.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_bedrock_model_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_bedrock_model_builder.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_configs.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_constants.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_deployment_progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_deployment_progress.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_fixtures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_fixtures.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_local_resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_local_resources.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_additional.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_advanced.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder_advanced.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder_build.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder_core.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_deploy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder_deploy.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder_integration.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_methods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder_methods.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_servers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder_servers.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder_utils.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_utils_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder_utils_new.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_utils_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder_utils_triton.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder_v3.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_v3_extended.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_model_builder_workflows.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_model_builder_workflows.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_parse_registry_accounts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_parse_registry_accounts.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_predictor_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_predictor_async.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/test_telemetry_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/test_telemetry_logger.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/utils/test_exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/utils/test_exceptions.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/utils/test_hardware_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/utils/test_hardware_detector.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/utils/test_hf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/utils/test_hf_utils.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/utils/test_lineage_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/utils/test_lineage_constants.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/utils/test_lineage_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/utils/test_lineage_utils.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/utils/test_local_hardware.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/utils/test_local_hardware.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/utils/test_logging_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/utils/test_logging_agent.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/utils/test_packaging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/utils/test_packaging.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/utils/test_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/utils/test_task.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/utils/test_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/utils/test_types.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/utils/test_uploader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/utils/test_uploader.py -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/validations/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /sagemaker-serve/tests/unit/validations/test_optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tests/unit/validations/test_optimization.py -------------------------------------------------------------------------------- /sagemaker-serve/tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-serve/tox.ini -------------------------------------------------------------------------------- /sagemaker-train/CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/CHANGELOG.md -------------------------------------------------------------------------------- /sagemaker-train/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/CONTRIBUTING.md -------------------------------------------------------------------------------- /sagemaker-train/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/LICENSE -------------------------------------------------------------------------------- /sagemaker-train/MANIFEST.in: -------------------------------------------------------------------------------- 1 | recursive-include src *.json 2 | -------------------------------------------------------------------------------- /sagemaker-train/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/README.rst -------------------------------------------------------------------------------- /sagemaker-train/VERSION: -------------------------------------------------------------------------------- 1 | 1.1.1 2 | -------------------------------------------------------------------------------- /sagemaker-train/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/pyproject.toml -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/ai_registry/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/ai_registry/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/ai_registry/air_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/ai_registry/air_constants.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/ai_registry/air_hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/ai_registry/air_hub.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/ai_registry/air_hub_entity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/ai_registry/air_hub_entity.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/ai_registry/air_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/ai_registry/air_utils.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/ai_registry/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/ai_registry/dataset.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/ai_registry/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/ai_registry/dataset_utils.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/ai_registry/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/ai_registry/evaluator.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/ai_registry/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/ai_registry/utils.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/base_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/common.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/common_utils/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/common_utils/constants.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/configs.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/constants.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/defaults.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/distributed.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/dpo_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/dpo_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/evaluate/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/evaluate/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/evaluate/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/evaluate/constants.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/evaluate/execution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/evaluate/execution.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/local/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/local/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/local/data.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/local/entities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/local/entities.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/local/local_container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/local/local_container.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/model_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/model_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/modules/model_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/modules/model_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/remote_function/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/remote_function/client.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/remote_function/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/remote_function/errors.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/remote_function/job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/remote_function/job.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/rlaif_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/rlaif_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/rlvr_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/rlvr_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/sft_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/sm_recipes/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/sm_recipes/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/sm_recipes/utils.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/templates.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/tuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/tuner.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/types.py -------------------------------------------------------------------------------- /sagemaker-train/src/sagemaker/train/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/src/sagemaker/train/utils.py -------------------------------------------------------------------------------- /sagemaker-train/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/_repack_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/_repack_model.py -------------------------------------------------------------------------------- /sagemaker-train/tests/data/custom_drivers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/custom_drivers/driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/custom_drivers/driver.py -------------------------------------------------------------------------------- /sagemaker-train/tests/data/local_script/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/local_script/data/test/x_test.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/local_script/data/test/x_test.npy -------------------------------------------------------------------------------- /sagemaker-train/tests/data/local_script/data/test/y_test.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/local_script/data/test/y_test.npy -------------------------------------------------------------------------------- /sagemaker-train/tests/data/local_script/pytorch_model_def.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/local_script/pytorch_model_def.py -------------------------------------------------------------------------------- /sagemaker-train/tests/data/params_script/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/params_script/hyperparameters.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/params_script/hyperparameters.json -------------------------------------------------------------------------------- /sagemaker-train/tests/data/params_script/hyperparameters.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/params_script/hyperparameters.yaml -------------------------------------------------------------------------------- /sagemaker-train/tests/data/params_script/requirements.txt: -------------------------------------------------------------------------------- 1 | omegaconf 2 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/params_script/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/params_script/train.py -------------------------------------------------------------------------------- /sagemaker-train/tests/data/params_script/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/params_script/train.sh -------------------------------------------------------------------------------- /sagemaker-train/tests/data/remote_function/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/remote_function/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/remote_function/config.yaml -------------------------------------------------------------------------------- /sagemaker-train/tests/data/remote_function/non_existent_requirements.txt: -------------------------------------------------------------------------------- 1 | does_not_exist 2 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/remote_function/old_deps_requirements.txt: -------------------------------------------------------------------------------- 1 | pandas==1.3.4 2 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/remote_function/pre_exec_commands: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/remote_function/pre_exec_commands -------------------------------------------------------------------------------- /sagemaker-train/tests/data/remote_function/requirements.txt: -------------------------------------------------------------------------------- 1 | scipy==1.10.1 2 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/remote_function/workdir/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/remote_function/workdir/data/data.csv: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/remote_function/workdir/module/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/remote_function/workdir/module/train.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/remote_function/workdir/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/remote_function/workdir/test/test.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/script_mode/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/script_mode/code.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/script_mode/code.tar.gz -------------------------------------------------------------------------------- /sagemaker-train/tests/data/script_mode/custom_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/script_mode/custom_script.py -------------------------------------------------------------------------------- /sagemaker-train/tests/data/script_mode/data/test/x_test.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/script_mode/data/test/x_test.npy -------------------------------------------------------------------------------- /sagemaker-train/tests/data/script_mode/data/test/y_test.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/script_mode/data/test/y_test.npy -------------------------------------------------------------------------------- /sagemaker-train/tests/data/script_mode/data/train/x_train.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/script_mode/data/train/x_train.npy -------------------------------------------------------------------------------- /sagemaker-train/tests/data/script_mode/data/train/y_train.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/script_mode/data/train/y_train.npy -------------------------------------------------------------------------------- /sagemaker-train/tests/data/script_mode/pytorch_model_def.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/script_mode/pytorch_model_def.py -------------------------------------------------------------------------------- /sagemaker-train/tests/data/script_mode/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/script_mode/requirements.txt -------------------------------------------------------------------------------- /sagemaker-train/tests/data/scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/data/scripts/entry_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/data/scripts/entry_script.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/ai_registry/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/ai_registry/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/ai_registry/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/ai_registry/conftest.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/ai_registry/test_air_hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/ai_registry/test_air_hub.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/ai_registry/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/ai_registry/test_dataset.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/ai_registry/test_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/ai_registry/test_evaluator.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/jumpstart/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/jumpstart/test_jumpstart_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/jumpstart/test_jumpstart_train.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/lock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/lock.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/train/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/train/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/train/conftest.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/train/test_benchmark_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/train/test_benchmark_evaluator.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/train/test_local_model_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/train/test_local_model_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/tests/integ/train/test_model_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/integ/train/test_model_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/ai_registry/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/ai_registry/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/ai_registry/test_air_hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/ai_registry/test_air_hub.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/ai_registry/test_air_hub_entity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/ai_registry/test_air_hub_entity.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/ai_registry/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/ai_registry/test_dataset.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/ai_registry/test_dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/ai_registry/test_dataset_utils.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/ai_registry/test_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/ai_registry/test_evaluator.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/common_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/common_utils/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/container_drivers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/evaluate/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/evaluate/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/evaluate/test_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/evaluate/test_constants.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/evaluate/test_execution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/evaluate/test_execution.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/evaluate/test_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/evaluate/test_init.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/local/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/local/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/local/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/local/test_data.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/local/test_entities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/local/test_entities.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/remote_function/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/remote_function/__init__.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/sm_recipes/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/sm_recipes/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/sm_recipes/test_utils.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/test_defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/test_defaults.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/test_dpo_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/test_dpo_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/test_model_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/test_model_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/test_rlaif_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/test_rlaif_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/test_rlvr_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/test_rlvr_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/test_sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/test_sft_trainer.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/test_tuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/test_tuner.py -------------------------------------------------------------------------------- /sagemaker-train/tests/unit/train/test_tuner_phase5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tests/unit/train/test_tuner_phase5.py -------------------------------------------------------------------------------- /sagemaker-train/tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/sagemaker-train/tox.ini -------------------------------------------------------------------------------- /v3-examples/inference-examples/huggingface-example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/inference-examples/huggingface-example.ipynb -------------------------------------------------------------------------------- /v3-examples/inference-examples/in-process-mode-example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/inference-examples/in-process-mode-example.ipynb -------------------------------------------------------------------------------- /v3-examples/inference-examples/inference-spec-example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/inference-examples/inference-spec-example.ipynb -------------------------------------------------------------------------------- /v3-examples/inference-examples/jumpstart-example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/inference-examples/jumpstart-example.ipynb -------------------------------------------------------------------------------- /v3-examples/inference-examples/local-mode-example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/inference-examples/local-mode-example.ipynb -------------------------------------------------------------------------------- /v3-examples/inference-examples/optimize-example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/inference-examples/optimize-example.ipynb -------------------------------------------------------------------------------- /v3-examples/ml-ops-examples/data/validation.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/ml-ops-examples/data/validation.csv -------------------------------------------------------------------------------- /v3-examples/ml-ops-examples/v3-processing-job-pytorch/scripts/preprocess/requirements.txt: -------------------------------------------------------------------------------- 1 | datasets 2 | transformers 3 | torch 4 | -------------------------------------------------------------------------------- /v3-examples/ml-ops-examples/v3-processing-job-sklearn.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/ml-ops-examples/v3-processing-job-sklearn.ipynb -------------------------------------------------------------------------------- /v3-examples/ml-ops-examples/v3-sagemaker-clarify.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/ml-ops-examples/v3-sagemaker-clarify.ipynb -------------------------------------------------------------------------------- /v3-examples/ml-ops-examples/v3-transform-job-example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/ml-ops-examples/v3-transform-job-example.ipynb -------------------------------------------------------------------------------- /v3-examples/model-customization-examples/benchmark_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/model-customization-examples/benchmark_demo.ipynb -------------------------------------------------------------------------------- /v3-examples/training-examples/docker-compose.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/training-examples/docker-compose.yaml -------------------------------------------------------------------------------- /v3-examples/training-examples/local-training-example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-python-sdk/HEAD/v3-examples/training-examples/local-training-example.ipynb --------------------------------------------------------------------------------