Skip to content

Commit

Permalink
add keras pytests to workflows
Browse files Browse the repository at this point in the history
  • Loading branch information
reuvenp committed Oct 12, 2024
1 parent ffe7c22 commit 93fe4c6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/run_keras_tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -23,12 +23,12 @@ jobs:
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
pip install tensorflow==${{ inputs.tf-version }} sony-custom-layers
pip install tensorflow==${{ inputs.tf-version }} sony-custom-layers pytest
- name: Run unittests
# Some tests are sensitive to memory because we use tf gradients on a multi-thread/process
# CPU environment (https://github.com/tensorflow/tensorflow/issues/41718).
# For this reason, if we run them in such an environment, we need to run them first non-parallel separately.
run: |
python -m unittest discover tests/keras_tests -v
pytest tests_pytest/keras

0 comments on commit 93fe4c6

Please sign in to comment.