Skip to content

Commit 6826d12

Browse files
committed
Remove unused test constants
1 parent c652e64 commit 6826d12

File tree

3 files changed

+3
-13
lines changed

3 files changed

+3
-13
lines changed

tests/integ/test_data_capture_config.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,23 +17,17 @@
1717
import sagemaker
1818
import tests.integ
1919
import tests.integ.timeout
20-
21-
from tests.integ.retry import retries
22-
from sagemaker.model_monitor import NetworkConfig
23-
24-
from sagemaker.model_monitor import DataCaptureConfig
20+
from sagemaker.model_monitor import DataCaptureConfig, NetworkConfig
2521
from sagemaker.tensorflow.serving import Model
2622
from sagemaker.utils import unique_name_from_base
23+
from tests.integ.retry import retries
2724

2825
ROLE = "SageMakerRole"
29-
REGION = "us-west-2"
3026
SKLEARN_FRAMEWORK = "scikit-learn"
3127

3228
INSTANCE_COUNT = 1
3329
INSTANCE_TYPE = "ml.m5.xlarge"
3430
VOLUME_SIZE_IN_GB = 20
35-
VOLUME_KMS_KEY = "arn:aws:kms:us-west-2:012345678901:key/volume-kms-key"
36-
OUTPUT_KMS_KEY = "arn:aws:kms:us-west-2:012345678901:key/output-kms-key"
3731
MAX_RUNTIME_IN_SECONDS = 2 * 60 * 60
3832
ENVIRONMENT = {"env_key_1": "env_value_1"}
3933
TAGS = [{"Key": "tag_key_1", "Value": "tag_value_1"}]

tests/integ/test_monitoring_files.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,13 +19,10 @@
1919

2020
import tests.integ
2121
import tests.integ.timeout
22-
2322
from sagemaker.model_monitor import Statistics, Constraints, ConstraintViolations
2423
from sagemaker.s3 import S3Uploader
2524
from tests.integ.kms_utils import get_or_create_kms_key
2625

27-
ROLE = "arn:aws:iam::142577830533:role/SageMakerRole"
28-
2926

3027
@pytest.fixture(scope="module")
3128
def monitoring_files_kms_key(sagemaker_session):
@@ -40,7 +37,7 @@ def test_statistics_object_creation_from_file_path_with_customizations(
4037
kms_key=monitoring_files_kms_key,
4138
sagemaker_session=sagemaker_session,
4239
)
43-
#
40+
4441
assert statistics.file_s3_uri.startswith("s3://")
4542
assert statistics.file_s3_uri.endswith("statistics.json")
4643

tests/integ/test_processing.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
# language governing permissions and limitations under the License.
1313
from __future__ import absolute_import
1414

15-
import logging
1615
import os
1716

1817
import pytest

0 commit comments

Comments
 (0)