feat: add logging

vndee · Jul 6, 2024 · 84c3f43 · 84c3f43
1 parent 98b07ce
commit 84c3f43
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/llm_sandbox/session.py b/llm_sandbox/session.py
@@ -104,6 +104,11 @@ def close(self):
                     self.image.remove(force=True)
                 else:
                     raise ValueError("Invalid image type")
+            else:
+                if self.verbose:
+                    print(
+                        f"Image {self.image.tags[-1]} is in use by other containers. Skipping removal."
+                    )
 
     def run(self, code: str, libraries: List = []):
         raise NotImplementedError