Call tf_keras for keras 3.x.

Signed-off-by: Jay Zhang <[email protected]>

Author: fatcat-z

tests/keras2onnx_unit_tests/conftest.py

@@ -13,6 +13,12 @@
 K = keras.backend
 
 
+def is_keras_3():
+    if hasattr(keras, '__version__'):
+        return keras.__version__.startswith("3.")
+
+    return False
+
 @pytest.fixture(scope='function')
 def runner():
     np.random.seed(42)
@@ -25,10 +31,15 @@ def runner():
     def runner_func(*args, **kwargs):
         return run_onnx_runtime(*args, model_files, **kwargs)
 
-    # Ensure Keras layer naming is reset for each function
-    K.reset_uids()
-    # Reset the TensorFlow session to avoid resource leaking between tests
-    K.clear_session()
+    if is_keras_3():
+        import tf_keras
+        tf_keras.backend.reset_uids()
+        tf_keras.backend.clear_session()
+    else:
+        # Ensure Keras layer naming is reset for each function
+        K.reset_uids()
+        # Reset the TensorFlow session to avoid resource leaking between tests
+        K.clear_session()
 
     # Provide wrapped run_onnx_runtime function
     yield runner_func