def test_endpoint_context_fail(): session = context_sagemaker_session(summaries=False) pdctr = Predictor(ENDPOINT, sagemaker_session=session) context = pdctr.endpoint_context() assert not context
def test_endpoint_context_success(): session = context_sagemaker_session() pdctr = Predictor(ENDPOINT, sagemaker_session=session) context = pdctr.endpoint_context() assert context