Gene
2 years ago
1 changed files with 0 additions and 4 deletions
Unified View
Diff Options
+ 0
- 4
tests/test_learnware_client/test_load_docker.py
View File
| @@ -37,8 +37,6 @@ class TestLearnwareLoad(unittest.TestCase): | |||||
| def test_load_multi_learnware_by_id(self): | def test_load_multi_learnware_by_id(self): | ||||
| learnware_list = self.client.load_learnware(learnware_id=self.learnware_ids, runnable_option="docker") | 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") | reuser = AveragingReuser(learnware_list, mode="vote_by_label") | ||||
| input_array = np.random.random(size=(20, 13)) | input_array = np.random.random(size=(20, 13)) | ||||
| print(reuser.predict(input_array)) | print(reuser.predict(input_array)) | ||||
| @@ -46,8 +44,6 @@ class TestLearnwareLoad(unittest.TestCase): | |||||
| for learnware in learnware_list: | for learnware in learnware_list: | ||||
| print(learnware.id, learnware.predict(input_array)) | 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): | def test_load_single_learnware_by_id_pip(self): | ||||
| learnware_id = "00000147" | learnware_id = "00000147" | ||||
| learnware = self.client.load_learnware(learnware_id=learnware_id, runnable_option="docker") | learnware = self.client.load_learnware(learnware_id=learnware_id, runnable_option="docker") | ||||