%load_ext autoreload
%autoreload 2
from rush import *
import matplotlib.pyplot as plt
import numpy as np
import time

s = State()
s.load_puzzle("jams/easy.txt")

tic = time.time() # Time the solution
solution = s.solve()
toc = time.time()
print("Elapsed time: {:.3f} seconds".format(toc-tic))
for i, s in enumerate(solution):
    plt.figure()
    s.plot()
    plt.title("Step {}".format(i))

Elapsed time: 16.948 seconds