mirror of https://github.com/milvus-io/milvus.git
27 lines
1.2 KiB
Python
27 lines
1.2 KiB
Python
from .insert import InsertRunner, BPInsertRunner
|
|
from .locust import LocustInsertRunner, LocustSearchRunner, LocustRandomRunner
|
|
from .search import SearchRunner, InsertSearchRunner
|
|
from .build import BuildRunner, InsertBuildRunner
|
|
from .get import InsertGetRunner
|
|
from .accuracy import AccuracyRunner
|
|
from .accuracy import AccAccuracyRunner
|
|
from .chaos import SimpleChaosRunner
|
|
|
|
|
|
def get_runner(name, env, metric):
|
|
return {
|
|
"insert_performance": InsertRunner(env, metric),
|
|
"bp_insert_performance": BPInsertRunner(env, metric),
|
|
"search_performance": SearchRunner(env, metric),
|
|
"insert_search_performance": InsertSearchRunner(env, metric),
|
|
"locust_insert_performance": LocustInsertRunner(env, metric),
|
|
"locust_search_performance": LocustSearchRunner(env, metric),
|
|
"locust_random_performance": LocustRandomRunner(env, metric),
|
|
"insert_build_performance": InsertBuildRunner(env, metric),
|
|
"insert_get_performance": InsertGetRunner(env, metric),
|
|
"build_performance": BuildRunner(env, metric),
|
|
"accuracy": AccuracyRunner(env, metric),
|
|
"ann_accuracy": AccAccuracyRunner(env, metric),
|
|
"simple_chaos": SimpleChaosRunner(env, metric)
|
|
}.get(name)
|