[source_epochs_fix] add support to epoch on source

leondavi · Jun 8, 2024 · 8e69b4c · 8e69b4c
1 parent 837cd67
commit 8e69b4c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src_py/apiServer/transmitter.py b/src_py/apiServer/transmitter.py
@@ -75,9 +75,10 @@ def update_csv(self, csv_files: list, source_pieces: list):
             target_workers = source_piece.get_target_workers()
             num_of_batches = source_piece.get_num_of_batches()
             nerltensor_type = source_piece.get_nerltensor_type()
+            phase_type = source_piece.get_phase()
             with open(csv_file, 'r') as file:
                 csvfile = file.read()
-                data_str = f'{index + 1}#{total_sources}#{source_name}#{target_workers}#{num_of_batches}#{nerltensor_type}#{csvfile}'
+                data_str = f'{index + 1}#{total_sources}#{source_name}#{target_workers}#{phase_type}#{num_of_batches}#{nerltensor_type}#{csvfile}'
                 try:
                     response = requests.post(self.updateCSVAddress, data = data_str)
                     if not response.ok: