Bug fix: convert functions correctly (#133)

exalearn · Apr 2, 2024 · b837436 · b837436
1 parent 0f9f3cf
commit b837436
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/colmena/task_server/parsl.py b/colmena/task_server/parsl.py
@@ -353,7 +353,7 @@ def __init__(self, methods: List[Union[Callable, Tuple[Callable, Dict]]],
                 logger.info(f'Using default executors for {function.__name__}: {default_executors}')
 
             # Convert the function to a Colmena task
-            function = convert_to_colmena_method(method)
+            function = convert_to_colmena_method(function)
             name = function.name
 
             # If the function is not an executable, submit it as a single task

diff --git a/colmena/task_server/tests/test_parsl.py b/colmena/task_server/tests/test_parsl.py
@@ -59,7 +59,7 @@ def store(tmpdir):
 @fixture(autouse=True)
 def server_and_queue(config, store) -> Tuple[ParslTaskServer, ColmenaQueues]:
     queues = PipeQueues(proxystore_name='store', proxystore_threshold=5000, serialization_method='pickle')
-    server = ParslTaskServer([f, capitalize, bad_task, EchoTask(), FakeMPITask(), count_nodes], queues, config)
+    server = ParslTaskServer([(f, {'executors': 'all'}), capitalize, bad_task, EchoTask(), FakeMPITask(), count_nodes], queues, config)
     yield server, queues
     if server.is_alive():
         queues.send_kill_signal()