public
/
ottertune
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Bug fixes

This commit is contained in:
arifiorino 2020-02-19 03:00:44 +00:00 committed by Dana Van Aken
parent a2a77f9a75
commit 96a7991766
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
server/website/website/tasks/async_tasks.py
View File

@ -11,6 +11,7 @@ import gpflow
import time import time
from pyDOE import lhs from pyDOE import lhs
from scipy.stats import uniform from scipy.stats import uniform
from pytz import timezone
from celery import shared_task, Task from celery import shared_task, Task
from celery.utils.log import get_task_logger from celery.utils.log import get_task_logger
@ -28,7 +29,8 @@ from analysis.gpr.predict import gpflow_predict
from analysis.preprocessing import Bin, DummyEncoder from analysis.preprocessing import Bin, DummyEncoder
from analysis.constraints import ParamConstraintHelper from analysis.constraints import ParamConstraintHelper
from django.utils.datetime_safe import datetime from django.utils.datetime_safe import datetime
from website.models import PipelineData, PipelineRun, Result, Workload, SessionKnob, MetricCatalog from website.models import (PipelineData, PipelineRun, Result, Workload, SessionKnob,
MetricCatalog, ExecutionTime)
from website import db from website import db
from website.types import PipelineTaskType, AlgorithmType, VarType from website.types import PipelineTaskType, AlgorithmType, VarType
from website.utils import DataUtil, JSONUtil from website.utils import DataUtil, JSONUtil
@ -801,7 +803,7 @@ def configuration_recommendation(recommendation_input):
LOG.debug('%s: Finished selecting the next config.\n\ndata=%s\n', LOG.debug('%s: Finished selecting the next config.\n\ndata=%s\n',
AlgorithmType.name(algorithm), JSONUtil.dumps(conf_map_res, pprint=True)) AlgorithmType.name(algorithm), JSONUtil.dumps(conf_map_res, pprint=True))
save_execution_time(start_ts, "configuration_recommendation", Result.objects.get(pk=result_id)) save_execution_time(start_ts, "configuration_recommendation", newest_result)
return conf_map_res return conf_map_res
@ -996,5 +998,5 @@ def map_workload(map_workload_input):
LOG.debug('%s: Finished mapping the workload.\n\ndata=%s\n', LOG.debug('%s: Finished mapping the workload.\n\ndata=%s\n',
AlgorithmType.name(algorithm), JSONUtil.dumps(target_data, pprint=True)) AlgorithmType.name(algorithm), JSONUtil.dumps(target_data, pprint=True))
save_execution_time(start_ts, "map_workload", Result.objects.get(pk=result_id)) save_execution_time(start_ts, "map_workload", newest_result)
return target_data, algorithm return target_data, algorithm

3
server/website/website/tasks/periodic_tasks.py
View File

@ -6,6 +6,7 @@
import copy import copy
import numpy as np import numpy as np
import time import time
from pytz import timezone
from celery import shared_task from celery import shared_task
from celery.utils.log import get_task_logger from celery.utils.log import get_task_logger
@ -19,7 +20,7 @@ from analysis.preprocessing import (Bin, get_shuffle_indices,
DummyEncoder, DummyEncoder,
consolidate_columnlabels) consolidate_columnlabels)
from django.utils.datetime_safe import datetime from django.utils.datetime_safe import datetime
from website.models import PipelineData, PipelineRun, Result, Workload from website.models import PipelineData, PipelineRun, Result, Workload, ExecutionTime
from website.settings import ENABLE_DUMMY_ENCODER, TIME_ZONE from website.settings import ENABLE_DUMMY_ENCODER, TIME_ZONE
from website.types import PipelineTaskType, WorkloadStatusType from website.types import PipelineTaskType, WorkloadStatusType
from website.utils import DataUtil, JSONUtil from website.utils import DataUtil, JSONUtil