Fixes the completion_criteria_config dict in the to_input_req method (aws#3744)

Co-authored-by: Konrad Semsch <[email protected]>

Author: 2 people

src/sagemaker/tuner.py

@@ -559,6 +559,7 @@ def to_input_req(self):
         """
         completion_criteria_config = {}
         if self.max_number_of_training_jobs_not_improving is not None:
+            completion_criteria_config[BEST_OBJECTIVE_NOT_IMPROVING] = {}
             completion_criteria_config[BEST_OBJECTIVE_NOT_IMPROVING][
                 MAX_NUMBER_OF_TRAINING_JOBS_NOT_IMPROVING
             ] = self.max_number_of_training_jobs_not_improving
@@ -569,6 +570,7 @@ def to_input_req(self):
             ] = self.target_objective_metric_value
 
         if self.complete_on_convergence is not None:
+            completion_criteria_config[CONVERGENCE_DETECTED] = {}
             completion_criteria_config[CONVERGENCE_DETECTED][COMPLETE_ON_CONVERGENCE_DETECTED] = (
                 "Enabled" if self.complete_on_convergence else "Disabled"
             )