How to export or load multi-agent policies?

[ardian-selmonaj]

In a multi-agent setting, when training e.g. MAPPO_Agents(), then calling MAPPO_Agents.save_model(model_name='model.pth') and finally loading the model MAPPO_Agents.load_model(path), how can I extract specific policies out of the model? e.g. when 3 agents were trained, "agent_1", "agent_2" and "agent_3", I would like to have the prediction only of a specific agent.

I didn't find out how to exactly do that and would be very thankful for any help.