Restructured in preparation for packaging

2023-06-01 16:16:13 -05:00 · 2023-06-01 16:16:13 -05:00 · 80a62a6e32
commit 80a62a6e32
parent d6c0130477
1 changed files with 2 additions and 2 deletions
--- a/src/mltraining/trainers.py
+++ b/src/mltraining/trainers.py
@ -57,7 +57,7 @@ class TokenClassificationTrainer:
        )

        labels = []
-        for i, label in enumerate(examples[f"ner_tags"]):
+        for i, label in enumerate(examples["ner_tags"]):
            word_ids = tokenized_inputs.word_ids(
                batch_index=i
            )  # Map tokens to their respective word.