plot_functions.py

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()