test: fix import in proxy test (#22)

isha97 · web-flow · commit 61e7b47ebab2 · 2025-01-21T13:40:06.000-08:00
diff --git a/tests/unit/test_proxy.py b/tests/unit/test_proxy.py
@@ -17,7 +17,7 @@
 
 import pytest
 
-from google.cloud.dataproc_spark_connect.client.proxy import connect_sockets
+from google.cloud.spark_connect.client.proxy import connect_sockets
 
 
 @pytest.fixture
