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)