We will be using a package called OpenAI Gym to build reinforcement learning agents. You can learn more about it here: https://gym.openai.com . We can install it using pip by running the following command on the Terminal:

$ pip3 install gym

You can find various tips and tricks related to its installation here: https://github.com/openai/gym#installation . Now that you have installed it, let's go ahead and write some code.

Create a new python file and import the following package:

import argparse 
 
import gym 

Define a function to parse the input arguments. We will be able to specify the type of environment we want to run:

def build_arg_parser(): 
    parser = argparse.ArgumentParser(description='Run an environment') 
    parser.add_argument('--input-env', dest='input_env', required=True, 
            choices=['cartpole', 'mountaincar', 'pendulum', 'taxi', 'lake'],  
            help='Specify the name of the environment') 
   ...