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"))