def generate_data(FLAGS): name_size = 'SZ%d-STP%d' % (FLAGS.sz, FLAGS.stp) config_size = Config(size=FLAGS.sz, max_steps=FLAGS.stp) for name_std, config_std in CONFIG_STDS.iteritems(): for name_drop, config_drop in CONFIG_DROPS.iteritems(): for name_direction, config_direction in CONFIG_DIRECTIONS.iteritems( ): config = Config() config.add('base', 'base', CONFIG_BASE) config.add('size', name_size, config_size) config.add('direction', name_direction, config_direction) config.add('drop', name_drop, config_drop) config.add('std', name_std, config_std) gridworld = GridWorld(name=config.get_name(), **config.get_kwargs()) for seed in GRIDWORLD_SEEDS: data_dir = '%s-SEED%d' % (config.get_name(), seed) gridworld.generate(data_dir=data_dir, seed=seed, splitting_seed=SPLITTING_SEED)