Browse Source

Fix annotations

Pavel Samygin 3 years ago
parent
commit
98f2ac1959
1 changed files with 3 additions and 1 deletions
  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)