When I changed the environment, some errors occurred

[chenci107]

When I change the environment from HalfCheetahEnv to Arm7DofEnv (or HalfCheetahBlocksEnv), when the first round of training is over, the following error appears: Traceback (most recent call last): File "run_scripts/run_grbal.py", line 109, in run_experiment(config) File "run_scripts/run_grbal.py", line 68, in run_experiment algo.train() File "/home/chenci/code/Recovery/learning_to_adapt-master (1)/learning_to_adapt/trainers/mb_trainer.py", line 107, in train self.log_diagnostics(env_paths, '') File "/home/chenci/code/Recovery/learning_to_adapt-master (1)/learning_to_adapt/trainers/mb_trainer.py", line 125, in log_diagnostics self.env.log_diagnostics(paths, prefix) File "/home/chenci/code/Recovery/learning_to_adapt-master (1)/learning_to_adapt/envs/normalized_env.py", line 83, in hooked result = orig_attr(*args, **kwargs) TypeError: log_diagnostics() takes 2 positional arguments but 3 were given What should I do to solve this problem? Thanks~