From f482a98e3a59d6c9d0b5f0a6772f21cb9ce2c6f9 Mon Sep 17 00:00:00 2001 From: justheuristic Date: Thu, 1 Sep 2022 04:04:43 +0300 Subject: [PATCH] fix imports --- tests/test_block_exact_match.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_block_exact_match.py b/tests/test_block_exact_match.py index e59a6dd..fad84ae 100644 --- a/tests/test_block_exact_match.py +++ b/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