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 357 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/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/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/iris/failure_script.py 3 tests/data/iris/iris-dnn-classifier.py 36 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/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/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/sagemaker_rl/__init__.py 1 tests/data/sagemaker_rl/coach_launcher.py 237 tests/data/sagemaker_rl/configuration_list.py 59 tests/data/sip/preprocessor.py 80 tests/data/sip/sklearn_source_dir/inference.py 56 tests/data/sip/training.py 56 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_v1.py 82 tests/data/tensorflow_mnist/mnist_v2.py 137 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/123/saved_model.pb 531 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/00000123/saved_model.pb 531 tests/data/tfs/tfs-test-model-with-inference/code/inference.py 10 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/emr-script.sh 2 tests/data/workflow/inference.py 1 tests/data/xgboost_abalone/abalone.py 34 tests/integ/__init__.py 132 tests/integ/auto_ml_utils.py 24 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 27 tests/integ/sagemaker/lineage/__init__.py 1 tests/integ/sagemaker/lineage/conftest.py 698 tests/integ/sagemaker/lineage/helpers.py 52 tests/integ/sagemaker/lineage/test_action.py 112 tests/integ/sagemaker/lineage/test_artifact.py 116 tests/integ/sagemaker/lineage/test_association.py 60 tests/integ/sagemaker/lineage/test_context.py 81 tests/integ/sagemaker/lineage/test_dataset_artifact.py 30 tests/integ/sagemaker/lineage/test_endpoint_context.py 69 tests/integ/sagemaker/lineage/test_image_artifact.py 10 tests/integ/sagemaker/lineage/test_lineage_trial_component.py 14 tests/integ/sagemaker/lineage/test_model_artifact.py 36 tests/integ/sagemaker/lineage/test_model_package_group_context.py 4 tests/integ/test_airflow_config.py 517 tests/integ/test_artifact_analytics.py 60 tests/integ/test_async_inference.py 81 tests/integ/test_auto_ml.py 275 tests/integ/test_byo_estimator.py 97 tests/integ/test_chainer.py 134 tests/integ/test_clarify.py 217 tests/integ/test_clarify_model_monitor.py 487 tests/integ/test_data_capture_config.py 216 tests/integ/test_data_upload.py 23 tests/integ/test_debugger.py 650 tests/integ/test_edge.py 55 tests/integ/test_experiments_analytics.py 209 tests/integ/test_factorization_machines.py 73 tests/integ/test_feature_store.py 252 tests/integ/test_git.py 227 tests/integ/test_horovod.py 73 tests/integ/test_horovod_mx.py 74 tests/integ/test_huggingface.py 151 tests/integ/test_inference_pipeline.py 177 tests/integ/test_ipinsights.py 38 tests/integ/test_kmeans.py 103 tests/integ/test_kmeans_efs_fsx.py 196 tests/integ/test_knn.py 65 tests/integ/test_lda.py 42 tests/integ/test_linear_learner.py 157 tests/integ/test_local_mode.py 334 tests/integ/test_marketplace.py 226 tests/integ/test_model_monitor.py 1299 tests/integ/test_model_quality_monitor.py 412 tests/integ/test_monitoring_files.py 310 tests/integ/test_multi_variant_endpoint.py 235 tests/integ/test_multidatamodel.py 390 tests/integ/test_mxnet.py 389 tests/integ/test_neo_mxnet.py 140 tests/integ/test_neo_pytorch.py 67 tests/integ/test_ntm.py 42 tests/integ/test_object2vec.py 51 tests/integ/test_pca.py 70 tests/integ/test_predict_with_inference_id.py 58 tests/integ/test_processing.py 752 tests/integ/test_profiler.py 356 tests/integ/test_pytorch.py 237 tests/integ/test_randomcutforest.py 32 tests/integ/test_record_set.py 17 tests/integ/test_rl.py 95 tests/integ/test_s3.py 175 tests/integ/test_serverless_inference.py 33 tests/integ/test_session.py 27 tests/integ/test_sklearn.py 228 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 275 tests/integ/test_sparkml_serving.py 43 tests/integ/test_tf.py 256 tests/integ/test_tf_efs_fsx.py 202 tests/integ/test_tfs.py 192 tests/integ/test_training_compiler.py 88 tests/integ/test_transformer.py 358 tests/integ/test_tuner.py 716 tests/integ/test_tuner_multi_algo.py 161 tests/integ/test_workflow.py 2543 tests/integ/test_workflow_experiment.py 168 tests/integ/test_workflow_retry.py 228 tests/integ/test_workflow_with_clarify.py 228 tests/integ/test_workflow_with_clarify_check_steps.py 248 tests/integ/test_workflow_with_quality_check_steps.py 410 tests/integ/test_xgboost.py 62 tests/integ/timeout.py 132 tests/integ/vpc_test_utils.py 154 tests/scripts/fs_mount_setup.sh 72 tests/scripts/run-notebook-test.sh 85 tests/unit/__init__.py 4 tests/unit/output_capturer.py 13 tests/unit/sagemaker/__init__.py 1 tests/unit/sagemaker/async_inference/__init__.py 1 tests/unit/sagemaker/async_inference/test_async_inference_config.py 61 tests/unit/sagemaker/async_inference/test_async_inference_response.py 65 tests/unit/sagemaker/async_inference/test_waiter_config.py 23 tests/unit/sagemaker/automl/test_auto_ml.py 591 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 386 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 240 tests/unit/sagemaker/feature_store/test_feature_definition.py 29 tests/unit/sagemaker/feature_store/test_feature_store.py 400 tests/unit/sagemaker/feature_store/test_inputs.py 57 tests/unit/sagemaker/huggingface/__init__.py 1 tests/unit/sagemaker/huggingface/huggingface_utils.py 19 tests/unit/sagemaker/huggingface/test_estimator.py 297 tests/unit/sagemaker/huggingface/test_processing.py 117 tests/unit/sagemaker/image_uris/__init__.py 1 tests/unit/sagemaker/image_uris/expected_uris.py 20 tests/unit/sagemaker/image_uris/regions.py 7 tests/unit/sagemaker/image_uris/test_algos.py 161 tests/unit/sagemaker/image_uris/test_autogluon.py 72 tests/unit/sagemaker/image_uris/test_data_wrangler.py 37 tests/unit/sagemaker/image_uris/test_debugger.py 36 tests/unit/sagemaker/image_uris/test_dlc_frameworks.py 331 tests/unit/sagemaker/image_uris/test_model_monitor.py 33 tests/unit/sagemaker/image_uris/test_neo.py 80 tests/unit/sagemaker/image_uris/test_retrieve.py 608 tests/unit/sagemaker/image_uris/test_rl.py 101 tests/unit/sagemaker/image_uris/test_sklearn.py 67 tests/unit/sagemaker/image_uris/test_sparkml.py 39 tests/unit/sagemaker/image_uris/test_xgboost.py 103 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 120 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 446 tests/unit/sagemaker/lineage/test_utils.py 5 tests/unit/sagemaker/lineage/test_visualizer.py 164 tests/unit/sagemaker/model/test_deploy.py 372 tests/unit/sagemaker/model/test_edge.py 46 tests/unit/sagemaker/model/test_framework_model.py 401 tests/unit/sagemaker/model/test_model.py 252 tests/unit/sagemaker/model/test_model_package.py 141 tests/unit/sagemaker/model/test_neo.py 269 tests/unit/sagemaker/monitor/__init__.py 1 tests/unit/sagemaker/monitor/test_clarify_model_monitor.py 1273 tests/unit/sagemaker/monitor/test_cron_expression_generator.py 15 tests/unit/sagemaker/monitor/test_data_capture_config.py 46 tests/unit/sagemaker/monitor/test_model_monitoring.py 1165 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 15 tests/unit/sagemaker/spark/test_processing.py 1065 tests/unit/sagemaker/tensorflow/test_estimator.py 454 tests/unit/sagemaker/tensorflow/test_estimator_attach.py 180 tests/unit/sagemaker/tensorflow/test_estimator_init.py 114 tests/unit/sagemaker/tensorflow/test_tfs.py 349 tests/unit/sagemaker/test_deserializers.py 150 tests/unit/sagemaker/test_serializers.py 209 tests/unit/sagemaker/test_studio.py 69 tests/unit/sagemaker/training_compiler/__init__.py 2 tests/unit/sagemaker/training_compiler/test_huggingface_compiler.py 440 tests/unit/sagemaker/workflow/__init__.py 1 tests/unit/sagemaker/workflow/helpers.py 8 tests/unit/sagemaker/workflow/test_callback_step.py 122 tests/unit/sagemaker/workflow/test_clarify_check_step.py 603 tests/unit/sagemaker/workflow/test_condition_step.py 61 tests/unit/sagemaker/workflow/test_conditions.py 131 tests/unit/sagemaker/workflow/test_emr_step.py 145 tests/unit/sagemaker/workflow/test_entities.py 26 tests/unit/sagemaker/workflow/test_execution_variables.py 5 tests/unit/sagemaker/workflow/test_functions.py 74 tests/unit/sagemaker/workflow/test_lambda_step.py 176 tests/unit/sagemaker/workflow/test_parameters.py 65 tests/unit/sagemaker/workflow/test_pipeline.py 403 tests/unit/sagemaker/workflow/test_pipeline_experiment_config.py 11 tests/unit/sagemaker/workflow/test_properties.py 68 tests/unit/sagemaker/workflow/test_quality_check_step.py 370 tests/unit/sagemaker/workflow/test_repack_model_script.py 105 tests/unit/sagemaker/workflow/test_retry.py 102 tests/unit/sagemaker/workflow/test_step_collections.py 786 tests/unit/sagemaker/workflow/test_steps.py 1210 tests/unit/sagemaker/workflow/test_utils.py 160 tests/unit/sagemaker/wrangler/__init__.py 1 tests/unit/sagemaker/wrangler/test_ingestion.py 156 tests/unit/sagemaker/wrangler/test_processing.py 104 tests/unit/test_airflow.py 1555 tests/unit/test_algorithm.py 845 tests/unit/test_amazon_estimator.py 350 tests/unit/test_analytics.py 291 tests/unit/test_artifact_analytics.py 37 tests/unit/test_chainer.py 530 tests/unit/test_clarify.py 1046 tests/unit/test_common.py 287 tests/unit/test_create_deploy_entities.py 97 tests/unit/test_default_bucket.py 65 tests/unit/test_deprecations.py 162 tests/unit/test_endpoint_from_job.py 80 tests/unit/test_endpoint_from_model_data.py 147 tests/unit/test_estimator.py 2926 tests/unit/test_exception_on_bad_status.py 77 tests/unit/test_experiments_analytics.py 301 tests/unit/test_fm.py 291 tests/unit/test_fw_utils.py 517 tests/unit/test_git_utils.py 411 tests/unit/test_hyperparameter.py 43 tests/unit/test_image.py 654 tests/unit/test_init.py 4 tests/unit/test_inputs.py 139 tests/unit/test_ipinsights.py 260 tests/unit/test_job.py 472 tests/unit/test_kmeans.py 229 tests/unit/test_knn.py 251 tests/unit/test_lambda_helper.py 264 tests/unit/test_lda.py 192 tests/unit/test_linear_learner.py 375 tests/unit/test_local_data.py 160 tests/unit/test_local_entities.py 116 tests/unit/test_local_session.py 701 tests/unit/test_local_utils.py 55 tests/unit/test_multidatamodel.py 258 tests/unit/test_mxnet.py 966 tests/unit/test_ntm.py 256 tests/unit/test_object2vec.py 259 tests/unit/test_pca.py 204 tests/unit/test_pipeline_model.py 300 tests/unit/test_predictor.py 461 tests/unit/test_predictor_async.py 250 tests/unit/test_processing.py 1117 tests/unit/test_profiler_config.py 497 tests/unit/test_pytorch.py 589 tests/unit/test_randomcutforest.py 186 tests/unit/test_rl.py 525 tests/unit/test_s3.py 94 tests/unit/test_session.py 2102 tests/unit/test_sklearn.py 496 tests/unit/test_sparkml_serving.py 40 tests/unit/test_timeout.py 252 tests/unit/test_transformer.py 453 tests/unit/test_tuner.py 1145 tests/unit/test_upload_data.py 97 tests/unit/test_upload_string_as_file_body.py 54 tests/unit/test_utils.py 502 tests/unit/test_vpc_utils.py 44 tests/unit/test_xgboost.py 566 tests/unit/tuner_test_utils.py 263