Remove node_map from the GraphInfo (#198)

nod-ai · Nov 21, 2023 · 372a1c1 · 372a1c1
1 parent 0daa8dd
commit 372a1c1
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/python/shark_turbine/importers/onnx_importer.py b/python/shark_turbine/importers/onnx_importer.py
@@ -88,7 +88,6 @@ class GraphInfo:
     def __init__(self, model_info: ModelInfo, graph_proto: onnx.GraphProto):
         self.model_info = model_info
         self.graph_proto = graph_proto
-        self.node_map: dict[str, onnx.NodeProto] = {n.name: n for n in graph_proto.node}
         self.initializer_map: dict[str, onnx.TensorProto] = {
             n.name: n for n in graph_proto.initializer
         }