Skip to content

Commit 05d11d1

Browse files
committed
fix pylint
1 parent ee8ecf4 commit 05d11d1

File tree

2 files changed

+4
-7
lines changed

2 files changed

+4
-7
lines changed

tests/test_internals.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
import tf2onnx
2020
import tf2onnx.utils
2121
from tf2onnx.graph_matcher import OpTypePattern, GraphMatcher
22-
from tf2onnx.graph import Graph, GraphUtil
22+
from tf2onnx.graph import GraphUtil
2323

2424
# pylint: disable=missing-docstring
2525

tools/onnx-experiments.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,9 @@
1919

2020
import numpy as np
2121
import onnx
22-
from onnx import numpy_helper
2322

2423
import tf2onnx.utils
25-
from tf2onnx.graph import Graph, GraphUtil
24+
from tf2onnx.graph import GraphUtil
2625

2726
logging.basicConfig(level=logging.INFO)
2827
log = logging.getLogger("onnx-experiments")
@@ -38,10 +37,8 @@ def get_args():
3837

3938

4039
def load_graph(fname):
41-
with open(fname, "rb") as f:
42-
data = f.read()
43-
model_proto = onnx.ModelProto()
44-
g = GraphUtil.create_graph_from_onnx_model(model_proto)
40+
model_proto = onnx.ModelProto()
41+
g = GraphUtil.create_graph_from_onnx_model(model_proto)
4542
return g, model_proto.producer_name
4643

4744

0 commit comments

Comments
 (0)