Small bug fix to print debug messages for inference logger (PySDK) (#246)

* Draft of inference logger bug fix

* Draft fix of inference logger for SDK

* Revert adding --debug flag

* Add debug parameter to failing unit tests

* Fix create_from_dict to not have hardcoded debug flag

Author: aviruthen

src/sagemaker/hyperpod/inference/hp_endpoint.py

@@ -54,6 +54,7 @@ def create(
             kind=INFERENCE_ENDPOINT_CONFIG_KIND,
             namespace=namespace,
             spec=spec,
+            debug=debug,
         )
 
         self.metadata = Metadata(
@@ -71,9 +72,10 @@ def create_from_dict(
         input: Dict,
         name: str = None,
         namespace: str = None,
+        debug=False
     ) -> None:
         logger = self.get_logger()
-        logger = setup_logging(logger)
+        logger = setup_logging(logger, debug)
 
         spec = _HPEndpoint.model_validate(input, by_name=True)
 
@@ -93,6 +95,7 @@ def create_from_dict(
             kind=INFERENCE_ENDPOINT_CONFIG_KIND,
             namespace=namespace,
             spec=spec,
+            debug=debug,
         )
 
         self.metadata = Metadata(

src/sagemaker/hyperpod/inference/hp_endpoint_base.py

@@ -63,6 +63,7 @@ def call_create_api(
         kind: str,
         namespace: str,
         spec: Union[_HPJumpStartEndpoint, _HPEndpoint],
+        debug: bool = False,
     ):
         """Create an inference endpoint using Kubernetes API.
 
@@ -104,7 +105,7 @@ def call_create_api(
         cls.verify_kube_config()
 
         logger = cls.get_logger()
-        logger = setup_logging(logger)
+        logger = setup_logging(logger, debug)
 
         custom_api = client.CustomObjectsApi()
 

src/sagemaker/hyperpod/inference/hp_jumpstart_endpoint.py

@@ -59,6 +59,7 @@ def create(
             kind=JUMPSTART_MODEL_KIND,
             namespace=namespace,
             spec=spec,
+            debug=debug,
         )
 
         self.metadata = Metadata(
@@ -76,9 +77,10 @@ def create_from_dict(
         input: Dict,
         name: str = None,
         namespace: str = None,
+        debug = False
     ) -> None:
         logger = self.get_logger()
-        logger = setup_logging(logger)
+        logger = setup_logging(logger, debug)
 
         spec = _HPJumpStartEndpoint.model_validate(input, by_name=True)
 
@@ -102,6 +104,7 @@ def create_from_dict(
             kind=JUMPSTART_MODEL_KIND,
             namespace=namespace,
             spec=spec,
+            debug=debug,
         )
 
         self.metadata = Metadata(

test/unit_tests/inference/test_hp_endpoint.py

@@ -104,6 +104,7 @@ def test_create(self, mock_create_api, mock_validate_instance_type):
             kind=INFERENCE_ENDPOINT_CONFIG_KIND,
             namespace="test-ns",
             spec=unittest.mock.ANY,
+            debug=False,
         )
         self.assertEqual(self.endpoint.metadata.name, "test-name")
 
@@ -115,7 +116,13 @@ def test_create_from_dict(self, mock_create_api, mock_validate_instance_type):
 
         self.endpoint.create_from_dict(input_dict, namespace="test-ns")
 
-        mock_create_api.assert_called_once()
+        mock_create_api.assert_called_once_with(
+            name=unittest.mock.ANY,
+            kind=INFERENCE_ENDPOINT_CONFIG_KIND,
+            namespace="test-ns",
+            spec=unittest.mock.ANY,
+            debug=False,
+        )
 
     @patch.object(HPEndpoint, "call_get_api")
     def test_refresh(self, mock_get_api):

test/unit_tests/inference/test_hp_jumpstart_endpoint.py

@@ -44,6 +44,7 @@ def test_create(self, mock_create_api, mock_validate_instance_type):
             kind=JUMPSTART_MODEL_KIND,
             namespace="test-ns",
             spec=unittest.mock.ANY,
+            debug=False,
         )
         self.assertEqual(self.endpoint.metadata.name, "test-name")
 
@@ -60,7 +61,13 @@ def test_create_from_dict(self, mock_create_api, mock_validate_instance_type):
             input_dict, name="test-name", namespace="test-ns"
         )
 
-        mock_create_api.assert_called_once()
+        mock_create_api.assert_called_once_with(
+            name="test-name",
+            kind=JUMPSTART_MODEL_KIND,
+            namespace="test-ns",
+            spec=unittest.mock.ANY,
+            debug=False,
+        )
 
     @patch.object(HPJumpStartEndpoint, "call_get_api")
     def test_refresh(self, mock_get_api):