Skip to content

Commit d1e0f5b

Browse files
committed
fix: fix black-check warnings
1 parent e96f157 commit d1e0f5b

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

sagemaker-train/tests/unit/train/test_tuner.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -581,8 +581,7 @@ def test_build_training_job_definition_includes_internal_channels(self):
581581
assert len(channel_names) == 4, "Should have exactly 4 channels"
582582

583583
def test_build_training_job_definition_includes_spot_params(self):
584-
"""Test that _build_training_job_definition includes spot parameters.
585-
"""
584+
"""Test that _build_training_job_definition includes spot parameters."""
586585
tuner = HyperparameterTuner(
587586
model_trainer=_create_mock_model_trainer(with_spot_training=True),
588587
objective_metric_name="accuracy",
@@ -594,4 +593,6 @@ def test_build_training_job_definition_includes_spot_params(self):
594593

595594
# Verify managed spot training enabled
596595
assert definition.enable_managed_spot_training is True, "Spot should be enabled"
597-
assert isinstance(definition.stopping_condition.max_wait_time_in_seconds, int), "Max wait time should be set"
596+
assert isinstance(
597+
definition.stopping_condition.max_wait_time_in_seconds, int
598+
), "Max wait time should be set"

0 commit comments

Comments
 (0)