use 'nf.yaml' meets some problems

[Lewisvan]

when i want to use 'nf.yaml' to create scenario,i have some problems
Traceback (most recent call last):
File "/home/cai/SafeBench/scripts/run.py", line 78, in
runner.run()
File "/home/cai/SafeBench/safebench/carla_runner.py", line 346, in run
self.train(data_loader, start_episode)
File "/home/cai/SafeBench/safebench/carla_runner.py", line 191, in train
scenario_init_action, additional_dict = self.scenario_policy.get_init_action(static_obs)
File "/home/cai/SafeBench/safebench/scenario/scenario_policy/normalizing_flow_policy.py", line 301, in get_init_action
action = self.model.inverse(mean, condition)
File "/home/cai/SafeBench/safebench/scenario/scenario_policy/normalizing_flow_policy.py", line 111, in inverse
z_a_c = torch.cat([z_a, c], dim=1)
RuntimeError: Tensors must have same number of dimensions: got 2 and 3

parser.add_argument('--mode', '-m', type=str, default='train_scenario', choices=['train_agent', 'train_scenario', 'eval'])
parser.add_argument('--agent_cfg', nargs='', type=str, default=['basic.yaml'])
parser.add_argument('--scenario_cfg', nargs='', type=str, default=['nf.yaml'])