浏览代码

fix imports

justheuristic 3 年之前
父节点
当前提交
f482a98e3a
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      tests/test_block_exact_match.py

+ 1 - 0
tests/test_block_exact_match.py

@@ -8,6 +8,7 @@ from hivemind import P2PHandlerError
 from test_utils import *
 
 import src
+from src import DistributedBloomConfig
 from src.bloom.from_pretrained import load_pretrained_block
 from src.client.remote_sequential import RemoteTransformerBlock
 from src.data_structures import UID_DELIMITER