Skip to content

Commit 6fac514

Browse files
committed
Fix type hints.
1 parent 8ce0efd commit 6fac514

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/nxp/quantizer/patterns.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -231,11 +231,11 @@ class SubTensorPattern(QuantizationPattern):
231231
Basic quantization for all inputs and output.
232232
"""
233233

234-
def partition_types(self) -> List[Type[torch.nn.Module]]:
234+
def partition_types(self) -> list[type[torch.nn.Module]]:
235235
return [torch.ops.aten.sub.Tensor]
236236

237237
def get_anchors(
238-
self, gm: fx.GraphModule, fused_partition: List[fx.GraphModule]
238+
self, gm: fx.GraphModule, fused_partition: list[fx.GraphModule]
239239
) -> PartitionAnchors | None:
240240
node = fused_partition[0].nodes[-1]
241241
inputs = [(node, 0)]

0 commit comments

Comments
 (0)