diff --git a/ml-agents/mlagents/trainers/trainer_util.py b/ml-agents/mlagents/trainers/trainer_util.py index 01fe654dbc..e788bf56ce 100644 --- a/ml-agents/mlagents/trainers/trainer_util.py +++ b/ml-agents/mlagents/trainers/trainer_util.py @@ -39,6 +39,11 @@ def __init__( self.ghost_controller = GhostController() def generate(self, brain_name: str) -> Trainer: + if brain_name not in self.trainer_config.keys(): + logger.warning( + f"Behavior name {brain_name} does not match any behaviors specified in the trainer configuration file:" + f"{sorted(self.trainer_config.keys())}" + ) return initialize_trainer( self.trainer_config[brain_name], brain_name,