(argv)
| 51 | |
| 52 | |
| 53 | def main(argv): |
| 54 | del argv |
| 55 | environment_name = FLAGS.environment_name |
| 56 | if environment_name is None: |
| 57 | print('\n '.join(['Available environments:'] + _ALL_NAMES)) |
| 58 | environment_name = prompt_environment_name( |
| 59 | 'Please select an environment name: ', _ALL_NAMES) |
| 60 | |
| 61 | index = _ALL_NAMES.index(environment_name) |
| 62 | domain_name, task_name = suite.ALL_TASKS[index] |
| 63 | |
| 64 | task_kwargs = {} |
| 65 | if not FLAGS.timeout: |
| 66 | task_kwargs['time_limit'] = float('inf') |
| 67 | |
| 68 | def loader(): |
| 69 | env = suite.load( |
| 70 | domain_name=domain_name, task_name=task_name, task_kwargs=task_kwargs) |
| 71 | env.task.visualize_reward = FLAGS.visualize_reward |
| 72 | if FLAGS.action_noise > 0: |
| 73 | env = action_noise.Wrapper(env, scale=FLAGS.action_noise) |
| 74 | return env |
| 75 | |
| 76 | viewer.launch(loader) |
| 77 | |
| 78 | |
| 79 | if __name__ == '__main__': |
nothing calls this directly
no test coverage detected
searching dependent graphs…