Add missing "self.did_contender_win" to rl_loop.py #180

QueensGambit · Jul 1, 2022 · d450309 · d450309
1 parent 5f05795
commit d450309
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/engine/src/rl/rl_loop.py b/engine/src/rl/rl_loop.py
@@ -121,7 +121,8 @@ def check_for_enough_train_data(self, number_files_to_update):
         """
         if self.file_io.get_number_generated_files() >= number_files_to_update:
             self.binary_io.stop_process()
-            self.file_io.prepare_data_for_training(self.rl_config.rm_nb_files, self.rl_config.rm_fraction_for_selection)
+            self.file_io.prepare_data_for_training(self.rl_config.rm_nb_files, self.rl_config.rm_fraction_for_selection,
+                                                   self.did_contender_win)
             # start training using a process to ensure memory clearing afterwards
             queue = Queue()  # start a subprocess to be memory efficient
             self.tc.device_id = self.args.device_id