recieves brain_name and identifier on python side

ml-agents/mlagents/trainers/ppo/trainer.py

 
 import numpy as np
 
-from mlagents.trainers.brain import BrainParameters, BrainInfo
+from mlagents.envs.brain import BrainParameters, BrainInfo
 from mlagents.trainers.ppo.policy import PPOPolicy
 from mlagents.trainers.ppo.multi_gpu_policy import MultiGpuPPOPolicy, get_devices
 from mlagents.trainers.rl_trainer import RLTrainer, AllRewardsOutput

ml-agents/mlagents/trainers/sac/trainer.py

 
 import numpy as np
 
-from mlagents.trainers.brain import BrainParameters, BrainInfo
-from mlagents.trainers.action_info import ActionInfoOutputs
+from mlagents.envs.brain import BrainParameters, BrainInfo
+from mlagents.envs.action_info import ActionInfoOutputs
 from mlagents.envs.timers import timed
 from mlagents.trainers.sac.policy import SACPolicy
 from mlagents.trainers.rl_trainer import RLTrainer, AllRewardsOutput

ml-agents/mlagents/trainers/trainer_util.py

     :return:
     """
     trainer_parameters = trainer_config["default"].copy()
-    brain_name, brain_name_identifiers = brain_parameters.brain_name.split("?")
     trainer_parameters["summary_path"] = "{basedir}/{name}".format(
         basedir=summaries_dir, name=str(run_id) + "_" + brain_name
     )