mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-27 02:40:41 +00:00
25 lines
898 B
Python
25 lines
898 B
Python
|
from retriever import benchmark_indexing, benchmark_querying
|
||
|
|
from reader import benchmark_reader
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
|
||
|
|
parser.add_argument('--reader', default=False, action="store_true",
|
||
|
|
help='Perform Reader benchmarks')
|
||
|
|
parser.add_argument('--retriever_index', default=False, action="store_true",
|
||
|
|
help='Perform Retriever indexing benchmarks')
|
||
|
|
parser.add_argument('--retriever_query', default=False, action="store_true",
|
||
|
|
help='Perform Retriever querying benchmarks')
|
||
|
|
parser.add_argument('--ci', default=False, action="store_true",
|
||
|
|
help='Perform a smaller subset of benchmarks that are quicker to run')
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
if args.retriever_index:
|
||
|
|
benchmark_indexing(ci)
|
||
|
|
if args.retriever_query:
|
||
|
|
benchmark_querying(ci)
|
||
|
|
if args.retriever_reader:
|
||
|
|
benchmark_reader(ci)
|
||
|
|
|