fix the import issue of warnings

pytorch · Jun 10, 2024 · c37a30e · c37a30e
1 parent 98ed700
commit c37a30e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchbenchmark/util/extra_args.py b/torchbenchmark/util/extra_args.py
@@ -1,5 +1,6 @@
 import argparse
 import enum
+import warnings
 from typing import List, Optional, Tuple
 
 from torchbenchmark.util.backends import BACKENDS, list_backends