Chess Challenge submission by graug
Browse files- 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 =
|
| 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 |
|