diff --git a/learnware/client/container.py b/learnware/client/container.py index ff6f778..5d164e4 100644 --- a/learnware/client/container.py +++ b/learnware/client/container.py @@ -78,3 +78,6 @@ class ModelEnvContainer(BaseModel): def finetune(self, X, y): self.run_model_with_script("finetune", X=X, y=y) + + def __del__(self): + system_execute(f"conda env remove -n {self.conda_env}") \ No newline at end of file diff --git a/tests/test_learnware_client/test_reuse.py b/tests/test_learnware_client/test_reuse.py index 1d381c9..6f3ff21 100644 --- a/tests/test_learnware_client/test_reuse.py +++ b/tests/test_learnware_client/test_reuse.py @@ -1,8 +1,5 @@ -import os - -from learnware.learnware import get_learnware_from_dirpath -from learnware.client.learnware_client import LearnwareClient - +from learnware.learnware import get_learnware_from_dirpath, Learnware +from learnware.client.container import ModelEnvContainer if __name__ == "__main__": semantic_specification = dict() @@ -12,7 +9,11 @@ if __name__ == "__main__": semantic_specification["Scenario"] = {"Type": "Tag", "Values": "Financial"} semantic_specification["Name"] = {"Type": "String", "Values": "test"} semantic_specification["Description"] = {"Type": "String", "Values": "test"} - - zip_path = "/home/bixd/workspace/learnware/Learnware/tests/test_workflow/learnware_pool/svm_0.zip" - - learnware = get_learnware_from_dirpath("test_id", semantic_specification, zip_path) + + zip_path = '/home/bixd/workspace/learnware/Learnware/tests/test_workflow/learnware_pool/svm_0.zip' + + learnware = get_learnware_from_dirpath('test_id', semantic_specification, zip_path) + + env_leanware = Learnware(id=learnware.id, model=ModelEnvContainer(learnware.get_model(), zip_path), specification=learnware.get_specification()) + + \ No newline at end of file