Skip to content

Commit 71adb25

Browse files
author
Brock Wade
committed
fix: formatting
1 parent 623d85b commit 71adb25

File tree

6 files changed

+15
-11
lines changed

6 files changed

+15
-11
lines changed

src/sagemaker/processing.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1705,6 +1705,7 @@ def _pack_and_upload_code(
17051705
):
17061706
"""Pack local code bundle and upload to Amazon S3."""
17071707
from sagemaker.workflow.utilities import _pipeline_config, hash_object
1708+
17081709
if code.startswith("s3://"):
17091710
return code, inputs, job_name
17101711

tests/data/framework_processor_data/evaluate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@
22
Integ test file evaluate.py
33
"""
44

5-
print("test evaluate script")
5+
print("test evaluate script")

tests/data/framework_processor_data/preprocess.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@
22
Integ test file preprocess.py
33
"""
44

5-
print("test preprocess script")
5+
print("test preprocess script")

tests/data/framework_processor_data/query_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@
22
Integ test file query_data.py
33
"""
44

5-
print("test query data script")
5+
print("test query data script")

tests/data/framework_processor_data/train_test_split.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@
22
Integ test file train_test_split.py
33
"""
44

5-
print("test train, test, split script")
5+
print("test train, test, split script")

tests/integ/sagemaker/workflow/test_workflow.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@
4848
ProcessingOutput,
4949
FeatureStoreOutput,
5050
ScriptProcessor,
51-
FrameworkProcessor
51+
FrameworkProcessor,
5252
)
5353
from sagemaker.s3 import S3Uploader
5454
from sagemaker.session import get_execution_role
@@ -86,7 +86,7 @@
8686
TransformInput,
8787
PropertyFile,
8888
TuningStep,
89-
CacheConfig
89+
CacheConfig,
9090
)
9191
from sagemaker.workflow.step_collections import RegisterModel
9292
from sagemaker.workflow.pipeline import Pipeline
@@ -1316,7 +1316,9 @@ def test_caching_behavior(
13161316
pass
13171317

13181318

1319-
def test_multi_step_framework_processing_pipeline_uploads(pipeline_session, role, pipeline_name, region_name):
1319+
def test_multi_step_framework_processing_pipeline_uploads(
1320+
pipeline_session, role, pipeline_name, region_name
1321+
):
13201322
default_bucket = pipeline_session.default_bucket()
13211323
cache_config = CacheConfig(enable_caching=True, expire_after="PT1H")
13221324
evaluation_report = PropertyFile(
@@ -1419,16 +1421,17 @@ def test_multi_step_framework_processing_pipeline_uploads(pipeline_session, role
14191421
)
14201422

14211423
pipeline = Pipeline(
1422-
name=pipeline_name,
1423-
steps=[query_step, prepare_step, split_step, evaluate_step]
1424+
name=pipeline_name, steps=[query_step, prepare_step, split_step, evaluate_step]
14241425
)
14251426
try:
14261427
pipeline.create(role)
14271428

14281429
definition = json.loads(pipeline.definition())
14291430

1430-
source_dir_tar_prefix = f"s3://{default_bucket}/{pipeline_name}" \
1431+
source_dir_tar_prefix = (
1432+
f"s3://{default_bucket}/{pipeline_name}"
14311433
f"/code/{hash_files_or_dirs([DATA_DIR + '/framework_processor_data'])}"
1434+
)
14321435

14331436
run_procs = []
14341437

@@ -1448,4 +1451,4 @@ def test_multi_step_framework_processing_pipeline_uploads(pipeline_session, role
14481451
try:
14491452
pipeline.delete()
14501453
except Exception:
1451-
pass
1454+
pass

0 commit comments

Comments
 (0)