fix: reverting some changes

Author: chrstfu

src/sagemaker/jumpstart/enums.py

@@ -82,12 +82,6 @@ class VariableTypes(str, Enum):
     BOOL = "bool"
 
 
-class HubContentCapability(str, Enum):
-    """Enum class for HubContent capabilities."""
-
-    BEDROCK_CONSOLE = "BEDROCK_CONSOLE"
-
-
 class JumpStartTag(str, Enum):
     """Enum class for tag keys to apply to JumpStart models."""
 
@@ -105,8 +99,6 @@ class JumpStartTag(str, Enum):
 
     HUB_CONTENT_ARN = "sagemaker-sdk:hub-content-arn"
 
-    BEDROCK = "sagemaker-sdk:bedrock"
-
 
 class SerializerType(str, Enum):
     """Enum class for serializers associated with JumpStart models."""

src/sagemaker/jumpstart/factory/model.py

@@ -41,7 +41,7 @@
 from sagemaker.model_metrics import ModelMetrics
 from sagemaker.metadata_properties import MetadataProperties
 from sagemaker.drift_check_baselines import DriftCheckBaselines
-from sagemaker.jumpstart.enums import JumpStartScriptScope, JumpStartModelType, HubContentCapability
+from sagemaker.jumpstart.enums import JumpStartScriptScope, JumpStartModelType
 from sagemaker.jumpstart.types import (
     HubContentType,
     JumpStartModelDeployKwargs,
@@ -51,9 +51,7 @@
 )
 from sagemaker.jumpstart.utils import (
     add_hub_content_arn_tags,
-    add_bedrock_store_tags,
     add_jumpstart_model_info_tags,
-    add_bedrock_store_tags,
     get_default_jumpstart_session_with_user_agent_suffix,
     get_neo_content_bucket,
     get_top_ranked_config_name,
@@ -490,10 +488,6 @@ def _add_tags_to_kwargs(kwargs: JumpStartModelDeployKwargs) -> Dict[str, Any]:
             )
         kwargs.tags = add_hub_content_arn_tags(kwargs.tags, hub_content_arn=hub_content_arn)
 
-    if hasattr(kwargs.specs, "capabilities") and kwargs.specs.capabilities is not None:
-        if HubContentCapability.BEDROCK_CONSOLE in kwargs.specs.capabilities:
-            kwargs.tags = add_bedrock_store_tags(kwargs.tags, compatibility="compatible")
-
     return kwargs
 
 

src/sagemaker/jumpstart/hub/utils.py

@@ -119,8 +119,8 @@ def generate_hub_arn_for_init_kwargs(
 
     hub_arn = None
     if hub_name:
-        # if hub_name == constants.JUMPSTART_MODEL_HUB_NAME:
-        #     return None
+        if hub_name == constants.JUMPSTART_MODEL_HUB_NAME:
+            return None
         match = re.match(constants.HUB_ARN_REGEX, hub_name)
         if match:
             hub_arn = hub_name