@@ -17,26 +17,26 @@ check_logs() {
17
17
run_for_framework () {
18
18
if [ " $zero_code_change_test " = " enable" ] ; then
19
19
# 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
21
21
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
23
23
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
27
27
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
29
29
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
32
32
fi
33
33
34
34
else
35
35
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
38
38
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
40
40
fi
41
41
fi
42
42
}
@@ -48,7 +48,7 @@ export SMDEBUG_LOG_LEVEL=info
48
48
49
49
export OUT_DIR=upload/$CURRENT_COMMIT_PATH
50
50
export 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
52
52
53
53
run_for_framework core
54
54
0 commit comments