Fix buildspec yaml file for TF integration tests (#66)

awslabs · Sep 3, 2020 · ed56c1d · ed56c1d
1 parent bd5e60b
commit ed56c1d
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/..._profiler_sagemaker_integration_tests.yml → ...agemaker_tensorflow_integration_tests.yml b/..._profiler_sagemaker_integration_tests.yml → ...agemaker_tensorflow_integration_tests.yml
@@ -22,7 +22,7 @@ phases:
       # build pip wheel of the latest smdebug
       - cd $CODEBUILD_SRC_DIR  && python setup.py bdist_wheel --universal && pip install --force-reinstall dist/*.whl
       # Runs the smprofiler sagemaker integration tests
-      - cd $CODEBUILD_SRC_DIR  && python -m pytest -v -s -W=ignore --html=$REPORT_DIR/profiler_report_analysis.html --self-contained-html tests/sagemaker/
+      - cd $CODEBUILD_SRC_DIR  && python -m pytest -v -s -W=ignore --html=$REPORT_DIR/profiler_report_analysis.html --self-contained-html tests/sagemaker/test_profiler_tensorflow.py
 
   post_build:
     commands:

diff --git a/tests/sagemaker/profiler_tests_config.yaml → ...ker/tensorflow_profiler_tests_config.yaml b/tests/sagemaker/profiler_tests_config.yaml → ...ker/tensorflow_profiler_tests_config.yaml
diff --git a/tests/sagemaker/test_profiler.py → tests/sagemaker/test_profiler_tensorflow.py b/tests/sagemaker/test_profiler.py → tests/sagemaker/test_profiler_tensorflow.py
@@ -72,7 +72,7 @@ def _download_artifacts(s3_path, out_dir):
 
 
 def _get_estimator_list(index, job_type):
-    fhandle = open("tests/sagemaker/profiler_tests_config.yaml")
+    fhandle = open("tests/sagemaker/tensorflow_profiler_tests_config.yaml")
     config_file = yaml.load(fhandle, Loader=yaml.FullLoader)
     fhandle.close()
     estimator_list = []