Skip to content

Commit 5c97926

Browse files
committed
Update the predictor test with the correct assert
1 parent 7d896a2 commit 5c97926

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/unit/test_predictor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,8 +62,8 @@ def test_predict_call_pass_through():
6262
result = predictor.predict(data)
6363

6464
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()
65+
assert sagemaker_session.sagemaker_client.describe_endpoint.not_called
66+
assert sagemaker_session.sagemaker_client.describe_endpoint_config.not_called
6767

6868
expected_request_args = {
6969
"Accept": DEFAULT_ACCEPT,

0 commit comments

Comments
 (0)