google-research / google-research/language
minor typo on preprocess_util of bert_extraction
Open
- Dominant language
- Python
- Stars
- 1.8k
- Forks
- 362
- PR merge metrics
- No merged PRs in 30d
Description
Hi,
When I run the bert_extraction of SST-2, there is a minor typo that gives an error : 'KeyError: 'sst-2''
In the file:
language/language/bert_extraction/steal_bert_classifier/data_generation/preprocess_util.py
the task_input_indices has some error (task name : 'sst2' which should be 'sst-2')
Could you check this? Thanks.
Best regards,
Contributor guide
Assessment
This issue has not been assessed yet.