flake

Signed-off-by: Boris Fomitchev <[email protected]>

Author: borisfom

tests/test_trt_compile.py

@@ -36,7 +36,7 @@ class ListAdd(torch.nn.Module):
     def __init__(self):
         super().__init__()
 
-    def forward(self, x: list[torch.Tensor], y: torch.Tensor, z: torch.Tensor, bs: float = float(0.1)):
+    def forward(self, x: list[torch.Tensor], y: torch.Tensor, z: torch.Tensor, bs: float = 0.1):
         y1 = y.clone()
         x1 = x.copy()
         z1 = z + y