diff --git a/tests/test_learnware_client/test_load_docker.py b/tests/test_learnware_client/test_load_docker.py index ac16afd..b64ffed 100644 --- a/tests/test_learnware_client/test_load_docker.py +++ b/tests/test_learnware_client/test_load_docker.py @@ -37,8 +37,6 @@ class TestLearnwareLoad(unittest.TestCase): def test_load_multi_learnware_by_id(self): learnware_list = self.client.load_learnware(learnware_id=self.learnware_ids, runnable_option="docker") - docker_container = learnware_list[0].get_model().docker_container - reuser = AveragingReuser(learnware_list, mode="vote_by_label") input_array = np.random.random(size=(20, 13)) print(reuser.predict(input_array)) @@ -46,8 +44,6 @@ class TestLearnwareLoad(unittest.TestCase): for learnware in learnware_list: print(learnware.id, learnware.predict(input_array)) - learnware_list[0].get_model()._destroy_docker_container(docker_container) - def test_load_single_learnware_by_id_pip(self): learnware_id = "00000147" learnware = self.client.load_learnware(learnware_id=learnware_id, runnable_option="docker")