@@ -1,8 +1,9 @@
# this code is in active development, interfaces may change
import os
import torch
-from typing import List, Optional, Union, Tuple
+from typing import List, Optional, Tuple, Union
+import torch
import hivemind
import torch.nn as nn