linter

golemfactory · Sep 11, 2019 · 9302310 · 9302310
1 parent edc4ff8
commit 9302310
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/golem/task/taskserver.py b/golem/task/taskserver.py
@@ -1,6 +1,5 @@
 # -*- coding: utf-8 -*-
 import functools
-import asyncio
 import itertools
 import logging
 import os
@@ -307,7 +306,9 @@ def resume(self):
         CoreTask.VERIFICATION_QUEUE.resume()
 
     def quit(self):
-        defer_rtm_quit = deferred_from_future(self.requested_task_manager.quit())
+        defer_rtm_quit = deferred_from_future(
+            self.requested_task_manager.quit()
+        )
         sync_wait(defer_rtm_quit)
         self.task_computer.quit()
 

diff --git a/tests/golem/task/test_rpc.py b/tests/golem/task/test_rpc.py
@@ -1124,7 +1124,7 @@ def test_removing_duplicate_subtasks(self, *_):
 
 class TestGetFragments(ProviderBase, TwistedTestCase):
     @defer.inlineCallbacks
-    def _create_task(self) -> taskbase.Task:
+    def _create_task(self) -> defer.Deferred:
         task = self.client.task_manager.create_task(self.t_dict)
         with mock.patch('os.path.getsize', return_value=123):
             result = yield rpc._prepare_task(self.client, task, force=False)