style: ruff format

AnswerDotAI · Feb 28, 2024 · 7d3ff09 · 7d3ff09
1 parent faa9eb0
commit 7d3ff09
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/ragatouille/RAGTrainer.py b/ragatouille/RAGTrainer.py
@@ -157,7 +157,9 @@ def prepare_training_data(
         )
         if len(self.data_processor.training_triplets) == 0:
             if mine_hard_negatives:
-                print("Warning: No training triplets were generated with setting mine_hard_negatives=='True'. This may be due to the data being too small or the hard negative miner not being able to find enough hard negatives.")
+                print(
+                    "Warning: No training triplets were generated with setting mine_hard_negatives=='True'. This may be due to the data being too small or the hard negative miner not being able to find enough hard negatives."
+                )
                 self.data_processor.process_raw_data(
                     data_type=data_type,
                     raw_data=raw_data,

diff --git a/ragatouille/data/training_data_processor.py b/ragatouille/data/training_data_processor.py
@@ -125,7 +125,9 @@ def _get_new_negatives(self, query, passages, mine_hard_negatives, n_new_negativ
         else:
             new_negatives = [
                 x
-                for x in random.sample(self.collection, min(n_new_negatives, len(self.collection)))
+                for x in random.sample(
+                    self.collection, min(n_new_negatives, len(self.collection))
+                )
                 if x not in passages["positives"] and x not in passages["negatives"]
             ]