tarzanagh commited on
Commit
4a4e780
·
verified ·
1 Parent(s): 124576e

Pass --max_documents to passage-entity

Browse files
Files changed (1) hide show
  1. benchmarks/run.py +2 -0
benchmarks/run.py CHANGED
@@ -136,6 +136,8 @@ def run_passage_entity(args):
136
  if args.force_build:
137
  sub_argv.append("--force_index")
138
  sub_argv.append("--force_openie")
 
 
139
 
140
  old_argv = sys.argv
141
  sys.argv = sub_argv
 
136
  if args.force_build:
137
  sub_argv.append("--force_index")
138
  sub_argv.append("--force_openie")
139
+ if args.max_documents:
140
+ sub_argv.extend(["--max_documents", str(args.max_documents)])
141
 
142
  old_argv = sys.argv
143
  sys.argv = sub_argv