Adjust HyperParameterTuningJobConfig values (apache#28368)

XD-DENG · Dec 14, 2022 · fbbd453 · fbbd453
1 parent 753b9ec
commit fbbd453
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/system/providers/amazon/aws/example_sagemaker.py b/tests/system/providers/amazon/aws/example_sagemaker.py
@@ -328,9 +328,8 @@ def set_up(env_id, role_arn):
                 "Type": "Maximize",
             },
             "ResourceLimits": {
-                # You would bump these up in production as appropriate.
-                "MaxNumberOfTrainingJobs": 2,
-                "MaxParallelTrainingJobs": 2,
+                "MaxNumberOfTrainingJobs": 10,
+                "MaxParallelTrainingJobs": 10,
             },
             "ParameterRanges": {
                 "CategoricalParameterRanges": [],