import matplotlib.pyplot as plt import matplotlib def plot_env(ax, info): ax.cla() env = info['env'] ax.plot([1, 1], [1, 2], '.', color='b', alpha=0.5, linewidth=5, markersize=20) # ax.set_xlim([min(n_agents_list) - 20, max(n_agents_list) + 20]) ax.set_xlim([0, 100]) ax.set_ylim([0, 100]) # ax.set_xticks(n_agents_list) # ax.set_xlabel('N agents', fontsize=27) # ax.set_ylabel('Success Rate', fontsize=27) # ax.set_title(f'{img_dir[:-4]} Map | time limit: {time_to_think_limit} sec.') # set_plot_title(ax, f'{img_dir[:-4]} Map | time limit: {time_to_think_limit} sec.', size=11) ax.set_title(f'Warehouse', fontweight="bold", size=30) # set_legend(ax, size=18) # labelsize = 20 # ax.xaxis.set_tick_params(labelsize=labelsize) # ax.yaxis.set_tick_params(labelsize=labelsize) plt.tight_layout()