Enable the integration tests for MIG (#337)

* integration test for jumpstart with mig profile

* template fix for mig with jumpstart

* skipped mig tests until instances setup finished

* enable the mig integration tests

---------

Co-authored-by: Chad Chiang <[email protected]>

Author: chad119

test/integration_tests/inference/cli/test_cli_jumpstart_inference_with_mig.py

@@ -28,7 +28,6 @@ def sagemaker_client():
     return boto3.client("sagemaker", region_name=REGION)
 
 # --------- JumpStart Endpoint Tests ---------
-@pytest.mark.skip(reason="Temporarily disabled")
 @pytest.mark.dependency(name="create")
 def test_js_create(runner, js_endpoint_name):
     result = runner.invoke(js_create, [
@@ -101,12 +100,10 @@ def test_custom_invoke(runner, js_endpoint_name):
     assert result.exit_code == 0
     assert "error" not in result.output.lower()
 
-@pytest.mark.skip(reason="Temporarily disabled")
 def test_js_get_operator_logs(runner):
     result = runner.invoke(js_get_operator_logs, ["--since-hours", "1"])
     assert result.exit_code == 0
 
-@pytest.mark.skip(reason="Temporarily disabled")
 def test_js_list_pods(runner):
     result = runner.invoke(js_list_pods, ["--namespace", NAMESPACE])
     assert result.exit_code == 0

test/integration_tests/inference/sdk/test_sdk_jumpstart_inference_with_mig.py

@@ -38,7 +38,6 @@ def endpoint_obj():
 
     return HPJumpStartEndpoint(metadata=metadata, model=model, server=server, sage_maker_endpoint=sm_endpoint)
 
-@pytest.mark.skip(reason="Temporarily disabled")
 @pytest.mark.dependency(name="create")
 def test_create_endpoint(endpoint_obj):
     endpoint_obj.create()
@@ -105,13 +104,11 @@ def mock_transform(data, shape, object_instance=None):
     assert "error" not in response.body.lower()
 
 
-@pytest.mark.skip(reason="Temporarily disabled")
 def test_get_operator_logs():
     ep = HPJumpStartEndpoint.get(name=ENDPOINT_NAME, namespace=NAMESPACE)
     logs = ep.get_operator_logs(since_hours=1)
     assert logs
 
-@pytest.mark.skip(reason="Temporarily disabled")
 def test_list_pods():
     ep = HPJumpStartEndpoint.get(name=ENDPOINT_NAME, namespace=NAMESPACE)
     pods = ep.list_pods(NAMESPACE)