import policy

tests/test_remote_sequential.py

@@ -4,7 +4,7 @@ from hivemind import DHT, get_logger, use_hivemind_log_handler
 from test_utils import *
 
 from petals.bloom.from_pretrained import load_pretrained_block
-from petals.client import RemoteSequential
+from petals.client import NoSpendingPolicy, RemoteSequential
 from petals.client.remote_model import DistributedBloomConfig
 
 use_hivemind_log_handler("in_root_logger")