Update exp_plot_reservoir_ret.py

bad imports