ucinlp / autoprompt

AutoPrompt: Automatic Prompt Construction for Masked Language Models.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

TypeErrors in label_search.py

olenmg opened this issue · comments

Hello, thanks for great codebase!

in autoprompt/label_search.py, I found that there are some TypeErrors(missing arguments).

  1. use_ctx is not denoted when loading dataset
  2. when initializing utils.TriggerTemplatizer, 'config' argument is missing.

First one can be fixed as just add 'use_ctx=False' since label_search.py is only used for sentiment analysis and NLI tasks. (not for relation extraction)

Fixed in PR #31 .

Thank you again.