FPS battle royale game AI using ML agents

For battle royale game (like PUBG), I want to train a team of shooters using self-play, I am unsure what to do when my player dies.

I want to understand how to handle the death of agents in the two scenarios -

1. Say 1 team has exactly 1 player, and at a time there are 4 players in the battleground, sharing the same model with different team ID, so what should I call if 1 Player dies?
OnEpisodeEnd() and gameObject.SetActive(false)?

2. Say 1 team has exactly 2+ players the what should I call on the death of a teammate?