Merge pull request #618 from onnx/gs/fixbuild

fix build

Author: guschmue

README.md

@@ -17,6 +17,8 @@ A list of models that we use for testing can be found [here](tests/run_pretraine
 Supported RNN classes and APIs: LSTMCell, BasicLSTMCell, GRUCell, GRUBlockCell, MultiRNNCell, and user defined RNN cells inheriting rnn_cell_impl.RNNCell, used along with DropoutWrapper, BahdanauAttention, AttentionWrapper.
 Check [tips](examples/rnn_tips.md) when converting RNN models.
 
+You find a list of supported Tensorflow ops and their mapping to ONNX [here](support_status.md).
+
 Tensorflow has broad functionality and occacional mapping it to ONNX creates issues.
 The common issues we run into we try to document here [Troubleshooting Guide](Troubleshooting.md).
 

tools/gen_doc.py

@@ -1,7 +1,13 @@
+"""
+tool to list tensorflow op to onnx op mapping in markdown
+"""
+
 import argparse
-from tf2onnx.handler import tf_op
+
 from collections import OrderedDict
 
+from tf2onnx.handler import tf_op
+
 parser = argparse.ArgumentParser()
 parser.add_argument('filename',
                     type=str,
@@ -30,6 +36,3 @@
         # Document support status according to the gathered mapping.
         for tf_op, supported_versions in tf_op_to_versions.items():
             doc_file.write("| {} | {} |\n".format(tf_op, supported_versions))
-
-    doc_file.close()
-