Matplotlib - RuntimeWarning: More than 20 figures have been opened.

RuntimeWarning: More than 20 figures have been opened. Figures created through the pyplot interface (`matplotlib.pyplot.figure
`) are retained until explicitly closed and may consume too much memory. (To control this warning, see the rcParam `figure.max_open_warning`).

Need to close the figures after using:

plt.close(fig)