Az-r-ow commited on
Commit
0747c82
·
1 Parent(s): b42f9d4

refactor(data_generation): raising exception in script if chars written != actual chars

Browse files
Files changed (1) hide show
  1. data/scripting_lcs_1/script.py +3 -1
data/scripting_lcs_1/script.py CHANGED
@@ -95,7 +95,9 @@ def generate_data(cities: List, file_out: str, nb_samples: int):
95
  new_line = line.replace("{depart}", departure_city)
96
  new_line = new_line.replace("{arrivee}", arrival_city)
97
  try:
98
- f_sortie.write(new_line)
 
 
99
  line_count += 1
100
  except Exception as e:
101
  print(e)
 
95
  new_line = line.replace("{depart}", departure_city)
96
  new_line = new_line.replace("{arrivee}", arrival_city)
97
  try:
98
+ n_chars_written = f_sortie.write(new_line)
99
+ if n_chars_written != len(new_line):
100
+ raise Exception("Error while writing line")
101
  line_count += 1
102
  except Exception as e:
103
  print(e)