def load_hanabi_env(game_config):
    assert isinstance(game_config, dict)

    pyhanabi_env = rl_env.HanabiEnv(game_config)

    if pyhanabi_env is not None:
        return pyhanabi_env_wrapper.PyhanabiEnvWrapper(pyhanabi_env)

    return None
Beispiel #2
0
def load_hanabi_env(env_name="Hanabi-Full", num_players=4):
  pyhanabi_env = rl_env.make(environment_name=env_name, num_players=num_players)
  py_env = pyhanabi_env_wrapper.PyhanabiEnvWrapper(pyhanabi_env)
  return py_env
def load_env(variant="Hanabi-Small", players=4):
    pyhanabi_env = rl_env.make(environment_name=variant, num_players=players)
    py_env = pyhanabi_env_wrapper.PyhanabiEnvWrapper(pyhanabi_env)

    return py_env