Skip to content

Commit bfd3f6b

Browse files
authored
Fix linting (#2250)
1 parent 82bcb89 commit bfd3f6b

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

integrations/amazon_sagemaker/tests/test_sagemaker.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import os
2+
import re
23
from unittest.mock import Mock
34

45
import pytest
@@ -86,8 +87,10 @@ def test_init_raises_boto_error(set_env_variables, mock_boto3_session): # noqa:
8687
mock_boto3_session.side_effect = BotoCoreError()
8788
with pytest.raises(
8889
AWSConfigurationError,
89-
match="Could not connect to SageMaker Inference Endpoint 'test-model'."
90-
"Make sure the Endpoint exists and AWS environment is configured.",
90+
match=re.escape(
91+
"Could not connect to SageMaker Inference Endpoint 'test-model'."
92+
"Make sure the Endpoint exists and AWS environment is configured."
93+
),
9194
):
9295
SagemakerGenerator(model="test-model")
9396

0 commit comments

Comments
 (0)