Missed lint issues

Author: markdlv-sf

src/datacustomcode/client.py

@@ -21,12 +21,10 @@
     Optional,
 )
 
-from pyspark.sql import SparkSession
-
-from datacustomcode.config import SparkConfig, config
-from datacustomcode.spark.default import DefaultSparkSessionProvider
+from datacustomcode.config import config
 from datacustomcode.file.path.default import DefaultFindFilePath
 from datacustomcode.io.reader.base import BaseDataCloudReader
+from datacustomcode.spark.default import DefaultSparkSessionProvider
 
 if TYPE_CHECKING:
     from pathlib import Path
@@ -127,7 +125,6 @@ def __new__(
                     raise ValueError(
                         "Spark config is required when reader/writer is not provided"
                     )
-                from datacustomcode.spark.base import BaseSparkSessionProvider
 
                 provider: BaseSparkSessionProvider
                 if spark_provider is not None:

tests/spark/test_session_provider.py

@@ -2,8 +2,6 @@
 
 from typing import TYPE_CHECKING, Any
 
-import builtins
-
 from datacustomcode.client import Client
 from datacustomcode.config import (
     AccessLayerObjectConfig,
@@ -92,10 +90,11 @@ def test_client_uses_provider_from_config(monkeypatch):
 
     global_config.update(cfg)
 
-    Client()  # noqa: F841
+    Client()
     assert MockReader.last_spark is SENTINEL_SPARK
     assert MockWriter.last_spark is SENTINEL_SPARK
 
+
 class ExplicitProvider(BaseSparkSessionProvider):
     CONFIG_NAME = "ExplicitProvider"
 
@@ -122,7 +121,6 @@ def test_client_explicit_provider_overrides_config(monkeypatch):
     global_config.update(cfg)
 
     provider = ExplicitProvider()
-    Client(spark_provider=provider)  # noqa: F841
+    Client(spark_provider=provider)
     assert MockReader.last_spark is SENTINEL_SPARK
     assert MockWriter.last_spark is SENTINEL_SPARK
-