Skip to content

Commit

Permalink
Black formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
harrykeightley committed Oct 17, 2023
1 parent 2f0df2f commit b8d49bb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions elpis/trainer/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -218,8 +218,8 @@ def create_vocab_from_batches(batches: Iterable[dict[str, Any]]) -> Vocab:
vocabs = map(create_vocab_from_batches, splits)
vocab = reduce(Vocab.merge, vocabs, Vocab({}))
else:
# Note: in this case the values() are datasets, which are treated
# as if they were batches.
# Note: in this case the values() are datasets, which are treated
# as if they were batches.
vocab = create_vocab_from_batches(vocab_datasets.values())

vocab.add(job.data_args.unk_token)
Expand Down

0 comments on commit b8d49bb

Please sign in to comment.