graug commited on
Commit
c710424
·
verified ·
1 Parent(s): 8e4b914

Chess Challenge submission by graug

Browse files
Files changed (1) hide show
  1. tokenizer.py +3 -3
tokenizer.py CHANGED
@@ -273,13 +273,13 @@ class ChessTokenizer(PreTrainedTokenizer):
273
  grouped_tokens = ["".join(tokens[i:i+3]) for i in range(0, len(tokens), 3)]
274
 
275
  # mettre une fois sur deux B avant les coups noirs sinon W
276
- final_tokens = grouped_tokens.copy()
277
- """
278
  for i, token in enumerate(grouped_tokens):
279
  if i % 2 == 0:
280
  final_tokens.append('W' + token)
281
  else:
282
- final_tokens.append('B' + token)"""
283
 
284
  return " ".join(final_tokens)
285
 
 
273
  grouped_tokens = ["".join(tokens[i:i+3]) for i in range(0, len(tokens), 3)]
274
 
275
  # mettre une fois sur deux B avant les coups noirs sinon W
276
+ final_tokens = []
277
+
278
  for i, token in enumerate(grouped_tokens):
279
  if i % 2 == 0:
280
  final_tokens.append('W' + token)
281
  else:
282
+ final_tokens.append('B' + token)
283
 
284
  return " ".join(final_tokens)
285