Pass --max_documents to passage-entity
Browse files- 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
|