From c44483694a6b3476d89968f6ee9463886d433211 Mon Sep 17 00:00:00 2001 From: Jerry Ge Date: Thu, 21 Mar 2024 17:33:55 +0000 Subject: Fix framework test generation - The framework test generation script is broken for some tests because TF2.16 has an issue with Keras 3.0 - Enforce the script to use Keras 2.0 and we also need to run pip install -U tf_keras together with this change Signed-off-by: Jerry Ge Change-Id: I519aa89c5d5bacf659a809d9eda456bd83b33527 --- verif/frameworks/test_builder.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/verif/frameworks/test_builder.py b/verif/frameworks/test_builder.py index 4bf8070..900f5fa 100644 --- a/verif/frameworks/test_builder.py +++ b/verif/frameworks/test_builder.py @@ -1,9 +1,13 @@ # Copyright (c) 2020-2024, ARM Limited. # SPDX-License-Identifier: Apache-2.0 +import os + import numpy as np import tensorflow as tf from frameworks.tensor_gen import TGen +os.environ["TF_USE_LEGACY_KERAS"] = "1" + class TBuilder: """The member functions build the tensorflow operators into small networks -- cgit v1.2.1