Skip to content

Commit e5fc2da

Browse files
committed
Enable keras v1 api tests.
Signed-off-by: Jay Zhang <[email protected]>
1 parent 0d6c98e commit e5fc2da

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

.github/actions/unit_test/action.yml

+2
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ runs:
4646
export TF2ONNX_SKIP_TFLITE_TESTS=${{ inputs.skip_tflite }}
4747
export TF2ONNX_SKIP_TFJS_TESTS=True
4848
export TF2ONNX_SKIP_TF_TESTS=False
49+
export TF_USE_LEGACY_KERAS=True
4950
python -m pytest --cov=tf2onnx --cov-report=term --disable-pytest-warnings -r s tests --cov-append --junitxml=junit/test-results.xml
5051
ls
5152
@@ -58,5 +59,6 @@ runs:
5859
set TF2ONNX_SKIP_TFLITE_TESTS=${{ inputs.skip_tflite }}
5960
set TF2ONNX_SKIP_TFJS_TESTS=True
6061
set TF2ONNX_SKIP_TF_TESTS=False
62+
set TF_USE_LEGACY_KERAS=True
6163
python -m pytest --cov=tf2onnx --cov-report=term --disable-pytest-warnings -r s tests --cov-append --junitxml=junit/test-results.xml
6264
ls

tests/utils/setup_test_env.sh

+2
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,8 @@ pip install tensorflow==$TF_VERSION
2828
pip uninstall -y protobuf
2929
pip install "protobuf~=3.20"
3030

31+
pip install tf_keras
32+
3133
python setup.py install
3234

3335
echo "----- List all of depdencies:"

0 commit comments

Comments
 (0)