File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed
sagemaker-train/tests/unit/train Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff 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"
You can’t perform that action at this time.
0 commit comments