Skip to content

Commit e4226a6

Browse files
committed
fix build
1 parent 0dd7f6d commit e4226a6

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

tools/gen_doc.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,13 @@
1+
"""
2+
tool to list tensorflow op to onnx op mapping in markdown
3+
"""
4+
15
import argparse
2-
from tf2onnx.handler import tf_op
6+
37
from collections import OrderedDict
48

9+
from tf2onnx.handler import tf_op
10+
511
parser = argparse.ArgumentParser()
612
parser.add_argument('filename',
713
type=str,
@@ -30,6 +36,3 @@
3036
# Document support status according to the gathered mapping.
3137
for tf_op, supported_versions in tf_op_to_versions.items():
3238
doc_file.write("| {} | {} |\n".format(tf_op, supported_versions))
33-
34-
doc_file.close()
35-

0 commit comments

Comments
 (0)