Bare minimum to interface to ALE and display the screen with Matplotlib

minim_ale.py

import numpy as np
import matplotlib
matplotlib.use('TkAgg')
import matplotlib.pyplot as plt
from ale_python_interface import ale_python_interface as ALE

def display_ale(im, screen):
    im.set_data(screen)
    plt.draw()

ale = ALE.ALEInterface()
ale.setInt(b'random_seed', 1234)
ale.loadROM(b'breakout.bin')

plt.ion()
fig = plt.figure()
ax = fig.add_subplot(111)
im = ax.imshow(ale.getScreenRGB())
plt.draw()

legal_actions = ale.getMinimalActionSet()
while not ale.game_over():
    ale.act(np.random.choice(legal_actions))
    display_ale(im, ale.getScreenRGB())