From 1f46eb1358f41e1b2eec18ce722b9d95304445bd Mon Sep 17 00:00:00 2001
From: bxdd <bxddream@gmail.com>
Date: Sun, 15 Oct 2023 17:14:10 +0800
Subject: [PATCH] [MNT] update model container remove env method

---
 learnware/client/container.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/learnware/client/container.py b/learnware/client/container.py
index 78b43e1..bdf238c 100644
--- a/learnware/client/container.py
+++ b/learnware/client/container.py
@@ -61,8 +61,13 @@ class ModelEnvContainer(BaseModel):
 
     def remove_env(self):
         if self.conda_env is not None:
+            conda_env = self.conda_env
             self.conda_env = None
-            remove_enviroment(self.conda_env)
+            try:
+                remove_enviroment(self.conda_env)
+            except Exception as err:
+                self.conda_env = conda_env
+                raise err
 
     def run_model_with_script(self, method, **kargs):
         with tempfile.TemporaryDirectory(prefix="learnware_") as tempdir: