fixing tests

AgnostiqHQ · Oct 19, 2023 · a02425a · a02425a
1 parent 40534cf
commit a02425a
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 6 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -35,11 +35,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 
 - Documentation and test cases for database triggers.
+- New Runner and executor API to bypass server-side memory when running tasks.
 
 ### Docs
 
 - Added federated learning showcase code
-- Updated tutorial for redispatching workflows with Streamlit 
+- Updated tutorial for redispatching workflows with Streamlit
 
 ### Tests
 
@@ -152,7 +153,6 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Fix for double locking file in configurations.
 - Introduced new data access layer
 - Introduced Shutil file transfer strategy for local file transfers
-- New Runner and executor API to bypass server-side memory when running tasks.
 
 ### Docs
 

diff --git a/covalent_dispatcher/_core/runner.py b/covalent_dispatcher/_core/runner.py
@@ -15,7 +15,7 @@
 # limitations under the License.
 
 """
-Defines the core functionality of the runner
+Defines the core functionality of the legacy runner
 """
 
 import asyncio

diff --git a/covalent_dispatcher/_core/runner_ng.py b/covalent_dispatcher/_core/runner_ng.py
@@ -1,4 +1,4 @@
-# Copyright 2021 Agnostiq Inc.
+# Copyright 2023 Agnostiq Inc.
 #
 # This file is part of Covalent.
 #
@@ -15,7 +15,7 @@
 # limitations under the License.
 
 """
-Defines the core functionality of the runner
+Defines the core functionality of the new improved runner
 """
 
 import asyncio
@@ -72,6 +72,7 @@ async def _submit_abstract_task_group(
 ) -> None:
     # Task sequence of the form {"function_id": task_id, "args_ids":
     # [node_ids], "kwargs_ids": {key: node_id}}
+
     task_ids = [task["function_id"] for task in task_seq]
     task_specs = []
     task_group_metadata = {

diff --git a/tests/covalent_tests/executor/base_test.py b/tests/covalent_tests/executor/base_test.py
@@ -26,9 +26,10 @@
 from covalent import DepsCall, TransportableObject
 from covalent._results_manager import Result
 from covalent._shared_files.exceptions import TaskCancelledError, TaskRuntimeError
-from covalent.executor import BaseExecutor, wrapper_fn
+from covalent.executor import BaseExecutor
 from covalent.executor.base import AsyncBaseExecutor
 from covalent.executor.utils.enums import Signals
+from covalent.executor.utils.wrappers import wrapper_fn
 
 
 class MockExecutor(BaseExecutor):