fix: account for possible race condition when creating /opt/ml/code (#231)

benieric · web-flow · commit 87f69b714a03 · 2025-01-31T15:30:56.000-08:00
diff --git a/src/sagemaker_training/environment.py b/src/sagemaker_training/environment.py
@@ -183,7 +183,10 @@ def _create_training_directories():
 def _create_code_dir():  # type: () -> None
     """Create /opt/ml/code when the module is imported."""
     if not os.path.exists(code_dir):
-        os.makedirs(code_dir)
+        try:
+            os.makedirs(code_dir)
+        except FileExistsError:
+            pass  # Another process created the directory
 
 
 _create_code_dir()
