multi-agent-rlenv 1.0.0

A strongly typed Multi-Agent Reinforcement Learning framework

RLEnv: yet another RL framework

This framework aims at high level abstractions of RL models, allowing to build algorithms on top of it.

Designing an environment

To create an environment that is compatible with RLEnv, you should inherit from the RLEnv class.

Instanciating an environment

Simple environments

import marlenv as menv
print(menv.__version__)

# From Gym
env = menv.make("CartPole-v1")

# From pettingzoo
from pettingzoo.sisl import pursuit_v4
env = menv.make(pursuit_v4.parallel_env())

Adding extra information to the observations

import marlenv as menv
# Building the environment with additional information
from pettingzoo.sisl import pursuit_v4
env = menv.Builder(pursuit_v4.parallel_env())\
    .with_agent_id()\
    .with_last_action()\
    .build()
# 8 agents  + 5 actions = 13 extras
assert env.extra_feature_shape == (13, )

Related projects

• MARL: multi-agent reinforcement framework https://github.com/yamoling/marl