From 93fe4c6f65e6e8246eec0d114e05211a7c1c7e64 Mon Sep 17 00:00:00 2001 From: reuvenp Date: Sat, 12 Oct 2024 14:05:34 +0300 Subject: [PATCH] add keras pytests to workflows --- .github/workflows/run_keras_tests.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/run_keras_tests.yml b/.github/workflows/run_keras_tests.yml index 045f8fc40..99523fd5d 100644 --- a/.github/workflows/run_keras_tests.yml +++ b/.github/workflows/run_keras_tests.yml @@ -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