Skip to content

Commit 7d896a2

Browse files
committed
Update unit test for predictor
1 parent dd78d53 commit 7d896a2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/unit/test_predictor.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +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()
6567

6668
expected_request_args = {
6769
"Accept": DEFAULT_ACCEPT,

0 commit comments

Comments
 (0)