diff --git a/shrinkwrap/utils/runtime.py b/shrinkwrap/utils/runtime.py index 9b564c021515c96c3afd8a571876a3751884a19d..52be6942a9c14c1f2edfbe43c5bca967d2d2f9fd 100644 --- a/shrinkwrap/utils/runtime.py +++ b/shrinkwrap/utils/runtime.py @@ -131,17 +131,20 @@ print(ip) if res.returncode == 0: return res.stdout.strip() return '127.0.0.1' - + def __enter__(self): global _instance assert _instance is None _instance = self return self - + def __exit__(self, exc_type, exc_val, exc_tb): global _instance assert _instance == self _instance = None + if self._rt: + self._rt.cleanup() + self._rt = None def get():