We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7d896a2 commit 5c97926Copy full SHA for 5c97926
tests/unit/test_predictor.py
@@ -62,8 +62,8 @@ def test_predict_call_pass_through():
62
result = predictor.predict(data)
63
64
assert sagemaker_session.sagemaker_runtime_client.invoke_endpoint.called
65
- sagemaker_session.describe_endpoint.assert_not_called()
66
- sagemaker_session.describe_endpoint_config.assert_not_called()
+ assert sagemaker_session.sagemaker_client.describe_endpoint.not_called
+ assert sagemaker_session.sagemaker_client.describe_endpoint_config.not_called
67
68
expected_request_args = {
69
"Accept": DEFAULT_ACCEPT,
0 commit comments