minor fixes

aseembits93 · aseembits93 · commit 4af844257031 · 2025-03-31T11:08:02.000-07:00
diff --git a/codeflash/verification/parse_line_profile_test_output.py b/codeflash/verification/parse_line_profile_test_output.py
@@ -76,7 +76,7 @@ def parse_line_profile_results(line_profiler_output_file: Path | None) -> dict:
     line_profiler_output_file = line_profiler_output_file.with_suffix(".lprof")
     stats_dict = {}
     if not line_profiler_output_file.exists():
-        return {'timings':{},'unit':0}
+        return {'timings':{},'unit':0, 'str_out':''}, None
     else:
         with open(line_profiler_output_file,'rb') as f:
             stats = pickle.load(f)
diff --git a/tests/test_instrument_tests.py b/tests/test_instrument_tests.py
@@ -21,7 +21,7 @@
     TestFiles,
     TestingMode,
     TestsInFile,
-    TestType,
+    TestType, CodeOptimizationContext,
 )
 from codeflash.optimization.function_optimizer import FunctionOptimizer
 from codeflash.verification.verification_utils import TestConfig

Original file line number	Diff line number	Diff line change
`@@ -21,7 +21,7 @@`
`21`	`21`	`TestFiles,`
`22`	`22`	`TestingMode,`
`23`	`23`	`TestsInFile,`
`24`		`- TestType,`
	`24`	`+ TestType, CodeOptimizationContext,`
`25`	`25`	`)`
`26`	`26`	`from codeflash.optimization.function_optimizer import FunctionOptimizer`
`27`	`27`	`from codeflash.verification.verification_utils import TestConfig`