def test_utils_stop_hyper_parameter_tuning_job(self): mock_sm_client = MagicMock() mock_sm_client.stop_hyper_parameter_tuning_job.return_value = None response = _utils.stop_hyperparameter_tuning_job( mock_sm_client, 'FakeJobName') mock_sm_client.stop_hyper_parameter_tuning_job.assert_called_once_with( HyperParameterTuningJobName='FakeJobName') self.assertEqual(response, None)
def signal_term_handler(signalNumber, frame): _utils.stop_hyperparameter_tuning_job(client, hpo_job_name) logging.info( f"HyperParameter Tuning Job: {hpo_job_name} request submitted to Stop" )