fix pylint

Author: RandyShuai

tf2onnx/custom_opsets/onnx_ml.py

@@ -4,32 +4,35 @@
 import numpy as np
 
 from onnx import onnx_pb
-from onnx.onnx_pb import TensorProto
-from tf2onnx import constants, utils
+from tf2onnx import constants
 from tf2onnx.handler import tf_op
 
 @tf_op("HashTableV2")
 class HashTable:
     @classmethod
     def version_11(cls, ctx, node, **kwargs):
+        """ HashTable will be removed """
         pass
 
 @tf_op("LookupTableFindV2")
 class LookupTableFind:
     @classmethod
     def version_11(cls, ctx, node, **kwargs):
+        """ convert lookup to category mapper """
         table_node = node.inputs[0]
         file_path = table_node.get_attr_value("shared_name")[11:-6]
         cats_int64s = []
         cats_strings = []
         with open(file_path, 'r') as f:
-            for i,s in enumerate(f.readlines()):
+            for i, s in enumerate(f.readlines()):
                 cats_int64s.append(i)
                 cats_strings.append(s.strip())
         node_inputs = node.input
         node_outputs = node.output
         ctx.remove_node(node.name)
-        ctx.make_node("CategoryMapper", domain=constants.AI_ONNX_ML_DOMAIN, inputs=node_inputs[1:2], outputs=node_outputs, attr={'cats_int64s':cats_int64s,'cats_strings':cats_strings})
+        ctx.make_node("CategoryMapper", domain=constants.AI_ONNX_ML_DOMAIN,
+                      inputs=node_inputs[1: 2], outputs=node_outputs,
+                      attr={'cats_int64s': cats_int64s, 'cats_strings': cats_strings})
         customer_nodes = ctx.find_output_consumers(table_node.output[0])
         if len(customer_nodes) == 0:
             ctx.remove_node(table_node.name)

tf2onnx/utils.py

@@ -179,7 +179,7 @@ def get_tf_shape_attr(node):
 
 def get_tf_output_shapes_attr(node):
     """Get output shapes from tensorflow attr "output_shapes"."""
-    dims = [] 
+    dims = []
     try:
         shapes = get_tf_node_attr(node, "output_shapes")
         for shape in shapes: