From fd7b078f589bb25e8c439c3214d6aca3d4619803 Mon Sep 17 00:00:00 2001 From: isabella618033 Date: Wed, 19 Apr 2023 15:47:23 +0000 Subject: [PATCH] fix --- tests/unit_tests/bittensor_tests/test_axon.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit_tests/bittensor_tests/test_axon.py b/tests/unit_tests/bittensor_tests/test_axon.py index 2ee510f226..68d8d6e7fe 100644 --- a/tests/unit_tests/bittensor_tests/test_axon.py +++ b/tests/unit_tests/bittensor_tests/test_axon.py @@ -853,12 +853,9 @@ def forward( inputs_x: torch.FloatTensor, synapses , model_output = None): synapses= [ syn.serialize_to_wire_proto() for syn in synapses ], hotkey = axon.wallet.hotkey.ss58_address, ) - start_time = time.time() executor = ThreadPoolExecutor(2) future = executor.submit(axon._forward, (request)) future2 = executor.submit(axon._forward, (request)) - response, code, synapses = future.result() - assert code == bittensor.proto.ReturnCode.Success try: future2.result(timeout = 1) @@ -866,6 +863,9 @@ def forward( inputs_x: torch.FloatTensor, synapses , model_output = None): pass else: raise AssertionError('Expected to Timeout') + + _, code, _ = future.result() + assert code == bittensor.proto.ReturnCode.Success axon.stop()