def get_agent_batch(self, batch_id):
        """
            Returns the agent_ids corresponding to batch_id

        Args:
            batch_id: index of batch

        Returns:
            agent_ids (list): ids of the agents in the batch
        """
        return range(self.agent_batch_size * int(batch_id),
                     self.agent_batch_size * (int(batch_id) + 1))

            Returns the agent_ids corresponding to batch_id

        Args:
            batch_id: index of batch

        Returns:
            agent_ids (list): ids of the agents in the batch
        
surreal/launch/launcher.py