Skip to content

Commit 6eeed4b

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent c6e11bb commit 6eeed4b

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/test_trt_compile.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313

1414
import tempfile
1515
import unittest
16-
from typing import List
1716

1817
import torch
1918
from parameterized import parameterized
@@ -37,7 +36,7 @@ class ListAdd(torch.nn.Module):
3736
def __init__(self):
3837
super().__init__()
3938

40-
def forward(self, x: List[torch.Tensor], y: torch.Tensor, z: torch.Tensor, bs: float = float(0.1)):
39+
def forward(self, x: list[torch.Tensor], y: torch.Tensor, z: torch.Tensor, bs: float = float(0.1)):
4140
y1 = y.clone()
4241
x1 = x.copy()
4342
z1 = z + y

0 commit comments

Comments
 (0)