diff --git a/libs/utils/wa_results_collector.py b/libs/utils/wa_results_collector.py index 77474b7b6362c0e3b1e90d71499c997461b2db6c..a1a4eeaaf8305afa50f59f85ce2d5a6d4ad9864d 100644 --- a/libs/utils/wa_results_collector.py +++ b/libs/utils/wa_results_collector.py @@ -23,6 +23,7 @@ import pandas as pd import subprocess import logging import warnings +import sqlite3 from scipy.stats import ttest_ind import matplotlib.cm as cm @@ -254,6 +255,10 @@ class WaResultsCollector(object): # If not, some workloads have a 'test' workload_parameter, try # using that test = job['workload_parameters']['test'] + elif 'test_ids' in job['workload_parameters']: + # If not, some workloads have a 'test_ids' workload_parameter, try + # using that + test = job['workload_parameters']['test_ids'] else: # Otherwise just use the workload name. # This isn't ideal because it means the results from jobs with @@ -446,6 +451,14 @@ class WaResultsCollector(object): df.loc[:, 'metric'] = 'frame_total_duration' df.loc[:, 'units'] = 'ms' + extra_metric_list.append(df) + elif 'jankbench-results' in artifacts: + con = sqlite3.connect(artifacts['jankbench-results']) + df = pd.read_sql_query("SELECT _id, name, run_id, iteration, total_duration, jank_frame from ui_results", con) + df = pd.DataFrame({'value': df['total_duration']}) + df.loc[:, 'metric'] = 'frame_total_duration' + df.loc[:, 'units'] = 'ms' + extra_metric_list.append(df) # WA's metrics model just exports overall energy metrics, not individual