Correct type hint

Erik-Lundell · web-flow · commit 124fc23c4f41 · 2024-09-24T10:50:29.000+02:00
diff --git a/backends/arm/passes/scalars_to_attribute_pass.py b/backends/arm/passes/scalars_to_attribute_pass.py
@@ -28,7 +28,7 @@ class ScalarsToAttributePass(ExportPass):
         torch.ops.aten.div.Tensor,
     ]
 
-    def call(self, graph_module: GraphModule) -> GraphModule:
+    def call(self, graph_module: GraphModule) -> PassResult:
         for n in graph_module.graph.nodes:
             n = cast(Node, n)
             if n.op != "call_function" or n.target not in self.targeted_ops:

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ class ScalarsToAttributePass(ExportPass):`
`28`	`28`	`torch.ops.aten.div.Tensor,`
`29`	`29`	`]`
`30`	`30`
`31`		`- def call(self, graph_module: GraphModule) -> GraphModule:`
	`31`	`+ def call(self, graph_module: GraphModule) -> PassResult:`
`32`	`32`	`for n in graph_module.graph.nodes:`
`33`	`33`	`n = cast(Node, n)`
`34`	`34`	`if n.op != "call_function" or n.target not in self.targeted_ops:`