diff options
Diffstat (limited to 'qtip/loader/plan.py')
-rw-r--r-- | qtip/loader/plan.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/qtip/loader/plan.py b/qtip/loader/plan.py index 0fd9ff52..e15651a3 100644 --- a/qtip/loader/plan.py +++ b/qtip/loader/plan.py @@ -9,12 +9,20 @@ from qtip.base.constant import BaseProp -from qtip.collector.base import CollectorProp as CProp -from qtip.loader.module import load_collector +from qtip.collector import CollectorProp as CProp +from qtip.collector.logfile import LogfileCollector from qtip.loader.yaml_file import YamlFileLoader from qtip.loader.qpi import QPISpec +# TODO(yujunz) more elegant way to load module dynamically +def load_collector(type_name): + if type_name == LogfileCollector.TYPE: + return LogfileCollector + else: + raise Exception("Invalid collector type: {}".format(type_name)) + + class Plan(YamlFileLoader): """ a benchmark plan is consist of configuration and a QPI list |