Skip to content

Commit

Permalink
update jenkinsfile
Browse files Browse the repository at this point in the history
  • Loading branch information
michaelpoluektov committed Oct 12, 2023
1 parent 4c33f11 commit c80526d
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion Jenkinsfile
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,10 @@ pipeline {
} }
stage("Notebook Tests") { steps { withVenv {
sh "pip install pytest nbmake"
sh "pytest --nbmake ./docs/notebooks/**/*.ipynb"
sh "pytest --nbmake ./docs/notebooks/*.ipynb"
// Test the pytorch to keras notebooks overnight? Need to manually install all requirements
// Also these train models so it takes a while
// sh "pytest --nbmake ./docs/notebooks/*.ipynb"
} } }
}
post { cleanup { xcoreCleanSandbox() } }
Expand All @@ -95,6 +98,10 @@ def runTests(String platform) {
}
if (platform == "device") {
sh "pytest integration_tests/runner.py --models_path integration_tests/models/complex_models/non-bnns -n 1 --junitxml=integration_tests/integration_device_junit.xml"
// lstms are always problematic
sh "pytest integration_tests/runner.py --models_path integration_tests/models/non-bnns/test_lstm -n 1"
// test a float32 layer
sh "pytest integration_tests/runner.py --models_path integration_tests/models/non-bnns/test_detection_postprocess -n 1"
} else if (platform == "host") {
sh "pytest integration_tests/runner.py --models_path integration_tests/models/non-bnns -n 8 --junitxml=integration_tests/integration_non_bnns_junit.xml"
sh "pytest integration_tests/runner.py --models_path integration_tests/models/bnns --bnn -n 8 --junitxml=integration_tests/integration_bnns_junit.xml"
Expand Down

0 comments on commit c80526d

Please sign in to comment.