add missing pytest import

varun-edachali-dbx · varun-edachali-dbx · commit 6aa2991b5171 · 2025-07-16T10:24:40.000+05:30
Signed-off-by: varun-edachali-dbx &lt;varun.edachali@databricks.com&gt;
diff --git a/tests/e2e/common/large_queries_mixin.py b/tests/e2e/common/large_queries_mixin.py
@@ -2,6 +2,8 @@
 import math
 import time
 
+import pytest
+
 log = logging.getLogger(__name__)
 
 
