updates

Author: Ye Shaokai

llava/action/ek_eval.py

@@ -342,7 +342,10 @@ def collate_fn(batch):
     if eval_args.learn_neighbor_actions:
         from llava.action.generate_interval_pred import  get_lookup_dict
         if eval_args.test_type.startswith('temporal_cot'):
-            lookup_table = get_lookup_dict(eval_args.val_metadata, test_type = eval_args.test_type, pseudo_folder = eval_args.pseudo_folder)
+            lookup_table = get_lookup_dict(eval_args.val_metadata, 
+                                           eval_args.action_representation,
+                                           test_type = eval_args.test_type, 
+                                           pseudo_folder = eval_args.pseudo_folder)
 
 
     for idx, (frames, mc_data, time_meta, global_index) in tqdm(enumerate(val_dataloader)):