We are trying to make a Tetris AI on a raspberry pi which is connected to a board with WS2812b LEDs. We defined the action_space and observation_space like this:

self.action_space = spaces.Discrete(6)
self.observation_space = spaces.Discrete(width * height * block_number * block_degree * stored ...