removed the changes in test_fetches

saishreeeee · saishreeeee · commit ae90deed6ecb · 2025-07-02T15:50:12.000+05:30
Signed-off-by: Sai Shree Pradhan &lt;saishree.pradhan@databricks.com&gt;
diff --git a/tests/unit/test_fetches.py b/tests/unit/test_fetches.py
@@ -1,6 +1,6 @@
 import unittest
 import pytest
-from unittest.mock import Mock, patch
+from unittest.mock import Mock
 
 try:
     import pyarrow as pa
@@ -58,7 +58,7 @@ def make_dummy_result_set_from_initial_results(initial_results):
             for col_id in range(num_cols)
         ]
         return rs
-    
+
     @staticmethod
     def make_dummy_result_set_from_batch_list(batch_list):
         batch_index = 0
