Path	Lines of Code
tests/__init__.py	1
tests/component/__init__.py	1
tests/component/test_mxnet_estimator.py	63
tests/component/test_tf_estimator.py	80
tests/conftest.py	601
tests/data/_repack_model.py	51
tests/data/chainer_mnist/distributed_mnist.py	128
tests/data/chainer_mnist/mnist.py	118
tests/data/coach_cartpole/mxnet_deploy.py	29
tests/data/coach_cartpole/preset_cartpole_clippedppo.py	65
tests/data/coach_cartpole/train_coach.py	16
tests/data/config/config.yaml	183
tests/data/config/config_file_with_code.yaml	3
tests/data/config/expected_output_config_after_merge.yaml	10
tests/data/config/invalid_config_file.yaml	10
tests/data/config/sample_additional_config_for_merge.yaml	7
tests/data/config/sample_config_for_merge.yaml	9
tests/data/dummy_code_bundle_with_reqs/local_module.py	1
tests/data/dummy_code_bundle_with_reqs/main_script.py	6
tests/data/dummy_script.py	2
tests/data/experiment/inference.py	62
tests/data/experiment/process_job_script_for_run_clz.py	33
tests/data/experiment/train_job_script_for_run_clz.py	67
tests/data/horovod/hvd_basic.py	9
tests/data/horovod/hvd_mnist_mxnet.py	9
tests/data/horovod/launcher.sh	1
tests/data/huggingface/run_glue.py	419
tests/data/huggingface/run_tf.py	78
tests/data/huggingface_byoc/run_glue.py	419
tests/data/inference_recommender/inference.py	29
tests/data/iris/failure_script.py	3
tests/data/iris/iris-dnn-classifier.py	36
tests/data/marketplace/iris/scoring_logic.py	76
tests/data/marketplace/iris/wsgi.py	3
tests/data/modules/custom_drivers/driver.py	25
tests/data/modules/local_script/local_training_script.py	87
tests/data/modules/local_script/pytorch_model_def.py	16
tests/data/modules/params_script/hyperparameters.yaml	19
tests/data/modules/params_script/train.py	193
tests/data/modules/params_script/train.sh	6
tests/data/modules/script_mode/custom_script.py	86
tests/data/modules/script_mode/pytorch_model_def.py	16
tests/data/modules/scripts/entry_script.py	14
tests/data/monitor/actuals.jsonl	44
tests/data/monitor/capture.jsonl	44
tests/data/monitor/captured-data.jsonl	2
tests/data/monitor/postprocessor.py	2
tests/data/monitor/preprocessor.py	16
tests/data/multimodel/container/dockerd-entrypoint.py	19
tests/data/multimodel/container/model_handler.py	20
tests/data/mxnet_mnist/check_env.py	4
tests/data/mxnet_mnist/code/evaluation.py	9
tests/data/mxnet_mnist/code/inference.py	36
tests/data/mxnet_mnist/failure_script.py	2
tests/data/mxnet_mnist/mnist.py	106
tests/data/mxnet_mnist/mnist_ei.py	33
tests/data/mxnet_mnist/mnist_gluon.py	96
tests/data/mxnet_mnist/mnist_hosting_with_custom_handlers.py	69
tests/data/mxnet_mnist/mnist_neo.py	120
tests/data/mxnet_mnist/my_custom_rule.py	10
tests/data/object2vec/train.jsonl	1000
tests/data/pipeline/model_step/pytorch_mnist/mnist.py	194
tests/data/pipeline/test_source_dir/script_1.py	6
tests/data/pipeline/test_source_dir/script_2.py	4
tests/data/pipeline/test_source_dir_2/script_2.py	4
tests/data/pytorch_ddp/mnist_pt.py	209
tests/data/pytorch_eia/empty_inference_script.py	1
tests/data/pytorch_mnist/mnist.py	194
tests/data/pytorch_neo/code/inference.py	57
tests/data/pytorch_source_dirs/train.py	11
tests/data/ray_cartpole/train_ray.py	15
tests/data/remote_function/config.yaml	22
tests/data/remote_function/workdir/module/train.py	1
tests/data/remote_function/workdir/test/test.py	1
tests/data/sagemaker_rl/__init__.py	1
tests/data/sagemaker_rl/coach_launcher.py	237
tests/data/sagemaker_rl/configuration_list.py	59
tests/data/serve_resources/config.yaml	11
tests/data/serve_resources/mlflow/pytorch/conda.yaml	24
tests/data/serve_resources/mlflow/pytorch/python_env.yaml	7
tests/data/serve_resources/mlflow/tensorflow/conda.yaml	11
tests/data/serve_resources/mlflow/tensorflow/python_env.yaml	7
tests/data/serve_resources/mlflow/xgboost/conda.yaml	15
tests/data/serve_resources/mlflow/xgboost/python_env.yaml	7
tests/data/sip/preprocessor.py	80
tests/data/sip/sklearn_source_dir/inference.py	56
tests/data/sip/training.py	57
tests/data/sip/xgboost_source_dir/inference.py	40
tests/data/sklearn_mnist/__init__.py	1
tests/data/sklearn_mnist/failure_script.py	2
tests/data/sklearn_mnist/mnist.py	48
tests/data/sklearn_processing/preprocessing.py	56
tests/data/sklearn_processing/train.py	15
tests/data/smdistributed_dataparallel/mnist_pt.py	209
tests/data/smdistributed_dataparallel/mnist_tf.py	51
tests/data/spark/code/java/hello-java-spark/com/amazonaws/sagemaker/spark/test/HelloJavaSparkApp.java	6
tests/data/spark/code/python/hello_py_spark/__init__.py	1
tests/data/spark/code/python/hello_py_spark/hello_py_spark_app.py	27
tests/data/spark/code/python/hello_py_spark/hello_py_spark_udfs.py	2
tests/data/spark/files/data.jsonl	1000
tests/data/tensorflow_mnist/mnist.py	37
tests/data/tensorflow_mnist/mnist_mwms.py	42
tests/data/tensorflow_mnist/mnist_v1.py	82
tests/data/tensorflow_mnist/mnist_v2.py	140
tests/data/tfs/tfs-test-entrypoint-and-dependencies/dependency.py	1
tests/data/tfs/tfs-test-entrypoint-and-dependencies/inference.py	11
tests/data/tfs/tfs-test-entrypoint-with-handler/inference.py	10
tests/data/tfs/tfs-test-entrypoint-with-handler/training.py	5
tests/data/tfs/tfs-test-model-with-inference/code/inference.py	10
tests/data/torch_distributed/mnist_mlp_trainium.py	64
tests/data/upload_data_tests/file1.py	1
tests/data/upload_data_tests/file2.py	1
tests/data/upload_data_tests/nested_dir/file3.py	1
tests/data/upload_data_tests/nested_dir/file4.py	1
tests/data/workflow/abalone/evaluation.py	32
tests/data/workflow/abalone/preprocessing.py	73
tests/data/workflow/config.yaml	10
tests/data/workflow/emr-script.sh	2
tests/data/workflow/inference.py	1
tests/data/workflow/notebook_job_step/config.yaml	14
tests/data/workflow/notebook_job_step/my-init.sh	2
tests/data/workflow/notebook_job_step/notebook1_happypath.ipynb	729
tests/data/workflow/notebook_job_step/simple.ipynb	615
tests/data/workflow/notebook_job_step/step2_notebook.ipynb	637
tests/data/workflow/notebook_job_step/subfolder/sub.ipynb	77
tests/data/workflow/workdir/module/train.py	1
tests/data/workflow/workdir/test/test.py	1
tests/data/xgboost_abalone/abalone.py	34
tests/data/xgboost_abalone/inference.py	23
tests/integ/__init__.py	194
tests/integ/auto_ml_utils.py	23
tests/integ/auto_ml_v2_utils.py	109
tests/integ/datasets.py	10
tests/integ/file_system_input_utils.py	253
tests/integ/kms_utils.py	116
tests/integ/lock.py	17
tests/integ/marketplace_utils.py	23
tests/integ/record_set.py	12
tests/integ/retry.py	12
tests/integ/s3_utils.py	37
tests/integ/sagemaker/__init__.py	1
tests/integ/sagemaker/conftest.py	241
tests/integ/sagemaker/experiments/__init__.py	1
tests/integ/sagemaker/experiments/conftest.py	138
tests/integ/sagemaker/experiments/helpers.py	33
tests/integ/sagemaker/experiments/test_experiment.py	33
tests/integ/sagemaker/experiments/test_metrics.py	20
tests/integ/sagemaker/experiments/test_run.py	698
tests/integ/sagemaker/experiments/test_trial.py	47
tests/integ/sagemaker/experiments/test_trial_component.py	107
tests/integ/sagemaker/feature_store/feature_processor/test_feature_processor.py	1111
tests/integ/sagemaker/jumpstart/__init__.py	1
tests/integ/sagemaker/jumpstart/conftest.py	112
tests/integ/sagemaker/jumpstart/constants.py	79
tests/integ/sagemaker/jumpstart/estimator/__init__.py	1
tests/integ/sagemaker/jumpstart/estimator/test_jumpstart_estimator.py	176
tests/integ/sagemaker/jumpstart/model/__init__.py	1
tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py	388
tests/integ/sagemaker/jumpstart/private_hub/__init__.py	1
tests/integ/sagemaker/jumpstart/private_hub/estimator/__init__.py	1
tests/integ/sagemaker/jumpstart/private_hub/estimator/test_jumpstart_private_hub_estimator.py	139
tests/integ/sagemaker/jumpstart/private_hub/model/__init__.py	1
tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py	126
tests/integ/sagemaker/jumpstart/private_hub/test_hub.py	29
tests/integ/sagemaker/jumpstart/private_hub/test_hub_content.py	27
tests/integ/sagemaker/jumpstart/retrieve_uri/__init__.py	1
tests/integ/sagemaker/jumpstart/retrieve_uri/inference.py	155
tests/integ/sagemaker/jumpstart/retrieve_uri/test_inference.py	53
tests/integ/sagemaker/jumpstart/retrieve_uri/test_transfer_learning.py	88
tests/integ/sagemaker/jumpstart/retrieve_uri/training.py	124
tests/integ/sagemaker/jumpstart/retrieve_uri/utils.py	21
tests/integ/sagemaker/jumpstart/script_mode_class/__init__.py	1
tests/integ/sagemaker/jumpstart/script_mode_class/test_inference.py	92
tests/integ/sagemaker/jumpstart/script_mode_class/test_transfer_learning.py	146
tests/integ/sagemaker/jumpstart/utils.py	132
tests/integ/sagemaker/lineage/__init__.py	1
tests/integ/sagemaker/lineage/conftest.py	743
tests/integ/sagemaker/lineage/helpers.py	116
tests/integ/sagemaker/lineage/test_action.py	122
tests/integ/sagemaker/lineage/test_artifact.py	118
tests/integ/sagemaker/lineage/test_association.py	60
tests/integ/sagemaker/lineage/test_context.py	81
tests/integ/sagemaker/lineage/test_dataset_artifact.py	34
tests/integ/sagemaker/lineage/test_endpoint_context.py	78
tests/integ/sagemaker/lineage/test_image_artifact.py	12
tests/integ/sagemaker/lineage/test_lineage_trial_component.py	18
tests/integ/sagemaker/lineage/test_lineage_visualize.py	197
tests/integ/sagemaker/lineage/test_model_artifact.py	41
tests/integ/sagemaker/lineage/test_model_package_group_context.py	6
tests/integ/sagemaker/modules/__init__.py	1
tests/integ/sagemaker/modules/conftest.py	19
tests/integ/sagemaker/modules/train/test_local_model_trainer.py	168
tests/integ/sagemaker/modules/train/test_model_trainer.py	132
tests/integ/sagemaker/remote_function/__init__.py	1
tests/integ/sagemaker/remote_function/helpers/__init__.py	1
tests/integ/sagemaker/remote_function/helpers/local_module.py	3
tests/integ/sagemaker/remote_function/helpers/nested_helper/local_module2.py	3
tests/integ/sagemaker/remote_function/test_auto_capture.py	20
tests/integ/sagemaker/remote_function/test_decorator.py	669
tests/integ/sagemaker/remote_function/test_executor.py	287
tests/integ/sagemaker/serve/__init__.py	1
tests/integ/sagemaker/serve/conftest.py	33
tests/integ/sagemaker/serve/constants.py	22
tests/integ/sagemaker/serve/test_base_model_builder_deploy.py	174
tests/integ/sagemaker/serve/test_schema_builder.py	202
tests/integ/sagemaker/serve/test_serve_byoc_happy.py	1
tests/integ/sagemaker/serve/test_serve_hf_happy.py	1
tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py	225
tests/integ/sagemaker/serve/test_serve_js_happy.py	154
tests/integ/sagemaker/serve/test_serve_mlflow_pytorch_flavor_happy.py	148
tests/integ/sagemaker/serve/test_serve_mlflow_tensorflow_flavor_happy.py	131
tests/integ/sagemaker/serve/test_serve_mlflow_xgboost_flavor_happy.py	132
tests/integ/sagemaker/serve/test_serve_model_builder_gpu.py	162
tests/integ/sagemaker/serve/test_serve_model_builder_handshake.py	165
tests/integ/sagemaker/serve/test_serve_model_builder_inference_component_happy.py	113
tests/integ/sagemaker/serve/test_serve_pt_happy.py	138
tests/integ/sagemaker/serve/test_serve_pt_timeout.py	1
tests/integ/sagemaker/serve/test_serve_pt_triton_happy.py	1
tests/integ/sagemaker/serve/test_serve_tei.py	62
tests/integ/sagemaker/serve/test_serve_tf_triton_happy.py	1
tests/integ/sagemaker/serve/test_serve_transformers.py	105
tests/integ/sagemaker/serve/test_serve_xgb_happy.py	1
tests/integ/sagemaker/serve/utils/test_hardware_detector.py	19
tests/integ/sagemaker/utilities/__init__.py	1
tests/integ/sagemaker/utilities/test_search_expression.py	44
tests/integ/sagemaker/workflow/__init__.py	1
tests/integ/sagemaker/workflow/conftest.py	36
tests/integ/sagemaker/workflow/helpers.py	82
tests/integ/sagemaker/workflow/test_automl_steps.py	155
tests/integ/sagemaker/workflow/test_callback_steps.py	85
tests/integ/sagemaker/workflow/test_clarify_check_steps.py	242
tests/integ/sagemaker/workflow/test_emr_steps.py	101
tests/integ/sagemaker/workflow/test_experiment.py	149
tests/integ/sagemaker/workflow/test_fail_steps.py	259
tests/integ/sagemaker/workflow/test_lambda_steps.py	99
tests/integ/sagemaker/workflow/test_model_create_and_registration.py	1324
tests/integ/sagemaker/workflow/test_model_steps.py	817
tests/integ/sagemaker/workflow/test_monitor_batch_transform_step.py	850
tests/integ/sagemaker/workflow/test_notebook_job_step.py	465
tests/integ/sagemaker/workflow/test_pipeline_var_behaviors.py	82
tests/integ/sagemaker/workflow/test_processing_steps.py	853
tests/integ/sagemaker/workflow/test_quality_check_steps.py	390
tests/integ/sagemaker/workflow/test_retry.py	232
tests/integ/sagemaker/workflow/test_selective_execution.py	234
tests/integ/sagemaker/workflow/test_step_decorator.py	809
tests/integ/sagemaker/workflow/test_training_steps.py	296
tests/integ/sagemaker/workflow/test_triggers.py	132
tests/integ/sagemaker/workflow/test_tuning_steps.py	276
tests/integ/sagemaker/workflow/test_workflow.py	1178
tests/integ/sagemaker/workflow/test_workflow_with_clarify.py	223
tests/integ/sagemaker/workflow/workdir_helpers/__init__.py	1
tests/integ/sagemaker/workflow/workdir_helpers/local_module.py	3
tests/integ/sagemaker/workflow/workdir_helpers/nested_helper/local_module2.py	3
tests/integ/test_airflow_config.py	535
tests/integ/test_artifact_analytics.py	60
tests/integ/test_async_inference.py	93
tests/integ/test_auto_ml.py	323
tests/integ/test_auto_ml_v2.py	461
tests/integ/test_byo_estimator.py	149
tests/integ/test_chainer.py	134
tests/integ/test_clarify.py	705
tests/integ/test_clarify_model_monitor.py	571
tests/integ/test_collection.py	171
tests/integ/test_data_capture_config.py	216
tests/integ/test_data_upload.py	23
tests/integ/test_debugger.py	678
tests/integ/test_edge.py	52
tests/integ/test_experiments_analytics.py	213
tests/integ/test_explainer.py	99
tests/integ/test_factorization_machines.py	106
tests/integ/test_feature_store.py	2132
tests/integ/test_git.py	227
tests/integ/test_horovod.py	76
tests/integ/test_horovod_mx.py	71
tests/integ/test_huggingface.py	202
tests/integ/test_huggingface_torch_distributed.py	35
tests/integ/test_inference_component_based_endpoint.py	182
tests/integ/test_inference_pipeline.py	207
tests/integ/test_inference_recommender.py	444
tests/integ/test_ipinsights.py	71
tests/integ/test_kmeans.py	153
tests/integ/test_kmeans_efs_fsx.py	196
tests/integ/test_knn.py	91
tests/integ/test_lda.py	76
tests/integ/test_linear_learner.py	213
tests/integ/test_local_mode.py	800
tests/integ/test_marketplace.py	338
tests/integ/test_model_card.py	363
tests/integ/test_model_monitor.py	2140
tests/integ/test_model_package.py	487
tests/integ/test_model_quality_monitor.py	415
tests/integ/test_monitoring_files.py	310
tests/integ/test_multi_variant_endpoint.py	235
tests/integ/test_multidatamodel.py	381
tests/integ/test_mxnet.py	390
tests/integ/test_neo_mxnet.py	140
tests/integ/test_neo_pytorch.py	67
tests/integ/test_ntm.py	68
tests/integ/test_object2vec.py	86
tests/integ/test_pca.py	99
tests/integ/test_predict_stream.py	75
tests/integ/test_predict_with_inference_id.py	58
tests/integ/test_processing.py	713
tests/integ/test_profiler.py	362
tests/integ/test_pytorch.py	258
tests/integ/test_pytorchddp.py	36
tests/integ/test_randomcutforest.py	61
tests/integ/test_record_set.py	17
tests/integ/test_rl.py	95
tests/integ/test_s3.py	200
tests/integ/test_sagemaker_config.py	358
tests/integ/test_serve.py	129
tests/integ/test_serverless_inference.py	44
tests/integ/test_session.py	84
tests/integ/test_sklearn.py	245
tests/integ/test_smdataparallel_pt.py	40
tests/integ/test_smdataparallel_tf.py	39
tests/integ/test_source_dirs.py	31
tests/integ/test_spark_processing.py	295
tests/integ/test_sparkml_serving.py	43
tests/integ/test_tf.py	379
tests/integ/test_tf_efs_fsx.py	202
tests/integ/test_tfs.py	182
tests/integ/test_torch_distributed.py	32
tests/integ/test_training_compiler.py	220
tests/integ/test_transformer.py	626
tests/integ/test_tuner.py	823
tests/integ/test_tuner_multi_algo.py	193
tests/integ/test_xgboost.py	137
tests/integ/timeout.py	169
tests/integ/utils.py	66
tests/integ/vpc_test_utils.py	154
tests/scripts/fs_mount_setup.sh	72
tests/unit/__init__.py	444
tests/unit/common.py	16
tests/unit/conftest.py	34
tests/unit/output_capturer.py	13
tests/unit/sagemaker/__init__.py	1
tests/unit/sagemaker/accept_types/__init__.py	1
tests/unit/sagemaker/accept_types/jumpstart/__init__.py	1
tests/unit/sagemaker/accept_types/jumpstart/test_accept_types.py	71
tests/unit/sagemaker/async_inference/__init__.py	1
tests/unit/sagemaker/async_inference/test_async_inference_config.py	73
tests/unit/sagemaker/async_inference/test_async_inference_response.py	145
tests/unit/sagemaker/async_inference/test_waiter_config.py	23
tests/unit/sagemaker/automl/test_auto_ml.py	916
tests/unit/sagemaker/automl/test_auto_ml_v2.py	1012
tests/unit/sagemaker/automl/test_config.py	213
tests/unit/sagemaker/cli/__init__.py	1
tests/unit/sagemaker/cli/compatibility/__init__.py	1
tests/unit/sagemaker/cli/compatibility/v2/__init__.py	1
tests/unit/sagemaker/cli/compatibility/v2/modifiers/__init__.py	1
tests/unit/sagemaker/cli/compatibility/v2/modifiers/ast_converter.py	6
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_airflow.py	80
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_algorithm_image_uris.py	75
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_distribution.py	37
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_estimator_create_model.py	36
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_framework_version.py	173
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_image_uri.py	76
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_matching.py	43
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_parsing.py	30
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_predictors.py	86
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_s3_session.py	40
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_serde.py	378
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_session_image_uri.py	54
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_shuffle_config.py	42
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_tensorboard_remove_param.py	33
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_tf_legacy_mode.py	146
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_tf_script_mode.py	23
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_tfs.py	66
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_training_input.py	74
tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_training_params.py	68
tests/unit/sagemaker/cli/compatibility/v2/test_ast_transformer.py	17
tests/unit/sagemaker/cli/compatibility/v2/test_file_updaters.py	76
tests/unit/sagemaker/cli/test_framework_upgrade_tool.py	241
tests/unit/sagemaker/config/__init__.py	1
tests/unit/sagemaker/config/conftest.py	241
tests/unit/sagemaker/config/test_config.py	393
tests/unit/sagemaker/config/test_config_schema.py	222
tests/unit/sagemaker/content_types/__init__.py	1
tests/unit/sagemaker/content_types/jumpstart/__init__.py	1
tests/unit/sagemaker/content_types/jumpstart/test_content_types.py	67
tests/unit/sagemaker/deserializers/__init__.py	1
tests/unit/sagemaker/deserializers/jumpstart/__init__.py	1
tests/unit/sagemaker/deserializers/jumpstart/test_deserializers.py	74
tests/unit/sagemaker/deserializers/test_deserializers.py	165
tests/unit/sagemaker/environment_variables/__init__.py	1
tests/unit/sagemaker/environment_variables/jumpstart/__init__.py	1
tests/unit/sagemaker/environment_variables/jumpstart/test_default.py	281
tests/unit/sagemaker/experiments/__init__.py	1
tests/unit/sagemaker/experiments/conftest.py	57
tests/unit/sagemaker/experiments/helpers.py	31
tests/unit/sagemaker/experiments/test_environment.py	67
tests/unit/sagemaker/experiments/test_experiment.py	258
tests/unit/sagemaker/experiments/test_helper.py	248
tests/unit/sagemaker/experiments/test_metrics.py	66
tests/unit/sagemaker/experiments/test_run.py	953
tests/unit/sagemaker/experiments/test_run_context.py	247
tests/unit/sagemaker/experiments/test_trial.py	239
tests/unit/sagemaker/experiments/test_trial_component.py	365
tests/unit/sagemaker/experiments/test_utils.py	19
tests/unit/sagemaker/explainer/__init__.py	1
tests/unit/sagemaker/explainer/test_explainer_config.py	119
tests/unit/sagemaker/feature_store/feature_processor/lineage/test_constants.py	354
tests/unit/sagemaker/feature_store/feature_processor/lineage/test_feature_group_lineage_entity_handler.py	45
tests/unit/sagemaker/feature_store/feature_processor/lineage/test_feature_processor_lineage.py	2743
tests/unit/sagemaker/feature_store/feature_processor/lineage/test_lineage_association_handler.py	184
tests/unit/sagemaker/feature_store/feature_processor/lineage/test_pipeline_lineage_entity_handler.py	52
tests/unit/sagemaker/feature_store/feature_processor/lineage/test_pipeline_trigger.py	15
tests/unit/sagemaker/feature_store/feature_processor/lineage/test_pipeline_version_lineage_entity_handler.py	45
tests/unit/sagemaker/feature_store/feature_processor/lineage/test_s3_lineage_entity_handler.py	339
tests/unit/sagemaker/feature_store/feature_processor/test_config_uploader.py	243
tests/unit/sagemaker/feature_store/feature_processor/test_data_helpers.py	129
tests/unit/sagemaker/feature_store/feature_processor/test_data_source.py	13
tests/unit/sagemaker/feature_store/feature_processor/test_env.py	83
tests/unit/sagemaker/feature_store/feature_processor/test_event_bridge_rule_helper.py	241
tests/unit/sagemaker/feature_store/feature_processor/test_event_bridge_scheduler_helper.py	68
tests/unit/sagemaker/feature_store/feature_processor/test_factory.py	50
tests/unit/sagemaker/feature_store/feature_processor/test_feature_processor.py	83
tests/unit/sagemaker/feature_store/feature_processor/test_feature_processor_config.py	25
tests/unit/sagemaker/feature_store/feature_processor/test_feature_processor_pipeline_events.py	14
tests/unit/sagemaker/feature_store/feature_processor/test_feature_scheduler.py	935
tests/unit/sagemaker/feature_store/feature_processor/test_input_loader.py	251
tests/unit/sagemaker/feature_store/feature_processor/test_input_offset_parser.py	112
tests/unit/sagemaker/feature_store/feature_processor/test_params_loader.py	51
tests/unit/sagemaker/feature_store/feature_processor/test_spark_session_factory.py	118
tests/unit/sagemaker/feature_store/feature_processor/test_udf_arg_provider.py	185
tests/unit/sagemaker/feature_store/feature_processor/test_udf_output_receiver.py	65
tests/unit/sagemaker/feature_store/feature_processor/test_udf_wrapper.py	50
tests/unit/sagemaker/feature_store/feature_processor/test_validation.py	120
tests/unit/sagemaker/feature_store/test_dataset_builder.py	747
tests/unit/sagemaker/feature_store/test_feature_definition.py	66
tests/unit/sagemaker/feature_store/test_feature_group.py	1435
tests/unit/sagemaker/feature_store/test_feature_store.py	204
tests/unit/sagemaker/feature_store/test_feature_utils.py	87
tests/unit/sagemaker/feature_store/test_inputs.py	178
tests/unit/sagemaker/huggingface/__init__.py	1
tests/unit/sagemaker/huggingface/huggingface_utils.py	36
tests/unit/sagemaker/huggingface/test_estimator.py	332
tests/unit/sagemaker/huggingface/test_llm_utils.py	72
tests/unit/sagemaker/huggingface/test_processing.py	121
tests/unit/sagemaker/hyperparameters/__init__.py	1
tests/unit/sagemaker/hyperparameters/jumpstart/__init__.py	1
tests/unit/sagemaker/hyperparameters/jumpstart/test_default.py	171
tests/unit/sagemaker/hyperparameters/jumpstart/test_validate.py	498
tests/unit/sagemaker/image_uris/__init__.py	1
tests/unit/sagemaker/image_uris/conftest.py	21
tests/unit/sagemaker/image_uris/expected_uris.py	77
tests/unit/sagemaker/image_uris/jumpstart/__init__.py	1
tests/unit/sagemaker/image_uris/jumpstart/conftest.py	18
tests/unit/sagemaker/image_uris/jumpstart/test_catboost.py	52
tests/unit/sagemaker/image_uris/jumpstart/test_common.py	199
tests/unit/sagemaker/image_uris/jumpstart/test_huggingface.py	59
tests/unit/sagemaker/image_uris/jumpstart/test_lightgbm.py	51
tests/unit/sagemaker/image_uris/jumpstart/test_mxnet.py	51
tests/unit/sagemaker/image_uris/jumpstart/test_pytorch.py	51
tests/unit/sagemaker/image_uris/jumpstart/test_sklearn.py	75
tests/unit/sagemaker/image_uris/jumpstart/test_tensorflow.py	50
tests/unit/sagemaker/image_uris/jumpstart/test_variants.py	60
tests/unit/sagemaker/image_uris/jumpstart/test_xgboost.py	64
tests/unit/sagemaker/image_uris/regions.py	7
tests/unit/sagemaker/image_uris/test_algos.py	31
tests/unit/sagemaker/image_uris/test_autogluon.py	49
tests/unit/sagemaker/image_uris/test_base_python.py	18
tests/unit/sagemaker/image_uris/test_data_wrangler.py	36
tests/unit/sagemaker/image_uris/test_debugger.py	13
tests/unit/sagemaker/image_uris/test_djl.py	27
tests/unit/sagemaker/image_uris/test_dlc_frameworks.py	126
tests/unit/sagemaker/image_uris/test_graviton.py	169
tests/unit/sagemaker/image_uris/test_huggingface_llm.py	122
tests/unit/sagemaker/image_uris/test_model_monitor.py	14
tests/unit/sagemaker/image_uris/test_neo.py	54
tests/unit/sagemaker/image_uris/test_retrieve.py	737
tests/unit/sagemaker/image_uris/test_rl.py	42
tests/unit/sagemaker/image_uris/test_sagemaker_distribution.py	29
tests/unit/sagemaker/image_uris/test_sagemaker_geospatial.py	35
tests/unit/sagemaker/image_uris/test_sagemaker_tritonserver.py	38
tests/unit/sagemaker/image_uris/test_sklearn.py	45
tests/unit/sagemaker/image_uris/test_smp_v2.py	50
tests/unit/sagemaker/image_uris/test_sparkml.py	15
tests/unit/sagemaker/image_uris/test_stabilityai.py	20
tests/unit/sagemaker/image_uris/test_trainium.py	50
tests/unit/sagemaker/image_uris/test_xgboost.py	34
tests/unit/sagemaker/inference_recommender/__init__.py	1
tests/unit/sagemaker/inference_recommender/constants.py	152
tests/unit/sagemaker/inference_recommender/constructs.py	61
tests/unit/sagemaker/inference_recommender/test_inference_recommender_mixin.py	650
tests/unit/sagemaker/instance_types/__init__.py	1
tests/unit/sagemaker/instance_types/jumpstart/__init__.py	1
tests/unit/sagemaker/instance_types/jumpstart/test_instance_types.py	225
tests/unit/sagemaker/iterators/test_iterators.py	88
tests/unit/sagemaker/jumpstart/__init__.py	1
tests/unit/sagemaker/jumpstart/estimator/__init__.py	1
tests/unit/sagemaker/jumpstart/estimator/test_estimator.py	2127
tests/unit/sagemaker/jumpstart/estimator/test_sagemaker_config.py	497
tests/unit/sagemaker/jumpstart/hub/__init__.py	1
tests/unit/sagemaker/jumpstart/hub/test_hub.py	208
tests/unit/sagemaker/jumpstart/hub/test_interfaces.py	1025
tests/unit/sagemaker/jumpstart/hub/test_marketplace_hub_content.py	106
tests/unit/sagemaker/jumpstart/hub/test_parser_utils.py	34
tests/unit/sagemaker/jumpstart/hub/test_utils.py	225
tests/unit/sagemaker/jumpstart/model/__init__.py	1
tests/unit/sagemaker/jumpstart/model/test_model.py	1898
tests/unit/sagemaker/jumpstart/model/test_sagemaker_config.py	347
tests/unit/sagemaker/jumpstart/test_accessors.py	217
tests/unit/sagemaker/jumpstart/test_artifacts.py	369
tests/unit/sagemaker/jumpstart/test_cache.py	1075
tests/unit/sagemaker/jumpstart/test_exceptions.py	49
tests/unit/sagemaker/jumpstart/test_filters.py	316
tests/unit/sagemaker/jumpstart/test_notebook_utils.py	679
tests/unit/sagemaker/jumpstart/test_payload_utils.py	94
tests/unit/sagemaker/jumpstart/test_predictor.py	164
tests/unit/sagemaker/jumpstart/test_session_utils.py	313
tests/unit/sagemaker/jumpstart/test_types.py	1471
tests/unit/sagemaker/jumpstart/test_utils.py	2111
tests/unit/sagemaker/jumpstart/utils.py	367
tests/unit/sagemaker/lineage/conftest.py	10
tests/unit/sagemaker/lineage/test_action.py	301
tests/unit/sagemaker/lineage/test_artifact.py	443
tests/unit/sagemaker/lineage/test_association.py	190
tests/unit/sagemaker/lineage/test_context.py	436
tests/unit/sagemaker/lineage/test_dataset_artifact.py	141
tests/unit/sagemaker/lineage/test_endpoint_context.py	159
tests/unit/sagemaker/lineage/test_image_artifact.py	44
tests/unit/sagemaker/lineage/test_lineage_trial_component.py	194
tests/unit/sagemaker/lineage/test_model_artifact.py	59
tests/unit/sagemaker/lineage/test_model_package_group_context.py	18
tests/unit/sagemaker/lineage/test_query.py	498
tests/unit/sagemaker/lineage/test_utils.py	5
tests/unit/sagemaker/lineage/test_visualizer.py	164
tests/unit/sagemaker/local/__init__.py	1
tests/unit/sagemaker/local/test_local_data.py	160
tests/unit/sagemaker/local/test_local_entities.py	251
tests/unit/sagemaker/local/test_local_image.py	1007
tests/unit/sagemaker/local/test_local_pipeline.py	1249
tests/unit/sagemaker/local/test_local_session.py	898
tests/unit/sagemaker/local/test_local_utils.py	163
tests/unit/sagemaker/metric_definitions/__init__.py	1
tests/unit/sagemaker/metric_definitions/jumpstart/__init__.py	1
tests/unit/sagemaker/metric_definitions/jumpstart/test_default.py	159
tests/unit/sagemaker/mlflow/__init__.py	1
tests/unit/sagemaker/mlflow/test_forward_sagemaker_metrics.py	197
tests/unit/sagemaker/mlflow/test_tracking_server.py	25
tests/unit/sagemaker/model/test_deploy.py	991
tests/unit/sagemaker/model/test_edge.py	51
tests/unit/sagemaker/model/test_framework_model.py	468
tests/unit/sagemaker/model/test_model.py	1272
tests/unit/sagemaker/model/test_model_package.py	425
tests/unit/sagemaker/model/test_neo.py	332
tests/unit/sagemaker/model_uris/__init__.py	1
tests/unit/sagemaker/model_uris/jumpstart/__init__.py	1
tests/unit/sagemaker/model_uris/jumpstart/test_combined_artifact.py	18
tests/unit/sagemaker/model_uris/jumpstart/test_common.py	154
tests/unit/sagemaker/model_uris/jumpstart/test_xgboost.py	27
tests/unit/sagemaker/modules/__init__.py	1
tests/unit/sagemaker/modules/local_core/test_local_container.py	136
tests/unit/sagemaker/modules/test_utils.py	117
tests/unit/sagemaker/modules/train/__init__.py	1
tests/unit/sagemaker/modules/train/container_drivers/__init__.py	1
tests/unit/sagemaker/modules/train/container_drivers/test_mpi_driver.py	133
tests/unit/sagemaker/modules/train/container_drivers/test_mpi_utils.py	67
tests/unit/sagemaker/modules/train/container_drivers/test_torchrun_driver.py	122
tests/unit/sagemaker/modules/train/container_drivers/test_utils.py	95
tests/unit/sagemaker/modules/train/sm_recipes/__init__.py	1
tests/unit/sagemaker/modules/train/sm_recipes/test_utils.py	177
tests/unit/sagemaker/modules/train/test_model_trainer.py	1094
tests/unit/sagemaker/monitor/__init__.py	1
tests/unit/sagemaker/monitor/test_clarify_model_monitor.py	1501
tests/unit/sagemaker/monitor/test_cron_expression_generator.py	17
tests/unit/sagemaker/monitor/test_data_capture_config.py	47
tests/unit/sagemaker/monitor/test_data_quality_monitoring_config.py	41
tests/unit/sagemaker/monitor/test_model_monitoring.py	1970
tests/unit/sagemaker/partner_app/__init__.py	1
tests/unit/sagemaker/partner_app/test_auth_provider.py	99
tests/unit/sagemaker/partner_app/test_auth_utils.py	72
tests/unit/sagemaker/remote_function/core/test_pipeline_variables.py	310
tests/unit/sagemaker/remote_function/core/test_serialization.py	398
tests/unit/sagemaker/remote_function/core/test_stored_function.py	328
tests/unit/sagemaker/remote_function/runtime_environment/test_bootstrap_runtime_environment.py	643
tests/unit/sagemaker/remote_function/runtime_environment/test_mpi_utils.py	94
tests/unit/sagemaker/remote_function/runtime_environment/test_runtime_environment_manager.py	391
tests/unit/sagemaker/remote_function/test_client.py	1158
tests/unit/sagemaker/remote_function/test_custom_file_filter.py	77
tests/unit/sagemaker/remote_function/test_errors.py	70
tests/unit/sagemaker/remote_function/test_invoke_function.py	185
tests/unit/sagemaker/remote_function/test_job.py	2215
tests/unit/sagemaker/remote_function/test_logging_config.py	10
tests/unit/sagemaker/remote_function/test_spark_config.py	29
tests/unit/sagemaker/resource_requirements/__init__.py	1
tests/unit/sagemaker/resource_requirements/jumpstart/__init__.py	1
tests/unit/sagemaker/resource_requirements/jumpstart/test_resource_requirements.py	132
tests/unit/sagemaker/script_uris/__init__.py	1
tests/unit/sagemaker/script_uris/jumpstart/__init__.py	1
tests/unit/sagemaker/script_uris/jumpstart/test_combined_script_artifact.py	18
tests/unit/sagemaker/script_uris/jumpstart/test_common.py	154
tests/unit/sagemaker/script_uris/jumpstart/test_pytorch.py	27
tests/unit/sagemaker/serializers/__init__.py	1
tests/unit/sagemaker/serializers/jumpstart/__init__.py	1
tests/unit/sagemaker/serializers/jumpstart/test_serializers.py	78
tests/unit/sagemaker/serializers/test_serializers.py	229
tests/unit/sagemaker/serve/__init__.py	1
tests/unit/sagemaker/serve/builder/__init__.py	1
tests/unit/sagemaker/serve/builder/test_djl_builder.py	352
tests/unit/sagemaker/serve/builder/test_js_builder.py	1766
tests/unit/sagemaker/serve/builder/test_model_builder.py	3645
tests/unit/sagemaker/serve/builder/test_requirements_manager.py	55
tests/unit/sagemaker/serve/builder/test_schema_builder.py	121
tests/unit/sagemaker/serve/builder/test_serve_settings.py	56
tests/unit/sagemaker/serve/builder/test_tei_builder.py	187
tests/unit/sagemaker/serve/builder/test_tensorflow_serving_builder.py	58
tests/unit/sagemaker/serve/builder/test_tgi_builder.py	254
tests/unit/sagemaker/serve/builder/test_transformers_builder.py	193
tests/unit/sagemaker/serve/constants.py	303
tests/unit/sagemaker/serve/detector/__init__.py	1
tests/unit/sagemaker/serve/detector/test_dependency_manager.py	98
tests/unit/sagemaker/serve/detector/test_image_detector.py	172
tests/unit/sagemaker/serve/detector/test_pickle_dependencies.py	194
tests/unit/sagemaker/serve/detector/test_pickler.py	9
tests/unit/sagemaker/serve/mode/__init__.py	1
tests/unit/sagemaker/serve/mode/test_in_process_mode.py	165
tests/unit/sagemaker/serve/mode/test_local_container_mode.py	3
tests/unit/sagemaker/serve/mode/test_sagemaker_endpoint_mode.py	3
tests/unit/sagemaker/serve/model_format/mlflow/test_mlflow_utils.py	460
tests/unit/sagemaker/serve/model_server/constants.py	20
tests/unit/sagemaker/serve/model_server/djl_serving/test_djl_prepare.py	148
tests/unit/sagemaker/serve/model_server/in_process_model_server/test_app.py	60
tests/unit/sagemaker/serve/model_server/multi_model_server/test_multi_model_server_prepare.py	78
tests/unit/sagemaker/serve/model_server/tei/__init__.py	1
tests/unit/sagemaker/serve/model_server/tei/test_server.py	113
tests/unit/sagemaker/serve/model_server/tensorflow_serving/test_tf_prepare.py	89
tests/unit/sagemaker/serve/model_server/tensorflow_serving/test_tf_server.py	76
tests/unit/sagemaker/serve/model_server/tgi/test_tgi_prepare.py	131
tests/unit/sagemaker/serve/model_server/tgi/test_tgi_serving.py	232
tests/unit/sagemaker/serve/model_server/torchserve/__init__.py	1
tests/unit/sagemaker/serve/model_server/torchserve/test_inference_py.py	3
tests/unit/sagemaker/serve/model_server/torchserve/test_prepare.py	83
tests/unit/sagemaker/serve/model_server/triton/__init__.py	1
tests/unit/sagemaker/serve/model_server/triton/test_server.py	138
tests/unit/sagemaker/serve/model_server/triton/test_triton_builder.py	98
tests/unit/sagemaker/serve/utils/test_hardware_detector.py	84
tests/unit/sagemaker/serve/utils/test_lineage_utils.py	338
tests/unit/sagemaker/serve/utils/test_optimize_utils.py	509
tests/unit/sagemaker/serve/utils/test_predictors.py	57
tests/unit/sagemaker/serve/utils/test_task.py	29
tests/unit/sagemaker/serve/utils/test_telemetry_logger.py	290
tests/unit/sagemaker/serve/validations/__init__.py	1
tests/unit/sagemaker/serve/validations/test_check_image_and_hardware_type.py	139
tests/unit/sagemaker/serverless/__init__.py	1
tests/unit/sagemaker/serverless/test_model.py	11
tests/unit/sagemaker/serverless/test_predictor.py	12
tests/unit/sagemaker/serverless/test_serverless_inference_config.py	43
tests/unit/sagemaker/spark/test_processing.py	1205
tests/unit/sagemaker/telemetry/test_telemetry_logging.py	259
tests/unit/sagemaker/tensorflow/test_estimator.py	620
tests/unit/sagemaker/tensorflow/test_estimator_attach.py	182
tests/unit/sagemaker/tensorflow/test_estimator_init.py	120
tests/unit/sagemaker/tensorflow/test_tfs.py	401
tests/unit/sagemaker/test_instance_types_gpu_info.py	10
tests/unit/sagemaker/test_studio.py	101
tests/unit/sagemaker/training_compiler/__init__.py	2
tests/unit/sagemaker/training_compiler/test_huggingface_pytorch_compiler.py	652
tests/unit/sagemaker/training_compiler/test_huggingface_tensorflow_compiler.py	534
tests/unit/sagemaker/training_compiler/test_pytorch_compiler.py	571
tests/unit/sagemaker/training_compiler/test_tensorflow_compiler.py	463
tests/unit/sagemaker/utilities/__init__.py	1
tests/unit/sagemaker/utilities/test_cache.py	138
tests/unit/sagemaker/utilities/test_search_expression.py	52
tests/unit/sagemaker/workflow/__init__.py	1
tests/unit/sagemaker/workflow/conftest.py	53
tests/unit/sagemaker/workflow/helpers.py	113
tests/unit/sagemaker/workflow/test_airflow.py	1557
tests/unit/sagemaker/workflow/test_automl_step.py	448
tests/unit/sagemaker/workflow/test_callback_step.py	139
tests/unit/sagemaker/workflow/test_clarify_check_step.py	594
tests/unit/sagemaker/workflow/test_condition_step.py	539
tests/unit/sagemaker/workflow/test_conditions.py	138
tests/unit/sagemaker/workflow/test_emr_step.py	430
tests/unit/sagemaker/workflow/test_entities.py	141
tests/unit/sagemaker/workflow/test_execution_variables.py	26
tests/unit/sagemaker/workflow/test_fail_step.py	97
tests/unit/sagemaker/workflow/test_function_step.py	396
tests/unit/sagemaker/workflow/test_functions.py	221
tests/unit/sagemaker/workflow/test_lambda_step.py	244
tests/unit/sagemaker/workflow/test_model_step.py	1209
tests/unit/sagemaker/workflow/test_notebook_job_step.py	550
tests/unit/sagemaker/workflow/test_parameters.py	73
tests/unit/sagemaker/workflow/test_pipeline.py	865
tests/unit/sagemaker/workflow/test_pipeline_definition_config.py	9
tests/unit/sagemaker/workflow/test_pipeline_experiment_config.py	11
tests/unit/sagemaker/workflow/test_pipeline_graph.py	647
tests/unit/sagemaker/workflow/test_pipeline_session.py	323
tests/unit/sagemaker/workflow/test_processing_step.py	1060
tests/unit/sagemaker/workflow/test_properties.py	122
tests/unit/sagemaker/workflow/test_quality_check_step.py	394
tests/unit/sagemaker/workflow/test_repack_model_script.py	106
tests/unit/sagemaker/workflow/test_retry.py	102
tests/unit/sagemaker/workflow/test_selective_execution_config.py	10
tests/unit/sagemaker/workflow/test_step_collections.py	1199
tests/unit/sagemaker/workflow/test_steps.py	1389
tests/unit/sagemaker/workflow/test_steps_compiler.py	853
tests/unit/sagemaker/workflow/test_training_step.py	808
tests/unit/sagemaker/workflow/test_transform_step.py	253
tests/unit/sagemaker/workflow/test_triggers.py	211
tests/unit/sagemaker/workflow/test_tuning_step.py	286
tests/unit/sagemaker/workflow/test_utilities.py	135
tests/unit/sagemaker/workflow/test_utils.py	206
tests/unit/sagemaker/wrangler/__init__.py	1
tests/unit/sagemaker/wrangler/test_ingestion.py	156
tests/unit/sagemaker/wrangler/test_processing.py	108
tests/unit/test_algorithm.py	869
tests/unit/test_amazon_estimator.py	389
tests/unit/test_analytics.py	294
tests/unit/test_artifact_analytics.py	37
tests/unit/test_chainer.py	582
tests/unit/test_clarify.py	2620
tests/unit/test_collection.py	102
tests/unit/test_common.py	287
tests/unit/test_create_deploy_entities.py	97
tests/unit/test_default_bucket.py	158
tests/unit/test_deprecations.py	162
tests/unit/test_djl_inference.py	149
tests/unit/test_endpoint_from_job.py	87
tests/unit/test_endpoint_from_model_data.py	172
tests/unit/test_estimator.py	5159
tests/unit/test_exception_on_bad_status.py	89
tests/unit/test_experiments_analytics.py	301
tests/unit/test_fm.py	295
tests/unit/test_fw_utils.py	896
tests/unit/test_git_utils.py	439
tests/unit/test_hyperparameter.py	43
tests/unit/test_init.py	4
tests/unit/test_inputs.py	185
tests/unit/test_ipinsights.py	264
tests/unit/test_job.py	668
tests/unit/test_kmeans.py	233
tests/unit/test_knn.py	255
tests/unit/test_lambda_helper.py	427
tests/unit/test_lda.py	196
tests/unit/test_linear_learner.py	379
tests/unit/test_model_card.py	1761
tests/unit/test_multidatamodel.py	260
tests/unit/test_mxnet.py	1019
tests/unit/test_ntm.py	260
tests/unit/test_object2vec.py	263
tests/unit/test_pca.py	208
tests/unit/test_pipeline_model.py	387
tests/unit/test_predictor.py	692
tests/unit/test_predictor_async.py	365
tests/unit/test_processing.py	1444
tests/unit/test_profiler_app.py	76
tests/unit/test_profiler_config.py	678
tests/unit/test_pytorch.py	1001
tests/unit/test_randomcutforest.py	190
tests/unit/test_rl.py	525
tests/unit/test_s3.py	247
tests/unit/test_session.py	6287
tests/unit/test_sklearn.py	496
tests/unit/test_sparkml_serving.py	44
tests/unit/test_tensorboard.py	663
tests/unit/test_timeout.py	255
tests/unit/test_transformer.py	656
tests/unit/test_tuner.py	1763
tests/unit/test_upload_data.py	97
tests/unit/test_upload_string_as_file_body.py	54
tests/unit/test_user_agent.py	42
tests/unit/test_utils.py	1791
tests/unit/test_vpc_utils.py	44
tests/unit/test_xgboost.py	566
tests/unit/tuner_test_utils.py	283