瀏覽代碼

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)