File tree Expand file tree Collapse file tree 1 file changed +0
-24
lines changed
tests/integ/sagemaker/jumpstart/model Expand file tree Collapse file tree 1 file changed +0
-24
lines changed Original file line number Diff line number Diff line change @@ -419,30 +419,6 @@ def test_jumpstart_session_with_config_name():
419419
420420
421421# Currently JumpStartModel does not pull from HubService for the Public Hub.
422- # def test_bedrock_store_model_tags_from_hub_service(setup):
423-
424- # model_id = "huggingface-llm-gemma-2b-instruct"
425-
426- # model = JumpStartModel(
427- # model_id=model_id,
428- # hub_name="SageMakerPublicHub",
429- # role=get_sm_session().get_caller_identity_arn(),
430- # sagemaker_session=get_sm_session(),
431- # )
432-
433- # predictor = model.deploy(
434- # tags=[{"Key": JUMPSTART_TAG, "Value": os.environ[ENV_VAR_JUMPSTART_SDK_TEST_SUITE_ID]}],
435- # accept_eula=True,
436- # )
437-
438- # endpoint_arn = (
439- # f"arn:aws:sagemaker:{get_sm_session().boto_region_name}:"
440- # f"{get_sm_session().account_id()}:endpoint/{predictor.endpoint_name}"
441- # )
442- # tags = get_sm_session().list_tags(endpoint_arn)
443- # expected_tag = {"Key": "sagemaker-sdk:bedrock", "Value": "compatible"}
444- # assert expected_tag in tags
445-
446422# def test_proprietary_from_hub_service(setup):
447423
448424# model_id = "upstage-solar-mini-chat"
You can’t perform that action at this time.
0 commit comments