milvus/tests/benchmark/milvus_benchmark/env/__init__.py
binbin eff75c7701
Replace sdk source and merge tests and tests20 (#7182)
Signed-off-by: Binbin Lv <binbin.lv@zilliz.com>
2021-08-20 11:00:56 +08:00

15 lines
333 B
Python

import logging
from .helm import HelmEnv
from .docker import DockerEnv
from .local import LocalEnv
logger = logging.getLogger("milvus_benchmark.env")
def get_env(env_mode, deploy_mode=None):
return {
"helm": HelmEnv(deploy_mode),
"docker": DockerEnv(None),
"local": LocalEnv(None),
}.get(env_mode)