Parcourir la source

fix: tests is not a package (#73)

justheuristic il y a 5 ans
Parent
commit
5db3d4c946
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      tests/test_utils/run_server.py

+ 1 - 1
tests/test_utils/run_server.py

@@ -6,7 +6,7 @@ import resource
 import torch
 
 import hivemind
-from tests.test_utils.layers import name_to_block, name_to_input
+from test_utils.layers import name_to_block, name_to_input
 
 
 def make_dummy_server(interface='0.0.0.0', port=None, num_experts=1, expert_cls='ffn', hidden_dim=1024,