fix the import issue of warnings (#2288)

Summary: Pull Request resolved: #2288 Reviewed By: aaronenyeshi Differential Revision: D58376939 Pulled By: xuzhao9 fbshipit-source-id: 696af9ab978da96b1a09605736d3a4d6cd6c6326
pytorch · Jun 10, 2024 · 417731f · 417731f
1 parent e08d888
commit 417731f
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