From 779c021d78983dc7d6a2bfe3babf1a1f01fbe75b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 5 Sep 2023 15:33:31 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(anyio.py):=20change=20return?= =?UTF-8?q?=20type=20annotation=20of=20result()=20method=20from=20'any'=20?= =?UTF-8?q?to=20'Any'=20for=20better=20type=20hinting=20=F0=9F=90=9B=20fix?= =?UTF-8?q?(anyio.py):=20change=20parameter=20type=20annotation=20of=20get?= =?UTF-8?q?=5Ftask()=20method=20from=20'int'=20to=20'str'=20to=20match=20t?= =?UTF-8?q?he=20actual=20type=20of=20task=5Fid?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/task/backends/anyio.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/services/task/backends/anyio.py b/src/backend/langflow/services/task/backends/anyio.py index 68c449d4e..20691212d 100644 --- a/src/backend/langflow/services/task/backends/anyio.py +++ b/src/backend/langflow/services/task/backends/anyio.py @@ -17,7 +17,7 @@ class AnyIOTaskResult: return self._status @property - def result(self) -> any: + def result(self) -> Any: return self._result def ready(self) -> bool: @@ -41,10 +41,10 @@ class AnyIOBackend(TaskBackend): ) -> Tuple[str, AnyIOTaskResult]: # sourcery skip: remove-unnecessary-cast async with anyio.create_task_group() as tg: task_result = AnyIOTaskResult(tg) - await tg.spawn(task_result.run, task_func, *args, **kwargs) + tg.start_soon(task_result.run, task_func, *args, **kwargs) task_id = str(id(task_result)) self.tasks[task_id] = task_result return task_id, task_result - def get_task(self, task_id: int) -> AnyIOTaskResult: + def get_task(self, task_id: str) -> Any: return self.tasks.get(task_id)