From 24cca9a8f22dadd71e7b108b94340c7b685391b3 Mon Sep 17 00:00:00 2001 From: Jay Zhang Date: Thu, 1 Sep 2022 14:28:40 +0800 Subject: [PATCH] Update left tests in test_subclassing for the opset version issue. Signed-off-by: Jay Zhang --- tests/keras2onnx_unit_tests/test_subclassing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/keras2onnx_unit_tests/test_subclassing.py b/tests/keras2onnx_unit_tests/test_subclassing.py index 93cd376ad..f4b8ea9d0 100644 --- a/tests/keras2onnx_unit_tests/test_subclassing.py +++ b/tests/keras2onnx_unit_tests/test_subclassing.py @@ -4,7 +4,7 @@ import mock_keras2onnx import numpy as np import tensorflow as tf -from tf2onnx.keras2onnx_api import convert_keras +from test_utils import convert_keras_for_test as convert_keras from mock_keras2onnx.proto import is_tensorflow_older_than if (not mock_keras2onnx.proto.is_tf_keras) or (not mock_keras2onnx.proto.tfcompat.is_tf2):