소스 검색

Fix annotations

Pavel Samygin 3 년 전
부모
커밋
98f2ac1959
1개의 변경된 파일3개의 추가작업 그리고 1개의 파일을 삭제
  1. 3 1
      hivemind/moe/client/expert.py

+ 3 - 1
hivemind/moe/client/expert.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 from concurrent.futures import Future
 from dataclasses import dataclass
@@ -168,7 +170,7 @@ class RemoteExpertWorker:
     @classmethod
     def spawn_experts_bulk_future(
         cls, infos: Future[Sequence[Sequence[Optional[RemoteExpertInfo]]]], dht: DHT
-    ) -> MPFuture[List[List[Optional[RemoteExpert]]]]:
+    ) -> Future[List[List[Optional[RemoteExpert]]]]:
         async def _unpack():
             return cls.spawn_experts_bulk(await infos, dht)