fix: add deprecation warning

Author: mariumof

src/sagemaker/debugger/profiler_config.py

@@ -13,11 +13,14 @@
 """Configuration for collecting system and framework metrics in SageMaker training jobs."""
 from __future__ import absolute_import
 
+import logging
 from typing import Optional, Union
 
 from sagemaker.debugger.framework_profile import FrameworkProfile
 from sagemaker.workflow.entities import PipelineVariable
 
+logger = logging.getLogger(__name__)
+
 
 class ProfilerConfig(object):
     """Configuration for collecting system and framework metrics of SageMaker training jobs.
@@ -81,6 +84,11 @@ class and SageMaker Framework estimators.
         self.framework_profile_params = framework_profile_params
         self.disable_profiler = disable_profiler
 
+        if self.framework_profile_params is not None:
+            logger.warning(
+                "Framework profiling will be deprecated from tensorflow 2.12 and pytorch 2.0"
+            )
+
     def _to_request_dict(self):
         """Generate a request dictionary using the parameters provided when initializing the object.
 

tests/unit/test_profiler_config.py

@@ -708,7 +708,6 @@ def test_create_pytorch_estimator_w_image_with_framework_profile(
     )
 
 
-"""
 def test_create_tf_estimator_with_framework_profile(
     sagemaker_session,
     tensorflow_inference_version,
@@ -720,13 +719,16 @@ def test_create_tf_estimator_with_framework_profile(
     tf = TensorFlow(
         entry_point=SCRIPT_PATH,
         role=ROLE,
-        framework_version=tensorflow_inference_version,
-        py_version=tensorflow_inference_py_version,
+        framework_version="2.8",
+        py_version="py39",
         sagemaker_session=sagemaker_session,
         instance_count=INSTANCE_COUNT,
         instance_type=INSTANCE_TYPE,
         profiler_config=profiler_config,
     )
+
+
+"""
 ... ValueError: TF 1.5 supports only legacy mode. 
 Please supply the image URI directly with 
 'image_uri=520713654638.dkr.ecr.us-west-2.amazonaws.com/sagemaker-tensorflow:1.5-cpu-py2'