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():
419
419
420
420
421
421
# 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
-
446
422
# def test_proprietary_from_hub_service(setup):
447
423
448
424
# model_id = "upstage-solar-mini-chat"
You can’t perform that action at this time.
0 commit comments