Merge pull request #44 from delta-mpc/zk

Zk
delta-mpc · Sep 30, 2022 · 0c75d89 · 0c75d89
2 parents c1b23c6 + b255f78
commit 0c75d89
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 9 deletions.
diff --git a/delta_node/runner/hlr/manager.py b/delta_node/runner/hlr/manager.py
@@ -124,6 +124,9 @@ async def finish(self, success: bool):
         if success and self.task_entity.enable_verify:
             await self.verify()
 
+        await pool.run_in_io(self.ctx.clear)
+        await pool.run_in_io(self.client.close)
+
     async def verify(self):
         # upload data commitment
         data = self.ctx.get_data()
@@ -215,8 +218,5 @@ async def run(self):
                 await self.finish(False)
                 raise
 
-        await pool.run_in_io(self.ctx.clear)
-        await pool.run_in_io(self.client.close)
-
     async def recv_event(self, event: entity.TaskEvent):
         await self.event_box.recv_event(event)
diff --git a/delta_node/runner/horizontal/manager.py b/delta_node/runner/horizontal/manager.py
@@ -121,6 +121,9 @@ async def finish(self, success: bool):
         if self.running_fut is not None:
             self.running_fut.cancel()
 
+        await pool.run_in_io(self.ctx.clear)
+        await pool.run_in_io(self.client.close)
+
     async def run(self):
         while True:
             try:
@@ -135,8 +138,5 @@ async def run(self):
                 await self.finish(False)
                 raise
 
-        await pool.run_in_io(self.ctx.clear)
-        await pool.run_in_io(self.client.close)
-
     async def recv_event(self, event: entity.TaskEvent):
         await self.event_box.recv_event(event)
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 aiosqlite==0.17.0
 async_lru==1.0.2
 cryptography==3.4.7
-delta-task==0.7.0rc1
+delta-task==0.8.0
 fastapi==0.70.1
 grpclib==0.4.2
 httpx==0.23.0

diff --git a/setup.py b/setup.py
@@ -30,7 +30,7 @@ def run_tests(self):
 
 setup(
     name="delta_node",
-    version="0.7.0rc4",
+    version="0.8.0",
     packages=find_packages(),
     package_data={"delta_node": ["dataset/examples/*.csv"]},
     include_package_data=True,
@@ -39,7 +39,7 @@ def run_tests(self):
         "aiosqlite==0.17.0",
         "async_lru==1.0.2",
         "cryptography==3.4.7",
-        "delta-task==0.7.0rc1",
+        "delta-task==0.8.0",
         "fastapi==0.70.1",
         "grpclib==0.4.2",
         "httpx==0.23.0",