diff --git a/src/sagemaker/local/image.py b/src/sagemaker/local/image.py index 32437a59c3..5ba029eb1f 100644 --- a/src/sagemaker/local/image.py +++ b/src/sagemaker/local/image.py @@ -227,6 +227,17 @@ def process( if _ecr_login_if_needed(self.sagemaker_session.boto_session, self.image): _pull_image(self.image) + + # Remove sagemaker-local network, compose_command will fail if an existing network is still around + remove_sagemaker_local = ["docker", "network", "rm", STUDIO_HOST_NAME] + remove_process = subprocess.Popen( + remove_sagemaker_local, stdout=subprocess.PIPE, stderr=subprocess.STDOUT + ) + try: + _stream_output(remove_process) + except RuntimeError: + # Continue even if rm sagemaker-local fails + pass compose_command = self._compose() process = subprocess.Popen(