fix converter test error (#3083)

pytorch · Aug 14, 2024 · 03092ba · 03092ba
1 parent 3ee2d81
commit 03092ba
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/py/dynamo/conversion/test_argmin_aten.py b/tests/py/dynamo/conversion/test_argmin_aten.py
@@ -2,6 +2,7 @@
 import torch.nn as nn
 from parameterized import parameterized
 from torch.testing._internal.common_utils import run_tests
+from torch_tensorrt import Input
 
 from .harness import DispatchTestCase