wasiahmad / PLBART

Official code of our work, Unified Pre-training for Program Understanding and Generation [NAACL 2021].
https://arxiv.org/abs/2103.06333
MIT License
186 stars 35 forks source link

Change js to javascript in convert.py #27

Closed gchhablani closed 2 years ago

gchhablani commented 2 years ago

The current script leads to an error:

Traceback (most recent call last):
  File "convert.py", line 102, in <module>
    sanity_check(args.output_dir, args.output_ckpt)
  File "convert.py", line 75, in sanity_check
    assert bart.task.source_dictionary[50004] == '__js__'
AssertionError

Changing it to __javascript__ fixes it.