Removing incorrect test and removing cast from split impl

apbose · apbose · commit e95e0e4fabfb · 2023-09-20T18:21:31.000-07:00
diff --git a/py/torch_tensorrt/dynamo/conversion/impl/split.py b/py/torch_tensorrt/dynamo/conversion/impl/split.py
@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Optional, Sequence, Tuple, Union, cast
+from typing import Any, Dict, List, Optional, Sequence, Tuple, Union
 
 import numpy as np
 import torch
diff --git a/tests/py/dynamo/conversion/test_split_aten.py b/tests/py/dynamo/conversion/test_split_aten.py
@@ -33,7 +33,6 @@ def forward(self, input):
     @parameterized.expand(
         [
             ("split_size_or_sections_list_no_dim_list", [1, 4]),
-            ("split_size_or_sections_list_no_dim_not_full_list", [1, 3]),
         ]
     )
     def test_split_list(self, _, split_size_or_tensor):

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from typing import Any, Dict, List, Optional, Sequence, Tuple, Union, cast`
	`1`	`+from typing import Any, Dict, List, Optional, Sequence, Tuple, Union`
`2`	`2`
`3`	`3`	`import numpy as np`
`4`	`4`	`import torch`
Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,6 @@ def forward(self, input):`
`33`	`33`	`@parameterized.expand(`
`34`	`34`	`[`
`35`	`35`	`("split_size_or_sections_list_no_dim_list", [1, 4]),`
`36`		`- ("split_size_or_sections_list_no_dim_not_full_list", [1, 3]),`
`37`	`36`	`]`
`38`	`37`	`)`
`39`	`38`	`def test_split_list(self, _, split_size_or_tensor):`