@@ -17,26 +17,26 @@ check_logs() {
1717run_for_framework () {
1818 if [ " $zero_code_change_test " = " enable" ] ; then
1919 # ignoring some test becuase they require multiple frmaeworks to be installed, these tests need to be broken down
20- python -m pytest --cov=./ --cov-append --durations=50 --html=$REPORT_DIR /report_$1 .html -v -s --self-contained-html --ignore=tests/core/test_paths.py --ignore=tests/core/test_index_utils.py --ignore=tests/core/test_collections.py tests/$1
20+ python -m pytest ${code_coverage_smdebug : + --cov=./ --cov-append} --durations=50 --html=$REPORT_DIR /report_$1 .html -v -s --self-contained-html --ignore=tests/core/test_paths.py --ignore=tests/core/test_index_utils.py --ignore=tests/core/test_collections.py tests/$1
2121 if [ " $1 " = " mxnet" ] ; then
22- python -m pytest --cov=./ --cov-append tests/zero_code_change/test_mxnet_gluon_integration.py
22+ python -m pytest ${code_coverage_smdebug : + --cov=./ --cov-append} tests/zero_code_change/test_mxnet_gluon_integration.py
2323 elif [ " $1 " = " pytorch" ] ; then
24- python -m pytest --cov=./ --cov-append tests/zero_code_change/test_pytorch_integration.py
25- python -m pytest --cov=./ --cov-append tests/zero_code_change/test_pytorch_multiprocessing.py
26- python -m pytest --cov=./ --cov-append tests/zero_code_change/test_training_with_no_grad_updates.py
24+ python -m pytest ${code_coverage_smdebug : + --cov=./ --cov-append} tests/zero_code_change/test_pytorch_integration.py
25+ python -m pytest ${code_coverage_smdebug : + --cov=./ --cov-append} tests/zero_code_change/test_pytorch_multiprocessing.py
26+ python -m pytest ${code_coverage_smdebug : + --cov=./ --cov-append} tests/zero_code_change/test_training_with_no_grad_updates.py
2727 elif [ " $1 " = " tensorflow" ] ; then
28- python -m pytest --cov=./ --cov-append tests/zero_code_change/test_tensorflow_integration.py
28+ python -m pytest ${code_coverage_smdebug : + --cov=./ --cov-append} tests/zero_code_change/test_tensorflow_integration.py
2929 elif [ " $1 " = " tensorflow2" ] ; then
30- python -m pytest --cov=./ --cov-append tests/zero_code_change/test_tensorflow2_gradtape_integration.py
31- python -m pytest --cov=./ --cov-append tests/zero_code_change/test_tensorflow2_integration.py
30+ python -m pytest ${code_coverage_smdebug : + --cov=./ --cov-append} tests/zero_code_change/test_tensorflow2_gradtape_integration.py
31+ python -m pytest ${code_coverage_smdebug : + --cov=./ --cov-append} tests/zero_code_change/test_tensorflow2_integration.py
3232 fi
3333
3434 else
3535 if [ " $1 " = " tensorflow2" ] ; then
36- python -m pytest --cov=./ --cov-append --durations=50 --html=$REPORT_DIR /report_$1 /eager_mode.html -v -s --self-contained-html tests/$1
37- python -m pytest --cov=./ --cov-append --durations=50 --non-eager --html=$REPORT_DIR /report_$1 /non_eager_mode.html -v -s --self-contained-html tests/$1
36+ python -m pytest ${code_coverage_smdebug : + --cov=./ --cov-append} --durations=50 --html=$REPORT_DIR /report_$1 /eager_mode.html -v -s --self-contained-html tests/$1
37+ python -m pytest ${code_coverage_smdebug : + --cov=./ --cov-append} --durations=50 --non-eager --html=$REPORT_DIR /report_$1 /non_eager_mode.html -v -s --self-contained-html tests/$1
3838 else
39- python -m pytest --cov=./ --cov-append --durations=50 --html=$REPORT_DIR /report_$1 .html -v -s --self-contained-html tests/$1
39+ python -m pytest ${code_coverage_smdebug : + --cov=./ --cov-append} --durations=50 --html=$REPORT_DIR /report_$1 .html -v -s --self-contained-html tests/$1
4040 fi
4141 fi
4242}
@@ -48,7 +48,7 @@ export SMDEBUG_LOG_LEVEL=info
4848
4949export OUT_DIR=upload/$CURRENT_COMMIT_PATH
5050export REPORT_DIR=$OUT_DIR /pytest_reports
51- python -m pytest --cov=./ --cov-append -v -W=ignore --durations=50 --html=$REPORT_DIR /report_analysis.html --self-contained-html tests/analysis
51+ python -m pytest ${code_coverage_smdebug : + --cov=./ --cov-append} -v -W=ignore --durations=50 --html=$REPORT_DIR /report_analysis.html --self-contained-html tests/analysis
5252
5353run_for_framework core
5454
0 commit comments