I see warning messages in prod for memory

Explore if this can help: ```python matplotlib.pyplot.close() ```