Two arguments missed in label_search.py

[Junhao Zhao]

Hello:
The first one is at line 70, config is not passed to utils.TriggerTemplatizer.init():

    templatizer = utils.TriggerTemplatizer(
        args.template,
        tokenizer,
        label_map=label_map,
        label_field=args.label_field,
        add_special_tokens=False
    )

The second one is at line 96, use_ctx is required:

train_dataset = utils.load_trigger_dataset(args.train, templatizer)

Regards.

[Robert L. Logan IV]

Thanks for catching this!