Vacuum World as a Gym environment
git clone [email protected]:BenedictWilkinsAI/vacuumworld-gym.git
cd vacuumworld-gym
pip install -e .
import random
from vwgym import VacuumWorld, Vectorise
env = VacuumWorld(3)
env = Vectorise(env)
state = env.reset()
for i in range(10):
action = random.randint(0,env.action_space.n-1)
state, reward, done, *_ = env.step(action)
For running instructions please refer here.