Skip to content

Commit dac219a

Browse files
navinsoninavinsoni
committed
Fix style
1 parent c81ac9c commit dac219a

File tree

4 files changed

+92
-59
lines changed

4 files changed

+92
-59
lines changed

‎src/sagemaker/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def name_from_base(base, max_length=63, short=False):
7979

8080
def unique_name_from_base(base, max_length=63):
8181
"""Placeholder Docstring"""
82-
unique = "%04x" % random.randrange(16**4) # 4-digit hex
82+
unique = "%04x" % random.randrange(16 ** 4) # 4-digit hex
8383
ts = str(int(time.time()))
8484
available_length = max_length - 2 - len(ts) - len(unique)
8585
trimmed = base[:available_length]

‎tests/integ/sagemaker/lineage/helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def retry(callable, num_attempts=8):
4242
if i == num_attempts - 1:
4343
raise ex
4444
print("Retrying", ex)
45-
time.sleep(2**i)
45+
time.sleep(2 ** i)
4646
assert False, "logic error in retry"
4747

4848

‎tests/unit/sagemaker/feature_store/test_feature_store.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -308,13 +308,16 @@ def test_as_hive_ddl(create_table_ddl, feature_group_dummy_definitions, sagemake
308308

309309
feature_group = FeatureGroup(name="MyGroup", sagemaker_session=sagemaker_session_mock)
310310
feature_group.feature_definitions = feature_group_dummy_definitions
311-
assert create_table_ddl.format(
312-
database="MyDatabase",
313-
table_name="MyTable",
314-
account="1234",
315-
region="us-west-2",
316-
feature_group_name="MyGroup",
317-
) == feature_group.as_hive_ddl(database="MyDatabase", table_name="MyTable")
311+
assert (
312+
create_table_ddl.format(
313+
database="MyDatabase",
314+
table_name="MyTable",
315+
account="1234",
316+
region="us-west-2",
317+
feature_group_name="MyGroup",
318+
)
319+
== feature_group.as_hive_ddl(database="MyDatabase", table_name="MyTable")
320+
)
318321

319322

320323
@patch(

‎tests/unit/sagemaker/jumpstart/test_utils.py

Lines changed: 80 additions & 50 deletions
Original file line numberDiff line numberDiff line change
@@ -169,29 +169,38 @@ def test_add_jumpstart_tags_inference():
169169
tags = [{"Key": "some", "Value": "tag"}]
170170
inference_model_uri = "dfsdfsd"
171171
inference_script_uri = "dfsdfs"
172-
assert utils.add_jumpstart_tags(
173-
tags=tags,
174-
inference_model_uri=inference_model_uri,
175-
inference_script_uri=inference_script_uri,
176-
) == [{"Key": "some", "Value": "tag"}]
172+
assert (
173+
utils.add_jumpstart_tags(
174+
tags=tags,
175+
inference_model_uri=inference_model_uri,
176+
inference_script_uri=inference_script_uri,
177+
)
178+
== [{"Key": "some", "Value": "tag"}]
179+
)
177180

178181
tags = None
179182
inference_model_uri = random_jumpstart_s3_uri("random_key")
180183
inference_script_uri = "dfsdfs"
181-
assert utils.add_jumpstart_tags(
182-
tags=tags,
183-
inference_model_uri=inference_model_uri,
184-
inference_script_uri=inference_script_uri,
185-
) == [{"Key": JumpStartTag.INFERENCE_MODEL_URI.value, "Value": inference_model_uri}]
184+
assert (
185+
utils.add_jumpstart_tags(
186+
tags=tags,
187+
inference_model_uri=inference_model_uri,
188+
inference_script_uri=inference_script_uri,
189+
)
190+
== [{"Key": JumpStartTag.INFERENCE_MODEL_URI.value, "Value": inference_model_uri}]
191+
)
186192

187193
tags = []
188194
inference_model_uri = random_jumpstart_s3_uri("random_key")
189195
inference_script_uri = "dfsdfs"
190-
assert utils.add_jumpstart_tags(
191-
tags=tags,
192-
inference_model_uri=inference_model_uri,
193-
inference_script_uri=inference_script_uri,
194-
) == [{"Key": JumpStartTag.INFERENCE_MODEL_URI.value, "Value": inference_model_uri}]
196+
assert (
197+
utils.add_jumpstart_tags(
198+
tags=tags,
199+
inference_model_uri=inference_model_uri,
200+
inference_script_uri=inference_script_uri,
201+
)
202+
== [{"Key": JumpStartTag.INFERENCE_MODEL_URI.value, "Value": inference_model_uri}]
203+
)
195204

196205
tags = [{"Key": "some", "Value": "tag"}]
197206
inference_model_uri = random_jumpstart_s3_uri("random_key")
@@ -208,20 +217,26 @@ def test_add_jumpstart_tags_inference():
208217
tags = None
209218
inference_script_uri = random_jumpstart_s3_uri("random_key")
210219
inference_model_uri = "dfsdfs"
211-
assert utils.add_jumpstart_tags(
212-
tags=tags,
213-
inference_model_uri=inference_model_uri,
214-
inference_script_uri=inference_script_uri,
215-
) == [{"Key": JumpStartTag.INFERENCE_SCRIPT_URI.value, "Value": inference_script_uri}]
220+
assert (
221+
utils.add_jumpstart_tags(
222+
tags=tags,
223+
inference_model_uri=inference_model_uri,
224+
inference_script_uri=inference_script_uri,
225+
)
226+
== [{"Key": JumpStartTag.INFERENCE_SCRIPT_URI.value, "Value": inference_script_uri}]
227+
)
216228

217229
tags = []
218230
inference_script_uri = random_jumpstart_s3_uri("random_key")
219231
inference_model_uri = "dfsdfs"
220-
assert utils.add_jumpstart_tags(
221-
tags=tags,
222-
inference_model_uri=inference_model_uri,
223-
inference_script_uri=inference_script_uri,
224-
) == [{"Key": JumpStartTag.INFERENCE_SCRIPT_URI.value, "Value": inference_script_uri}]
232+
assert (
233+
utils.add_jumpstart_tags(
234+
tags=tags,
235+
inference_model_uri=inference_model_uri,
236+
inference_script_uri=inference_script_uri,
237+
)
238+
== [{"Key": JumpStartTag.INFERENCE_SCRIPT_URI.value, "Value": inference_script_uri}]
239+
)
225240

226241
tags = [{"Key": "some", "Value": "tag"}]
227242
inference_script_uri = random_jumpstart_s3_uri("random_key")
@@ -349,29 +364,38 @@ def test_add_jumpstart_tags_training():
349364
tags = [{"Key": "some", "Value": "tag"}]
350365
training_model_uri = "dfsdfsd"
351366
training_script_uri = "dfsdfs"
352-
assert utils.add_jumpstart_tags(
353-
tags=tags,
354-
training_model_uri=training_model_uri,
355-
training_script_uri=training_script_uri,
356-
) == [{"Key": "some", "Value": "tag"}]
367+
assert (
368+
utils.add_jumpstart_tags(
369+
tags=tags,
370+
training_model_uri=training_model_uri,
371+
training_script_uri=training_script_uri,
372+
)
373+
== [{"Key": "some", "Value": "tag"}]
374+
)
357375

358376
tags = None
359377
training_model_uri = random_jumpstart_s3_uri("random_key")
360378
training_script_uri = "dfsdfs"
361-
assert utils.add_jumpstart_tags(
362-
tags=tags,
363-
training_model_uri=training_model_uri,
364-
training_script_uri=training_script_uri,
365-
) == [{"Key": JumpStartTag.TRAINING_MODEL_URI.value, "Value": training_model_uri}]
379+
assert (
380+
utils.add_jumpstart_tags(
381+
tags=tags,
382+
training_model_uri=training_model_uri,
383+
training_script_uri=training_script_uri,
384+
)
385+
== [{"Key": JumpStartTag.TRAINING_MODEL_URI.value, "Value": training_model_uri}]
386+
)
366387

367388
tags = []
368389
training_model_uri = random_jumpstart_s3_uri("random_key")
369390
training_script_uri = "dfsdfs"
370-
assert utils.add_jumpstart_tags(
371-
tags=tags,
372-
training_model_uri=training_model_uri,
373-
training_script_uri=training_script_uri,
374-
) == [{"Key": JumpStartTag.TRAINING_MODEL_URI.value, "Value": training_model_uri}]
391+
assert (
392+
utils.add_jumpstart_tags(
393+
tags=tags,
394+
training_model_uri=training_model_uri,
395+
training_script_uri=training_script_uri,
396+
)
397+
== [{"Key": JumpStartTag.TRAINING_MODEL_URI.value, "Value": training_model_uri}]
398+
)
375399

376400
tags = [{"Key": "some", "Value": "tag"}]
377401
training_model_uri = random_jumpstart_s3_uri("random_key")
@@ -388,20 +412,26 @@ def test_add_jumpstart_tags_training():
388412
tags = None
389413
training_script_uri = random_jumpstart_s3_uri("random_key")
390414
training_model_uri = "dfsdfs"
391-
assert utils.add_jumpstart_tags(
392-
tags=tags,
393-
training_model_uri=training_model_uri,
394-
training_script_uri=training_script_uri,
395-
) == [{"Key": JumpStartTag.TRAINING_SCRIPT_URI.value, "Value": training_script_uri}]
415+
assert (
416+
utils.add_jumpstart_tags(
417+
tags=tags,
418+
training_model_uri=training_model_uri,
419+
training_script_uri=training_script_uri,
420+
)
421+
== [{"Key": JumpStartTag.TRAINING_SCRIPT_URI.value, "Value": training_script_uri}]
422+
)
396423

397424
tags = []
398425
training_script_uri = random_jumpstart_s3_uri("random_key")
399426
training_model_uri = "dfsdfs"
400-
assert utils.add_jumpstart_tags(
401-
tags=tags,
402-
training_model_uri=training_model_uri,
403-
training_script_uri=training_script_uri,
404-
) == [{"Key": JumpStartTag.TRAINING_SCRIPT_URI.value, "Value": training_script_uri}]
427+
assert (
428+
utils.add_jumpstart_tags(
429+
tags=tags,
430+
training_model_uri=training_model_uri,
431+
training_script_uri=training_script_uri,
432+
)
433+
== [{"Key": JumpStartTag.TRAINING_SCRIPT_URI.value, "Value": training_script_uri}]
434+
)
405435

406436
tags = [{"Key": "some", "Value": "tag"}]
407437
training_script_uri = random_jumpstart_s3_uri("random_key")

0 commit comments

Comments
 (0)