saattrupdan commited on
Commit
a73e53c
·
1 Parent(s): 576340d

feat: Fix colour for each model (up to retakes), reduce logging

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -264,9 +264,9 @@ def main() -> None:
264
  f"{min_colour_distance}."
265
  )
266
 
267
- random.seed(4242 + i)
268
  retries_left = 10 * len(all_models)
269
  for model_id in all_models:
 
270
  r, g, b = 0, 0, 0
271
  too_bright, similar_to_other_model = True, True
272
  while (too_bright or similar_to_other_model) and retries_left > 0:
@@ -279,7 +279,6 @@ def main() -> None:
279
  for colour in colour_mapping.values()
280
  )
281
  retries_left -= 1
282
- logger.info(f"Retries left to find a colour mapping: {retries_left}")
283
  colour_mapping[model_id] = (r, g, b)
284
 
285
  if retries_left:
 
264
  f"{min_colour_distance}."
265
  )
266
 
 
267
  retries_left = 10 * len(all_models)
268
  for model_id in all_models:
269
+ random.seed(hash(model_id) + i)
270
  r, g, b = 0, 0, 0
271
  too_bright, similar_to_other_model = True, True
272
  while (too_bright or similar_to_other_model) and retries_left > 0:
 
279
  for colour in colour_mapping.values()
280
  )
281
  retries_left -= 1
 
282
  colour_mapping[model_id] = (r, g, b)
283
 
284
  if retries_left: