def __init__(self, environment_name: str, bias: int):
self._environment_name = environment_name
self._bias = bias
self.envs = [
GymClient("BipedalWalkerHardcore-v3", False, 24, 4, bias, 1, 2000),
GymClient("BipedalWalker-v3", False, 24, 4, bias, 1, 1600)
]
def create_mountain_car(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 3 if max_trials is None else max_trials
max_episodes = 500 if max_episodes is None else max_episodes
return GymClient("MountainCar-v0", True, 2, 3, bias, max_trials,
max_episodes)
def create_bipedal_walker(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 1 if max_trials is None else max_trials
max_episodes = 1600 if max_episodes is None else max_episodes
return GymClient("BipedalWalker-v3", False, 24, 4, bias, max_trials,
max_episodes)
def create_lunar_lander_continuous(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 3 if max_trials is None else max_trials
max_episodes = 500 if max_episodes is None else max_episodes
return GymClient("LunarLanderContinuous-v2", False, 8, 2, bias,
max_trials, max_episodes)
def create_mountain_car_continuous(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 3 if max_trials is None else max_trials
max_episodes = 500 if max_episodes is None else max_episodes
return GymClient("MountainCarContinuous-v0", False, 2, 1, bias,
max_trials, max_episodes)
def create_lunar_lander(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 3 if max_trials is None else max_trials
max_episodes = 500 if max_episodes is None else max_episodes
return GymClient("LunarLander-v2", True, 8, 4, bias, max_trials,
max_episodes)
def create_cart_pole(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 5 if max_trials is None else max_trials
max_episodes = 200 if max_episodes is None else max_episodes
return GymClient("CartPole-v0", True, 4, 2, bias, max_trials,
max_episodes)
def create_double_cart_pole(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 1 if max_trials is None else max_trials
max_episodes = 500 if max_episodes is None else max_episodes
return GymClient("DoubleCartPole", True, 6, 2, bias, max_trials,
max_episodes)
def create_acrobot(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 2 if max_trials is None else max_trials
max_episodes = 500 if max_episodes is None else max_episodes
return GymClient("Acrobot-v1", True, 6, 3, bias, max_trials,
max_episodes)
def create_cart_pole_swingup(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 1 if max_trials is None else max_trials
max_episodes = 500 if max_episodes is None else max_episodes
return GymClient("CartPoleSwingUp-v0", False, 5, 1, bias, max_trials,
max_episodes)
def create_flappy_bird(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 1 if max_trials is None else max_trials
max_episodes = 1000 if max_episodes is None else max_episodes
return GymClient("FlappyBird-PLE-v0", False, 1, 3, bias, max_trials,
max_episodes)
def create_car_racing(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 1 if max_trials is None else max_trials
max_episodes = 1000 if max_episodes is None else max_episodes
return GymClient("CarRacing-v0", False, 96 * 96 * 3, 3, bias,
max_trials, max_episodes)
def create_double_inverted_pendulum(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 1 if max_trials is None else max_trials
max_episodes = 250 if max_episodes is None else max_episodes
return GymClient("InvertedDoublePendulum-v2", False, 11, 1, bias,
max_trials, max_episodes)
def create_pacman(bias: int = None,
max_trials: int = None,
max_episodes: int = None):
bias = 0 if bias is None else bias
max_trials = 1 if max_trials is None else max_trials
max_episodes = 10000 if max_episodes is None else max_episodes
return GymClient("MsPacman-ram-v0", True, 128, 9, bias, max_trials,
max_episodes)