Skip to content

Commit 73a9d34

Browse files
committed
minor changes
1 parent 6b11e35 commit 73a9d34

File tree

2 files changed

+5
-12
lines changed

2 files changed

+5
-12
lines changed

tests/common.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -124,11 +124,11 @@ def _append_message(reason, message):
124124
return reason
125125

126126

127-
def check_tf_max_version(max_required_version, message=""):
127+
def check_tf_max_version(max_accepted_version, message=""):
128128
""" Skip if tf_version > max_required_version """
129129
config = get_test_config()
130-
reason = _append_message("conversion requires tf <= {}".format(max_required_version), message)
131-
return unittest.skipIf(config.tf_version > LooseVersion(max_required_version), reason)
130+
reason = _append_message("conversion requires tf <= {}".format(max_accepted_version), message)
131+
return unittest.skipIf(config.tf_version > LooseVersion(max_accepted_version), reason)
132132

133133

134134
def check_tf_min_version(min_required_version, message=""):

tf2onnx/onnx_opset/math.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -32,21 +32,14 @@ class RealDiv(common.BroadcastOp):
3232
pass
3333

3434

35-
@tf_op(["Abs", "Ceil", "Elu", "Exp", "Floor", "Log", "LogSoftmax", "Neg", "Relu", "Sigmoid", "Sqrt", "Tanh",
36-
"Softplus", "Softsign", "Reciprocal"])
35+
@tf_op(["Abs", "Ceil", "Elu", "Exp", "Floor", "LeakyRelu", "Log", "LogSoftmax", "Neg", "Relu", "Sigmoid", "Sqrt",
36+
"Tanh", "Softplus", "Softsign", "Reciprocal"])
3737
class DirectOp:
3838
@classmethod
3939
def version_4(cls, ctx, node, **kwargs):
4040
pass
4141

4242

43-
@tf_op(["LeakyRelu"])
44-
class TrigOpSinceOpset6:
45-
@classmethod
46-
def version_6(cls, ctx, node, **kwargs):
47-
pass
48-
49-
5043
@tf_op(["Acos", "Asin", "Atan", "Cos", "Sin", "Tan"])
5144
class TrigOpSinceOpset7:
5245
@classmethod

0 commit comments

Comments
 (0)