From 93d6fa53395cd7ba71502a56c8aec561df45c5b8 Mon Sep 17 00:00:00 2001 From: yangdsh Date: Wed, 22 Apr 2020 14:58:31 +0000 Subject: [PATCH] fix unit test --- server/website/website/tasks/periodic_tasks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/server/website/website/tasks/periodic_tasks.py b/server/website/website/tasks/periodic_tasks.py index 90842d9..d247a86 100644 --- a/server/website/website/tasks/periodic_tasks.py +++ b/server/website/website/tasks/periodic_tasks.py @@ -252,7 +252,7 @@ def aggregate_data(wkld_results): return knob_data, metric_data -def run_workload_characterization(metric_data, dbms): +def run_workload_characterization(metric_data, dbms=None): # Performs workload characterization on the metric_data and returns # a set of pruned metrics. # @@ -268,7 +268,7 @@ def run_workload_characterization(metric_data, dbms): columnlabels = metric_data['columnlabels'] LOG.debug("Workload characterization ~ initial data size: %s", matrix.shape) - views = VIEWS_FOR_PRUNING.get(dbms.type, None) + views = None if dbms is None else VIEWS_FOR_PRUNING.get(dbms.type, None) if views is not None: useful_labels = [] for label in columnlabels: