fix yapf check

InternLM · Jul 9, 2024 · a222889 · a222889
1 parent e999025
commit a222889
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/xtuner/engine/hooks/dataset_info_hook.py b/xtuner/engine/hooks/dataset_info_hook.py
@@ -25,7 +25,8 @@ def __init__(self, tokenizer, is_intern_repo_dataset=False):
         self.is_intern_repo_dataset = is_intern_repo_dataset
 
     def log(self, runner, dataset, mode='train'):
-        def _log(input_ids, log_prefix=""):
+
+        def _log(input_ids, log_prefix=''):
             if self.is_intern_repo_dataset:
                 input_ids = [abs(x) for x in input_ids]
             # Try to split list to be compatible with IMAGE token
@@ -40,8 +41,8 @@ def _log(input_ids, log_prefix=""):
         runner.logger.info(f'Num {mode} samples {len(dataset)}')
         runner.logger.info(f'{mode} example:')
         if 'chosen_ids' in dataset[0]:
-            _log(dataset[0]['chosen_ids'], log_prefix="chosen: ")
-            _log(dataset[0]['rejected_ids'], log_prefix="rejected: ")
+            _log(dataset[0]['chosen_ids'], log_prefix='chosen: ')
+            _log(dataset[0]['rejected_ids'], log_prefix='rejected: ')
         else:
             _log(dataset[0]['input_ids'])