Merge branch 'master' into feat/inference-plus-training-script

Author: evakravi

doc/amazon_sagemaker_debugger.rst

@@ -4,6 +4,12 @@ Amazon SageMaker Debugger
 #########################
 
 
+.. warning::
+
+  This page is no longer supported for maintenence. The live documentation is at `Debug and Profile Training Jobs Using Amazon SageMaker Debugger <https://docs.aws.amazon.com/sagemaker/latest/dg/train-debugger.html>`_
+  and `Debugger API <https://sagemaker.readthedocs.io/en/stable/api/training/debugger.html>`_.
+
+
 Amazon SageMaker Debugger allows you to detect anomalies while training your machine learning model by emitting relevant data during training, storing the data and then analyzing it.
 
 .. contents::

doc/api/training/debugger.rst

@@ -10,8 +10,13 @@ Configure the Debugger-specific parameters when constructing
 a SageMaker estimator to gain visibility and insights
 into your training job.
 
+.. contents::
+
 .. currentmodule:: sagemaker.debugger
 
+Debugger Rule APIs
+~~~~~~~~~~~~~~~~~~
+
 .. autoclass:: get_rule_container_image_uri
     :show-inheritance:
 
@@ -44,6 +49,9 @@ into your training job.
     :show-inheritance:
     :inherited-members:
 
+Debugger Configuration APIs
+~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
 .. autoclass:: CollectionConfig
     :show-inheritance:
 
@@ -56,6 +64,21 @@ into your training job.
 .. autoclass:: ProfilerConfig
     :show-inheritance:
 
+Debugger Configuration APIs for Framework Profiling (Deprecated)
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. warning::
+
+  SageMaker Debugger deprecates the framework profiling feature starting from TensorFlow 2.11 and PyTorch 2.0. You can still use the feature in the previous versions of the frameworks and SDKs as follows.
+
+  * SageMaker Python SDK <= v2.130.0
+  * PyTorch >= v1.6.0, < v2.0
+  * TensorFlow >= v2.3.1, < v2.11
+
+  With the deprecation, SageMaker Debugger discontinues support for the APIs below this note.
+
+  See also `Amazon SageMaker Debugger Release Notes: March 16, 2023 <https://docs.aws.amazon.com/sagemaker/latest/dg/debugger-release-notes.html#debugger-release-notes-20230315>`_.
+
 .. autoclass:: FrameworkProfile
     :show-inheritance:
 

src/sagemaker/utils.py

@@ -429,7 +429,7 @@ def create_tar_file(source_files, target=None):
 def _tmpdir(suffix="", prefix="tmp", directory=None):
     """Create a temporary directory with a context manager.
 
-    The file is deleted when the context exits.
+    The file is deleted when the context exits, even when there's an exception.
     The prefix, suffix, and dir arguments are the same as for mkstemp().
 
     Args:
@@ -449,8 +449,10 @@ def _tmpdir(suffix="", prefix="tmp", directory=None):
             f"directory does not exist: '{directory}'"
         )
     tmp = tempfile.mkdtemp(suffix=suffix, prefix=prefix, dir=directory)
-    yield tmp
-    shutil.rmtree(tmp)
+    try:
+        yield tmp
+    finally:
+        shutil.rmtree(tmp)
 
 
 def repack_model(