From ddedb5269600a2c130b58649ed0ade282eb25595 Mon Sep 17 00:00:00 2001 From: torzdf <36920800+torzdf@users.noreply.github.com> Date: Tue, 20 Dec 2022 15:35:10 +0000 Subject: [PATCH] Update tests for tf2.10 --- tests/startup_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/startup_test.py b/tests/startup_test.py index 85a92a50..ece5af1e 100644 --- a/tests/startup_test.py +++ b/tests/startup_test.py @@ -31,5 +31,5 @@ def test_backend(dummy): # pylint:disable=unused-argument def test_keras(dummy): # pylint:disable=unused-argument """ Sanity check to ensure that tensorflow keras is being used for CPU and standard keras for AMD. """ - assert ((_BACKEND == "cpu" and keras.__version__ in ("2.7.0", "2.8.0", "2.9.0")) or + assert ((_BACKEND == "cpu" and keras.__version__ in ("2.7.0", "2.8.0", "2.9.0", "2.10.0")) or (_BACKEND == "amd" and keras.__version__ == "2.2.4"))