Update discover_unit_tests.py

Author: KRRT7

codeflash/discovery/discover_unit_tests.py

@@ -315,10 +315,8 @@ def discover_unit_tests(
     # Extract all functions to optimize for import filtering
     functions_to_optimize = None
     if file_to_funcs_to_optimize:
-        functions_to_optimize = [
-            func for funcs_list in file_to_funcs_to_optimize.values() for func in funcs_list
-        ]
-    
+        functions_to_optimize = [func for funcs_list in file_to_funcs_to_optimize.values() for func in funcs_list]
+
     return strategy(cfg, discover_only_these_tests, functions_to_optimize)