Skip to content

Commit

Permalink
to enable running headlessly with nlprun
Browse files Browse the repository at this point in the history
  • Loading branch information
Jemoka committed Jan 25, 2025
1 parent 3d356f3 commit acb64d2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion stanza/utils/datasets/coref/convert_ontonotes.py
Original file line number Diff line number Diff line change
Expand Up @@ -198,7 +198,7 @@ def process_dataset(short_name, ontonotes_path, coref_output_path):
# possible Singleton conlls that maybe in the folder
(Path(ontonotes_path) / "cache").mkdir(exist_ok=True)

dataset = load_dataset("conll2012_ontonotesv5", config_name, cache_dir=str(Path(ontonotes_path) / "cache"))
dataset = load_dataset("conll2012_ontonotesv5", config_name, cache_dir=str(Path(ontonotes_path) / "cache"), trust_remote_code=True)
for section, hf_name in SECTION_NAMES.items():
# for section, hf_name in [("test", "test")]:
print("Processing %s" % section)
Expand Down

0 comments on commit acb64d2

Please sign in to comment.