Revert "debug"

This reverts commit e61d445.

Author: KRRT7

codeflash/discovery/discover_unit_tests.py

@@ -1,9 +1,7 @@
 # ruff: noqa: SLF001
 from __future__ import annotations
 
-import concurrent.futures
 import hashlib
-import multiprocessing
 import os
 import pickle
 import re
@@ -25,8 +23,6 @@
 from codeflash.models.models import CodePosition, FunctionCalledInTest, TestsInFile, TestType
 
 if TYPE_CHECKING:
-    from multiprocessing.synchronize import Lock
-
     from codeflash.verification.verification_utils import TestConfig
 
 
@@ -288,33 +284,23 @@ def discover_parameters_unittest(function_name: str) -> tuple[bool, str, str | N
 
 
 def _process_single_test_file(
-    test_file: Path,
-    functions: list[TestsInFile],
-    project_root_path: Path,
-    test_framework: str,
-    jedi_lock: Optional[Lock] = None,
+    test_file: Path, functions: list[TestsInFile], project_root_path: Path, test_framework: str
 ) -> tuple[Path, set]:
     import jedi
 
     local_function_to_test_map = set()
 
     try:
-        if jedi_lock is not None:
-            jedi_lock.acquire()
-        try:
-            jedi_project = jedi.Project(path=project_root_path)
-            script = jedi.Script(path=test_file, project=jedi_project)
-            test_functions = set()
-
-            all_names = script.get_names(all_scopes=True, references=True)
-            all_defs = script.get_names(all_scopes=True, definitions=True)
-            all_names_top = script.get_names(all_scopes=True)
-
-            top_level_functions = {name.name: name for name in all_names_top if name.type == "function"}
-            top_level_classes = {name.name: name for name in all_names_top if name.type == "class"}
-        finally:
-            if jedi_lock is not None:
-                jedi_lock.release()
+        jedi_project = jedi.Project(path=project_root_path)
+        script = jedi.Script(path=test_file, project=jedi_project)
+        test_functions = set()
+
+        all_names = script.get_names(all_scopes=True, references=True)
+        all_defs = script.get_names(all_scopes=True, definitions=True)
+        all_names_top = script.get_names(all_scopes=True)
+
+        top_level_functions = {name.name: name for name in all_names_top if name.type == "function"}
+        top_level_classes = {name.name: name for name in all_names_top if name.type == "class"}
     except Exception as e:
         logger.debug(f"Failed to get jedi script for {test_file}: {e}")
         return test_file, local_function_to_test_map
@@ -393,13 +379,7 @@ def _process_single_test_file(
             continue
 
         try:
-            if jedi_lock is not None:
-                jedi_lock.acquire()
-            try:
-                definition = name.goto(follow_imports=True, follow_builtin_imports=False)
-            finally:
-                if jedi_lock is not None:
-                    jedi_lock.release()
+            definition = name.goto(follow_imports=True, follow_builtin_imports=False)
         except Exception as e:
             logger.debug(str(e))
             continue
@@ -453,25 +433,17 @@ def process_test_files(
     test_framework = cfg.test_framework
 
     function_to_test_map = defaultdict(set)
-    jedi_lock = multiprocessing.Manager().Lock()
-
-    with (
-        test_files_progress_bar(total=len(file_to_test_map), description="Processing test files") as (
-            progress,
-            task_id,
-        ),
-        concurrent.futures.ProcessPoolExecutor() as executor,
+
+    with test_files_progress_bar(total=len(file_to_test_map), description="Processing test files") as (
+        progress,
+        task_id,
     ):
-        futures = {
-            executor.submit(
-                _process_single_test_file, test_file, functions, project_root_path, test_framework, jedi_lock
-            ): (test_file, functions)
-            for test_file, functions in file_to_test_map.items()
-        }
-        for future in concurrent.futures.as_completed(futures):
-            _, local_results = future.result()
+        for test_file, functions in file_to_test_map.items():
+            _, local_results = _process_single_test_file(test_file, functions, project_root_path, test_framework)
+
             for qualified_name, function_called_in_test in local_results:
                 function_to_test_map[qualified_name].add(function_called_in_test)
+
             progress.advance(task_id)
 
     function_to_tests_dict = {function: list(tests) for function, tests in function_to_test_map.items()}