From 4d74fbc613cc8f2aa6a184d17397dad6b80ae5e3 Mon Sep 17 00:00:00 2001
From: Afroz Alam <afroz.alam@snowflake.com>
Date: Sun, 6 Oct 2024 21:33:15 -0700
Subject: [PATCH] fix test

---
 tests/integ/modin/conftest.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/integ/modin/conftest.py b/tests/integ/modin/conftest.py
index 514a2725460..207d640c702 100644
--- a/tests/integ/modin/conftest.py
+++ b/tests/integ/modin/conftest.py
@@ -14,7 +14,6 @@
 from pytest import fail
 
 import snowflake.snowpark.modin.plugin  # noqa: F401
-from tests.conftest import SKIP_SQL_COUNT_CHECK
 from tests.integ.modin.pandas_api_coverage import PandasAPICoverageGenerator
 from tests.integ.utils.sql_counter import (
     SqlCounter,
@@ -80,6 +79,8 @@ def auto_annotate_sql_counter(request):
 
 @pytest.fixture(autouse=True)
 def check_sql_counter_invoked(request):
+    from tests.conftest import SKIP_SQL_COUNT_CHECK
+
     do_check = (
         INTEG_PANDAS_SUBPATH in request.node.location[0]
         and running_on_public_ci()