columns as args in common cols

run_speech_recognition_ctc.py

@@ -418,9 +418,9 @@ def main():
     # 1. First, let's load the dataset
     raw_datasets = DatasetDict()
 
-    def common_cols(dataset_a, dataset_b):
-        col_a = set(dataset_a.column_names)
-        col_b = set(dataset_b.column_names)
+    def common_cols(columns_a, columns_b):
+        col_a = set(columns_a)
+        col_b = set(columns_b)
         return [col for col in col_a if col in col_b]
 
     if training_args.do_train: