fix linting

Author: KRRT7

codeflash/optimization/function_optimizer.py

@@ -1778,9 +1778,8 @@ def run_and_parse_tests(
             )
             # Return the test results for async throughput calculation
             return results, coverage_results, results if isinstance(results, TestResults) else None
-        else:
-            results, coverage_results = parse_line_profile_results(line_profiler_output_file=line_profiler_output_file)
-            return results, coverage_results, None
+        results, coverage_results = parse_line_profile_results(line_profiler_output_file=line_profiler_output_file)
+        return results, coverage_results, None
 
     def submit_test_generation_tasks(
         self,

codeflash/verification/parse_test_output.py

@@ -83,7 +83,8 @@ def calculate_async_throughput_from_stdout(stdout: str, async_function_names: se
 
 
 def calculate_function_throughput_from_stdout(stdout: str, function_name: str) -> int:
-    """A completed execution is defined as having both a start tag and matching end tag:
+    """Calculate function throughput from stdout. A completed execution is defined as having both a start tag and matching end tag.
+
     Start: !$######test_module:test_function:function_name:loop_index:iteration_id######$!
     End:   !######test_module:test_function:function_name:loop_index:iteration_id######!
     """

codeflash/verification/pytest_plugin.py

@@ -468,7 +468,7 @@ def pytest_runtest_setup(self, item: pytest.Item) -> None:
         os.environ["CODEFLASH_TEST_FUNCTION"] = test_function_name
 
     @pytest.hookimpl(trylast=True)
-    def pytest_runtest_teardown(self, item: pytest.Item) -> None:
+    def pytest_runtest_teardown(self, item: pytest.Item) -> None:  # noqa: ARG002
         """Clean up test context environment variables after each test."""
         for var in ["CODEFLASH_TEST_MODULE", "CODEFLASH_TEST_CLASS", "CODEFLASH_TEST_FUNCTION"]:
             os.environ.pop(var, None)