[FIX][execute_futures_dict]

kyegomez · Dec 2, 2023 · e35b253 · e35b253
1 parent 452bf7d
commit e35b253
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/futures.py b/futures.py
@@ -1,7 +1,7 @@
 import concurrent.futures
 import time
 import random
-import swarms.utils.execute_futures
+from swarms.utils.futures import execute_futures_dict
 
 
 def f(x):
@@ -10,5 +10,6 @@ def f(x):
 
 
 with concurrent.futures.ThreadPoolExecutor() as executor:
+    """Create a dictionary of futures."""
     fs_dict = {str(i): executor.submit(f, i) for i in range(10)}
-    print(swarms.utils.execute_futures.execute_futures_dict(fs_dict))
+    print(execute_futures_dict(fs_dict))
diff --git a/swarms/utils/__init__.py b/swarms/utils/__init__.py
@@ -1,5 +1,4 @@
 from swarms.utils.markdown_message import display_markdown_message
-from swarms.utils.execute_futures import execute_futures_dict
 from swarms.utils.code_interpreter import SubprocessCodeInterpreter
 from swarms.utils.parse_code import (
     extract_code_in_backticks_in_string,
@@ -8,7 +7,6 @@
 
 __all__ = [
     "display_markdown_message",
-    "execute_futures_dict",
     "SubprocessCodeInterpreter",
     "extract_code_in_backticks_in_string",
     "pdf_to_text",