fix linting CI failures

Author: atulikumwenayo

src/datacustomcode/auth.py

@@ -170,7 +170,7 @@ def do_oauth_browser_flow(
 
     # Start callback server
     click.echo(f"\nStarting local callback server on {redirect_uri}...")
-    server, actual_port = _run_oauth_callback_server(redirect_uri, auth_code_queue)
+    server, _actual_port = _run_oauth_callback_server(redirect_uri, auth_code_queue)
 
     # Build authorization URL with final redirect_uri
     auth_url = (

src/datacustomcode/cmd.py

@@ -104,6 +104,6 @@ def _cmd_output(
 
 
 def cmd_output(*cmd: str, **kwargs: Any) -> Union[str, None]:
-    returncode, stdout_b, stderr_b = _cmd_output(*cmd, **kwargs)
+    _returncode, stdout_b, _stderr_b = _cmd_output(*cmd, **kwargs)
     stdout = stdout_b.decode() if stdout_b is not None else None
     return stdout

src/datacustomcode/config.py

@@ -37,8 +37,8 @@
 # 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: TCH002
-from datacustomcode.io.writer.base import BaseDataCloudWriter  # noqa: TCH002
+from datacustomcode.io.reader.base import BaseDataCloudReader
+from datacustomcode.io.writer.base import BaseDataCloudWriter
 from datacustomcode.spark.base import BaseSparkSessionProvider
 
 if TYPE_CHECKING:
@@ -53,7 +53,7 @@ class AccessLayerObjectConfig(BaseObjectConfig, 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):
@@ -78,7 +78,7 @@ class SparkProviderConfig(BaseObjectConfig, Generic[_P]):
 
     def to_object(self) -> _P:
         type_ = self.type_base.subclass_from_config_name(self.type_config_name)
-        return cast(_P, type_(**self.options))
+        return cast("_P", type_(**self.options))
 
 
 class ClientConfig(BaseConfig):

src/datacustomcode/einstein_predictions_config.py

@@ -37,7 +37,7 @@ class EinsteinPredictionsObjectConfig(BaseObjectConfig, Generic[_E]):
 
     def to_object(self) -> _E:
         type_ = self.type_base.subclass_from_config_name(self.type_config_name)
-        return cast(_E, type_(**self.options))
+        return cast("_E", type_(**self.options))
 
 
 class EinsteinPredictionsConfig(BaseConfig):

src/datacustomcode/llm_gateway/default.py

@@ -19,12 +19,10 @@
     Optional,
 )
 
-from loguru import logger
 import requests
+from loguru import logger
 
 from datacustomcode.einstein_platform_client import EinsteinPlatformClient
-
-
 from datacustomcode.llm_gateway.base import LLMGateway
 from datacustomcode.llm_gateway.types.generate_text_request import GenerateTextRequest
 from datacustomcode.llm_gateway.types.generate_text_response import GenerateTextResponse
@@ -50,9 +48,7 @@ def generate_text(self, request: GenerateTextRequest) -> GenerateTextResponse:
             f"{request.model_name}/generations"
         )
 
-        payload: Dict[str, Any] = {
-            "prompt": request.prompt
-        }
+        payload: Dict[str, Any] = {"prompt": request.prompt}
 
         if request.localization:
             payload["localization"] = request.localization
@@ -76,6 +72,5 @@ def generate_text(self, request: GenerateTextRequest) -> GenerateTextResponse:
             raise RuntimeError(f"Generate text request failed: {e}") from e
 
         return GenerateTextResponse(
-            status_code=response.status_code,
-            data=self.parse_response(response)
+            status_code=response.status_code, data=self.parse_response(response)
         )

src/datacustomcode/llm_gateway_config.py

@@ -37,7 +37,7 @@ class LLMGatewayObjectConfig(BaseObjectConfig, Generic[_E]):
 
     def to_object(self) -> _E:
         type_ = self.type_base.subclass_from_config_name(self.type_config_name)
-        return cast(_E, type_(**self.options))
+        return cast("_E", type_(**self.options))
 
 
 class LLMGatewayConfig(BaseConfig):

src/datacustomcode/templates/function/payload/entrypoint.py

@@ -60,22 +60,28 @@ def make_einstein_prediction(runtime: Runtime) -> None:
 
     prediction_response = runtime.einstein_predictions.predict(prediction_request)
     print(
-        f"Einstein prediction results - success: [{prediction_response.is_success}] \
-response data: {prediction_response.data}"
+        f"Einstein prediction results - success: [{prediction_response.is_success}] "
+        f"response data: {prediction_response.data}"
     )
 
 
 def generate_text(runtime: Runtime):
     builder = GenerateTextRequestBuilder()
-    llm_request = builder.set_prompt("Generate 2 dog names").\
-        set_model("sfdc_ai__DefaultGPT52").build()
+    llm_request = (
+        builder.set_prompt("Generate 2 dog names")
+        .set_model("sfdc_ai__DefaultGPT52")
+        .build()
+    )
     llm_response = runtime.llm_gateway.generate_text(llm_request)
 
     if llm_response.is_success:
         print(llm_response.text)
     else:
         print(llm_response.error_code)
 
+
+
+
 def function(request: dict, runtime: Runtime) -> dict:
     logger.info("Inside Function")
     logger.info(request)

tests/file/test_path_default.py

@@ -72,7 +72,7 @@ def test_find_file_path_file_not_found(self):
 
             with pytest.raises(
                 FileNotFoundError,
-                match="File 'test.txt' not found in any search location",
+                match=r"File 'test\.txt' not found in any search location",
             ):
                 finder.find_file_path("test.txt")
 

tests/test_auth.py

@@ -148,7 +148,7 @@ def test_run_oauth_callback_server_valid_uri(
         mock_server.server_address = ("localhost", 5555)
         mock_tcpserver.return_value = mock_server
 
-        server, port = _run_oauth_callback_server(redirect_uri, auth_code_queue)
+        _server, port = _run_oauth_callback_server(redirect_uri, auth_code_queue)
 
         assert port == 5555
         mock_tcpserver.assert_called_once()
@@ -186,7 +186,7 @@ def test_run_oauth_callback_server_different_port(
         mock_server.server_address = ("localhost", 8080)
         mock_tcpserver.return_value = mock_server
 
-        server, port = _run_oauth_callback_server(redirect_uri, auth_code_queue)
+        _server, port = _run_oauth_callback_server(redirect_uri, auth_code_queue)
 
         assert port == 8080
         mock_tcpserver.assert_called_once()

tests/test_deploy.py

@@ -907,7 +907,7 @@ def test_verify_data_transform_config_missing(self, mock_exists):
         mock_exists.return_value = False
         with pytest.raises(
             FileNotFoundError,
-            match="config.json not found at /test/dir/payload/config.json",
+            match=r"config\.json not found at /test/dir/payload/config\.json",
         ):
             get_config("/test/dir/payload")
 
@@ -918,7 +918,7 @@ def test_verify_data_transform_config_invalid_json(self, mock_file, mock_exists)
         mock_exists.return_value = True
         with pytest.raises(
             ValueError,
-            match="config.json at /test/dir/payload/config.json is not valid JSON",
+            match=r"config\.json at /test/dir/payload/config\.json is not valid JSON",
         ):
             get_config("/test/dir/payload")
 
@@ -929,8 +929,8 @@ def test_verify_data_transform_config_missing_fields(self, mock_file, mock_exist
         mock_exists.return_value = True
         with pytest.raises(
             ValueError,
-            match="config.json at /test/dir/payload/config.json is missing "
-            "required fields: entryPoint, dataspace, permissions",
+            match=r"config\.json at /test/dir/payload/config\.json is missing "
+            r"required fields: entryPoint, dataspace, permissions",
         ):
             get_config("/test/dir/payload")