formating

Author: malavhs

tests/integ/sagemaker/jumpstart/conftest.py

@@ -30,7 +30,6 @@
     HubContentType,
 )
 
-from sagemaker.jumpstart.constants import JUMPSTART_LOGGER
 
 from tests.integ.sagemaker.jumpstart.utils import (
     get_test_artifact_bucket,
@@ -46,14 +45,15 @@ def _setup():
     test_suit_id = get_test_suite_id()
     test_hub_name = f"{HUB_NAME_PREFIX}{test_suit_id}"
     test_hub_description = "PySDK Integ Test Private Hub"
+
     os.environ.update({ENV_VAR_JUMPSTART_SDK_TEST_SUITE_ID: test_suit_id})
     os.environ.update({ENV_VAR_JUMPSTART_SDK_TEST_HUB_NAME: test_hub_name})
+
+    # Create a private hub to use for the test session
     hub = Hub(
         hub_name=os.environ[ENV_VAR_JUMPSTART_SDK_TEST_HUB_NAME], sagemaker_session=get_sm_session()
     )
     hub.create(description=test_hub_description)
-    describe_hub_response = hub.describe()
-    JUMPSTART_LOGGER.info(f"Describe Hub {describe_hub_response}")
 
 
 def _teardown():
@@ -62,7 +62,6 @@ def _teardown():
     test_cache_bucket = get_test_artifact_bucket()
 
     test_suite_id = os.environ[ENV_VAR_JUMPSTART_SDK_TEST_SUITE_ID]
-    test_hub_name = os.environ[ENV_VAR_JUMPSTART_SDK_TEST_HUB_NAME]
 
     boto3_session = boto3.Session(region_name=JUMPSTART_DEFAULT_REGION_NAME)
 
@@ -146,7 +145,6 @@ def _delete_hubs(sagemaker_session):
         if hub["HubName"] != SM_JUMPSTART_PUBLIC_HUB_NAME:
             # delete all hub contents first
             _delete_hub_contents(sagemaker_session, hub["HubName"])
-            JUMPSTART_LOGGER.info(f"Deleting {hub['HubName']}")
             sagemaker_session.delete_hub(hub["HubName"])
 
 
@@ -155,7 +153,6 @@ def _delete_hub_contents(sagemaker_session, test_hub_name):
     list_hub_content_response = sagemaker_session.list_hub_contents(
         hub_name=test_hub_name, hub_content_type=HubContentType.MODEL_REFERENCE.value
     )
-    JUMPSTART_LOGGER.info(f"Listing HubContents {list_hub_content_response}")
 
     # delete hub_contents for the given hub
     for models in list_hub_content_response["HubContentSummaries"]:

tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py

@@ -56,9 +56,6 @@
 
 def test_non_prepacked_jumpstart_model(setup):
 
-    JUMPSTART_LOGGER.info("starting test")
-    JUMPSTART_LOGGER.info(f"get identity {get_sm_session().get_caller_identity_arn()}")
-
     model_id = "catboost-classification-model"
 
     model = JumpStartModel(

tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py

@@ -12,12 +12,8 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
-import io
 import os
-import sys
 import time
-from unittest import mock
-import logging
 
 import pytest
 from sagemaker.enums import EndpointType

tests/integ/sagemaker/jumpstart/private_hub/test_hub.py

@@ -1,8 +1,5 @@
 import pytest
-import os
-from unittest.mock import MagicMock, patch
 from sagemaker.jumpstart.hub.hub import Hub
-from sagemaker.jumpstart.constants import JUMPSTART_LOGGER
 
 from tests.integ.sagemaker.jumpstart.utils import (
     get_sm_session,

tests/integ/sagemaker/jumpstart/private_hub/test_hub_content.py

@@ -1,4 +1,3 @@
-import pytest
 import os
 from sagemaker.jumpstart.hub.hub import Hub
 
@@ -10,7 +9,6 @@
 from tests.integ.sagemaker.jumpstart.constants import (
     ENV_VAR_JUMPSTART_SDK_TEST_HUB_NAME,
 )
-import tests
 
 
 def test_hub_model_reference(setup):