From c37a30e4b271cfc8e50c4f53d55517191cdcfec9 Mon Sep 17 00:00:00 2001
From: "Liao, Wei" <wei.liao@intel.com>
Date: Mon, 10 Jun 2024 15:28:49 +0800
Subject: [PATCH] fix the import issue of warnings

---
 torchbenchmark/util/extra_args.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/torchbenchmark/util/extra_args.py b/torchbenchmark/util/extra_args.py
index 7e84986426..84975b3923 100644
--- 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