diff --git a/llm_sandbox/session.py b/llm_sandbox/session.py index 474eee3..a9d5924 100644 --- 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