Skip to content

Commit 0052e50

Browse files
authored
Update left tests in test_subclassing for the opset version issue. (#2034)
Signed-off-by: Jay Zhang <[email protected]>
1 parent 34c8512 commit 0052e50

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/keras2onnx_unit_tests/test_subclassing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import mock_keras2onnx
55
import numpy as np
66
import tensorflow as tf
7-
from tf2onnx.keras2onnx_api import convert_keras
7+
from test_utils import convert_keras_for_test as convert_keras
88
from mock_keras2onnx.proto import is_tensorflow_older_than
99

1010
if (not mock_keras2onnx.proto.is_tf_keras) or (not mock_keras2onnx.proto.tfcompat.is_tf2):

0 commit comments

Comments
 (0)