snake-gym-grid 0.0.5

An implementation on Snake game with grid display

snake-gym

This package is my own implementation on Snake game with Gym integration.

Usage

import gym
import snake_gym_grid.snake_gym_grid

env = gym.make("snake-gym-10x20-v0", new_step_api=True, render_mode="human")
env.reset()

done = False # whether the snake dies

"""
action ranges from 0 to 3.
0 - LEFT
1 - RIGHT
2 - UP
3 - DOWN
"""
action = ...

observation = env.reset()

while not done:
    env.render()
    # I just use truncate because it's recommended by Gym
    # However, you may not need it
    # It signifies whether the game is running for more than 100000 steps
    observation, reward, done, truncate, info = env.step(action)
env.close()

Output at each time step

• observation: Pixel image of the game
• reward: 1.0 when a food is eaten 0.0 otherwise
• done: whether the snake dies
• truncate: whether the game is running for more than 100000 time steps
• info: an empty dictionary for now