Fixing lint error

Author: ritaagarwala-sf

src/datacustomcode/cli.py

@@ -249,7 +249,8 @@ def deploy(
             logger.info(f"Inferred use_in_feature: {use_in_feature}")
         else:
             click.secho(
-                "Error: Could not infer function invoke options. Please provide --use-in-feature",
+                "Error: Could not infer function invoke options. "
+                "Please provide --use-in-feature",
                 fg="red",
             )
             raise click.Abort()
@@ -331,7 +332,8 @@ def init(directory: str, code_type: str, use_in_feature: Optional[str]):
             click.echo(
                 "Test your function locally with "
                 + click.style(
-                    f"datacustomcode run {entrypoint_path} --test-with {test_json_path}",
+                    f"datacustomcode run {entrypoint_path} "
+                    f"--test-with {test_json_path}",
                     fg="blue",
                     bold=True,
                 )

src/datacustomcode/function_utils.py

@@ -215,7 +215,9 @@ def inspect_function_types(
         Either can be None if not found or on error
 
     Example:
-        >>> request_name, response_name = inspect_function_types("payload/entrypoint.py")
+        >>> request_name, response_name = inspect_function_types(
+        ...     "payload/entrypoint.py"
+        ... )
         >>> print(request_name)  # "SearchIndexChunkingV1Request"
         >>> print(response_name)  # "SearchIndexChunkingV1Response"
     """

src/datacustomcode/run.py

@@ -164,7 +164,8 @@ def run_function_with_test(entrypoint: str, test_file: str) -> None:
         load_function_module,
     )
 
-    # Import the entrypoint module in the current environment (with all dependencies loaded)
+    # Import the entrypoint module in the current environment
+    # (with all dependencies loaded)
     module = load_function_module(entrypoint, "entrypoint_module")
     function_callable = get_function_callable(module)
     request_type = get_request_type(entrypoint)

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

@@ -66,6 +66,9 @@ def function(
         output=chunks,
         status=SearchIndexStatusResponse(
             status_type="success",
-            status_message=f"Successfully chunked {len(request.input)} documents into {len(chunks)} chunks",
+            status_message=(
+                f"Successfully chunked {len(request.input)} documents "
+                f"into {len(chunks)} chunks"
+            ),
         ),
     )

tests/test_function_utils.py

@@ -28,9 +28,7 @@
 @pytest.fixture
 def sample_entrypoint():
     """Create a temporary entrypoint file with a function."""
-    with tempfile.NamedTemporaryFile(
-        mode="w", suffix=".py", delete=False
-    ) as temp_file:
+    with tempfile.NamedTemporaryFile(mode="w", suffix=".py", delete=False) as temp_file:
         entrypoint_content = textwrap.dedent(
             """
             from typing import List
@@ -62,9 +60,7 @@ def function(request: SampleRequest) -> SampleResponse:
 @pytest.fixture
 def entrypoint_no_annotations():
     """Create an entrypoint with no type annotations."""
-    with tempfile.NamedTemporaryFile(
-        mode="w", suffix=".py", delete=False
-    ) as temp_file:
+    with tempfile.NamedTemporaryFile(mode="w", suffix=".py", delete=False) as temp_file:
         entrypoint_content = textwrap.dedent(
             """
             def function(request):
@@ -117,6 +113,7 @@ def test_inspect_function_types_static(sample_entrypoint, entrypoint_no_annotati
     assert req_name is None
     assert resp_name is None
 
+
 def test_inspect_function_types(sample_entrypoint):
     """Test dynamic inspection of function types."""
     req_name, resp_name = function_utils.inspect_function_types(sample_entrypoint)
@@ -139,7 +136,7 @@ def test_get_request_type(sample_entrypoint, entrypoint_no_annotations):
 
 
 def test_generate_test_json():
-    """Test generating test.json file from entrypoint with simple and complex nested types."""
+    """Test generating test.json with simple and complex nested types."""
     temp_dir = tempfile.mkdtemp()
     models_file = os.path.join(temp_dir, "test_models.py")
 
@@ -235,4 +232,4 @@ def function(request: ComplexRequest):
         if temp_dir in sys.path:
             sys.path.remove(temp_dir)
         if os.path.exists(temp_dir):
-            shutil.rmtree(temp_dir)
+            shutil.rmtree(temp_dir)

tests/test_sf_cli_contract.py

@@ -188,6 +188,7 @@ def test_accepts_network_flag(
         result = runner.invoke(deploy, [*self._BASE_ARGS, "--network", "custom"])
         assert result.exit_code != 2, result.output
 
+
 class TestRunArgContract:
     """
     SF CLI spawn: