[ROCm] Remove exit on error (#167)

Author: Ruturaj4

build/rocm/ci_build

@@ -53,7 +53,6 @@ def dist_wheels(
         print(f"Image {image} removed successfully.")
     except subprocess.CalledProcessError as e:
         print(f"Failed to remove Docker image {image}: {e}")
-        exit(1)
 
     cmd = [
         "docker",

build/rocm/tools/get_rocm.py

@@ -74,9 +74,8 @@ def install_packages(self, package_specs):
         env = dict(os.environ)
         if self.pkgbin == "apt":
             env["DEBIAN_FRONTEND"] = "noninteractive"
-
-        # Update indexes.
-        subprocess.check_call(["apt-get", "update"])
+            # Update indexes.
+            subprocess.check_call(["apt-get", "update"])
 
         LOG.info("Running %r" % cmd)
         subprocess.check_call(cmd, env=env)