Add -mindepth

Author: mollyheamazon

sagemaker-core/src/sagemaker/core/modules/local_core/local_container.py

@@ -75,7 +75,7 @@ def _rmtree(path, image=None, is_studio=False):
             cmd = ["docker", "run", "--rm"]
             if is_studio:
                 cmd += ["--network", "sagemaker"]
-            cmd += ["-v", f"{path}:/delete", image, "rm", "-rf", "/delete"]
+            cmd += ["-v", f"{path}:/delete", image, "sh", "-c", "find /delete -mindepth 1 -delete"]
             subprocess.run(cmd, check=True, capture_output=True)
             # The mount point directory itself may remain — clean it up
             if os.path.exists(path):

sagemaker-train/src/sagemaker/train/local/local_container.py

@@ -83,7 +83,7 @@ def _rmtree(path, image=None, is_studio=False):
             cmd = ["docker", "run", "--rm"]
             if is_studio:
                 cmd += ["--network", "sagemaker"]
-            cmd += ["-v", f"{path}:/delete", image, "rm", "-rf", "/delete"]
+            cmd += ["-v", f"{path}:/delete", image, "sh", "-c", "find /delete -mindepth 1 -delete"]
             subprocess.run(cmd, check=True, capture_output=True)
             # The mount point directory itself may remain — clean it up
             if os.path.exists(path):

sagemaker-train/tests/unit/train/local/test_local_container.py

@@ -37,7 +37,11 @@ def test_rmtree_permission_error_docker_fallback(self, mock_exists, mock_run, mo
         _rmtree("/tmp/test", IMAGE)
 
         mock_run.assert_called_once_with(
-            ["docker", "run", "--rm", "-v", "/tmp/test:/delete", IMAGE, "rm", "-rf", "/delete"],
+            [
+                "docker", "run", "--rm",
+                "-v", "/tmp/test:/delete", IMAGE,
+                "sh", "-c", "find /delete -mindepth 1 -delete",
+            ],
             check=True,
             capture_output=True,
         )
@@ -55,7 +59,8 @@ def test_rmtree_studio_adds_network(self, mock_exists, mock_run, mock_rmtree):
             [
                 "docker", "run", "--rm",
                 "--network", "sagemaker",
-                "-v", "/tmp/test:/delete", IMAGE, "rm", "-rf", "/delete",
+                "-v", "/tmp/test:/delete", IMAGE,
+                "sh", "-c", "find /delete -mindepth 1 -delete",
             ],
             check=True,
             capture_output=True,