|
20 | 20 |
|
21 | 21 | defaults = {
|
22 | 22 | 'bucket': 'ldbc-snb-datagen-store',
|
23 |
| - 'use_spot': False, |
| 23 | + 'use_spot': True, |
24 | 24 | 'master_instance_type': 'm5d.2xlarge',
|
25 | 25 | 'instance_type': 'i3.4xlarge',
|
26 | 26 | 'sf_ratio': 100.0, # ratio of SFs and machines. a ratio of 50.0 for SF100 yields 2 machines
|
@@ -214,10 +214,16 @@ def submit_datagen_job(name,
|
214 | 214 | help='scale factor (used to calculate cluster size)')
|
215 | 215 | parser.add_argument('format', type=str, help='the required output format')
|
216 | 216 | parser.add_argument('mode', type=str, help='output mode')
|
217 |
| - parser.add_argument('--use-spot', |
218 |
| - default=defaults['use_spot'], |
219 |
| - action='store_true', |
220 |
| - help='Use SPOT workers') |
| 217 | + market_args = parser.add_mutually_exclusive_group() |
| 218 | + market_args.add_argument('--use-spot', |
| 219 | + default=defaults['use_spot'], |
| 220 | + action='store_true', |
| 221 | + help='Use SPOT workers') |
| 222 | + market_args.add_argument('--no-use-spot', |
| 223 | + default=not defaults['use_spot'], |
| 224 | + dest='use_spot', |
| 225 | + action='store_false', |
| 226 | + help='Do not use SPOT workers') |
221 | 227 | parser.add_argument('--az',
|
222 | 228 | default=defaults['az'],
|
223 | 229 | help=f'Cluster availability zone. Default: {defaults["az"]}')
|
|
0 commit comments