Skip to content
Snippets Groups Projects

Parse a tokens yaml instead of hardcoding the tokens

Merged Thibault Lavigne requested to merge parse-tokens into main
All threads resolved!
+ 1
1
@@ -359,7 +359,7 @@ def process_batch(
text = result["text"]
tokens = parse_tokens(tokens)
start_tokens, end_tokens = zip(*list(tokens.values()))
end_tokens = list(filter(lambda item: item != "", end_tokens))
end_tokens = list(filter(bool, end_tokens))
if len(end_tokens):
assert len(start_tokens) == len(
Loading