From 40613297349587eece2bc07e327a90e1578c6cf3 Mon Sep 17 00:00:00 2001
From: Cemberk <Cem.bozkus@amd.com>
Date: Fri, 26 Apr 2024 14:14:44 -0500
Subject: [PATCH] Update test_modeling_common.py

---
 tests/test_modeling_common.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py
index 77c73f0f011..835c542fa4b 100755
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -28,6 +28,7 @@
 from typing import Dict, List, Tuple
 
 import numpy as np
+import pytest
 from pytest import mark
 
 import transformers