Fixing lint errors

Author: ritaagarwala-sf

src/datacustomcode/client.py

@@ -196,7 +196,7 @@ def read_dlo(self, name: str, row_limit: int = 1000) -> PySparkDataFrame:
             A PySpark DataFrame containing the DLO data.
         """
         self._record_dlo_access(name)
-        return self._reader.read_dlo(name, row_limit=row_limit)
+        return self._reader.read_dlo(name, row_limit=row_limit)  # type: ignore[no-any-return]
 
     def read_dmo(self, name: str, row_limit: int = 1000) -> PySparkDataFrame:
         """Read a DMO from Data Cloud.
@@ -209,7 +209,7 @@ def read_dmo(self, name: str, row_limit: int = 1000) -> PySparkDataFrame:
             A PySpark DataFrame containing the DMO data.
         """
         self._record_dmo_access(name)
-        return self._reader.read_dmo(name, row_limit=row_limit)
+        return self._reader.read_dmo(name, row_limit=row_limit)  # type: ignore[no-any-return]
 
     def write_to_dlo(
         self, name: str, dataframe: PySparkDataFrame, write_mode: WriteMode, **kwargs
@@ -222,7 +222,7 @@ def write_to_dlo(
             write_mode: The write mode to use for writing to the DLO.
         """
         self._validate_data_layer_history_does_not_contain(DataCloudObjectType.DMO)
-        return self._writer.write_to_dlo(name, dataframe, write_mode, **kwargs)
+        return self._writer.write_to_dlo(name, dataframe, write_mode, **kwargs)  # type: ignore[no-any-return]
 
     def write_to_dmo(
         self, name: str, dataframe: PySparkDataFrame, write_mode: WriteMode, **kwargs
@@ -235,17 +235,17 @@ def write_to_dmo(
             write_mode: The write mode to use for writing to the DMO.
         """
         self._validate_data_layer_history_does_not_contain(DataCloudObjectType.DLO)
-        return self._writer.write_to_dmo(name, dataframe, write_mode, **kwargs)
+        return self._writer.write_to_dmo(name, dataframe, write_mode, **kwargs)  # type: ignore[no-any-return]
 
     def call_llm_gateway(self, LLM_MODEL_ID: str, prompt: str, maxTokens: int) -> str:
         if self._proxy is None:
             raise ValueError("No proxy configured; set proxy or proxy_config")
-        return self._proxy.call_llm_gateway(LLM_MODEL_ID, prompt, maxTokens)
+        return self._proxy.call_llm_gateway(LLM_MODEL_ID, prompt, maxTokens)  # type: ignore[no-any-return]
 
     def find_file_path(self, file_name: str) -> Path:
         """Return a file path"""
 
-        return self._file.find_file_path(file_name)
+        return self._file.find_file_path(file_name)  # type: ignore[no-any-return]
 
     def _validate_data_layer_history_does_not_contain(
         self, data_cloud_object_type: DataCloudObjectType

src/datacustomcode/config.py

@@ -36,10 +36,7 @@
 # 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
 from datacustomcode.proxy.base import BaseProxyAccessLayer
-from datacustomcode.proxy.client.base import BaseProxyClient  # noqa: TCH001
 from datacustomcode.spark.base import BaseSparkSessionProvider
 
 DEFAULT_CONFIG_NAME = "config.yaml"
@@ -48,6 +45,10 @@
 if TYPE_CHECKING:
     from pyspark.sql import SparkSession
 
+    from datacustomcode.io.reader.base import BaseDataCloudReader
+    from datacustomcode.io.writer.base import BaseDataCloudWriter
+    from datacustomcode.proxy.client.base import BaseProxyClient
+
 
 class ForceableConfig(BaseModel):
     force: bool = Field(

src/datacustomcode/io/reader/query_api.py

@@ -179,7 +179,7 @@ def read_dlo(
             self, "_sf_cli_reader", None
         )
         if sf_cli_reader is not None:
-            return sf_cli_reader.read_dlo(name, schema, row_limit)
+            return sf_cli_reader.read_dlo(name, schema, row_limit)  # type: ignore[no-any-return]
 
         query = SQL_QUERY_TEMPLATE.format(name, row_limit)
 
@@ -214,7 +214,7 @@ def read_dmo(
             self, "_sf_cli_reader", None
         )
         if sf_cli_reader is not None:
-            return sf_cli_reader.read_dmo(name, schema, row_limit)
+            return sf_cli_reader.read_dmo(name, schema, row_limit)  # type: ignore[no-any-return]
 
         query = SQL_QUERY_TEMPLATE.format(name, row_limit)
 

src/datacustomcode/llm_gateway/base.py

@@ -15,13 +15,19 @@
 from __future__ import annotations
 
 from abc import abstractmethod
+from typing import TYPE_CHECKING
 
-from datacustomcode.llm_gateway.types.generate_text_request import GenerateTextRequest
-from datacustomcode.llm_gateway.types.generate_text_response import GenerateTextResponse
+if TYPE_CHECKING:
+    from datacustomcode.llm_gateway.types.generate_text_request import (
+        GenerateTextRequest,
+    )
+    from datacustomcode.llm_gateway.types.generate_text_response import (
+        GenerateTextResponse,
+    )
 
 
 class LLMGateway:
-    def __init__(self):
+    def __init__(self) -> None:
         pass
 
     @abstractmethod