lint ruff failures

Author: joroscoSF

src/datacustomcode/__init__.py

@@ -17,4 +17,4 @@
 from datacustomcode.io.reader.query_api import QueryAPIDataCloudReader
 from datacustomcode.io.writer.print import PrintDataCloudWriter
 
-__all__ = ["Client", "QueryAPIDataCloudReader", "PrintDataCloudWriter"]
+__all__ = ["Client", "PrintDataCloudWriter", "QueryAPIDataCloudReader"]

src/datacustomcode/config.py

@@ -36,14 +36,14 @@
 # This lets all readers and writers to be findable via config
 from datacustomcode.io import *  # noqa: F403
 from datacustomcode.io.base import BaseDataAccessLayer
-from datacustomcode.io.reader.base import BaseDataCloudReader  # noqa: TCH001
-from datacustomcode.io.writer.base import BaseDataCloudWriter  # noqa: TCH001
 
 DEFAULT_CONFIG_NAME = "config.yaml"
 
 
 if TYPE_CHECKING:
     from pyspark.sql import SparkSession
+    from datacustomcode.io.reader.base import BaseDataCloudReader
+    from datacustomcode.io.writer.base import BaseDataCloudWriter
 
 
 class ForceableConfig(BaseModel):
@@ -72,7 +72,7 @@ class AccessLayerObjectConfig(ForceableConfig, Generic[_T]):
 
     def to_object(self, spark: SparkSession) -> _T:
         type_ = self.type_base.subclass_from_config_name(self.type_config_name)
-        return cast(_T, type_(spark=spark, **self.options))
+        return cast("_T", type_(spark=spark, **self.options))
 
 
 class SparkConfig(ForceableConfig):
@@ -90,8 +90,8 @@ class SparkConfig(ForceableConfig):
 
 
 class ClientConfig(BaseModel):
-    reader_config: Union[AccessLayerObjectConfig[BaseDataCloudReader], None] = None
-    writer_config: Union[AccessLayerObjectConfig[BaseDataCloudWriter], None] = None
+    reader_config: Union[AccessLayerObjectConfig["BaseDataCloudReader"], None] = None
+    writer_config: Union[AccessLayerObjectConfig["BaseDataCloudWriter"], None] = None
     spark_config: Union[SparkConfig, None] = None
 
     def update(self, other: ClientConfig) -> ClientConfig:

tests/test_deploy.py

@@ -1,7 +1,6 @@
 """Tests for the deploy module."""
 
 import shutil
-import sys
 from unittest.mock import (
     MagicMock,
     call,
@@ -242,18 +241,18 @@ def test_prepare_dependency_archive_file_copy_failure(
         # Create a custom mock for shutil.copy that raises FileNotFoundError
         # only for the specific calls we want to test
         original_copy = shutil.copy
-        
+
         def mock_copy(src, dst):
             if src == "requirements.txt" or src == "build_native_dependencies.sh":
                 raise FileNotFoundError("File not found")
             return original_copy(src, dst)
-        
+
         with patch("datacustomcode.deploy.shutil.copy", side_effect=mock_copy):
             # Call the function - it should catch the FileNotFoundError and call sys.exit(1)
             # We expect it to raise SystemExit (which is what sys.exit(1) does)
             with pytest.raises(SystemExit) as exc_info:
                 prepare_dependency_archive("/test/dir")
-            
+
             # Verify the exit code is 1
             assert exc_info.value.code == 1