Spaces:
Runtime error
Runtime error
# Isolated functions to reload the leaderboard data and plot the results | |
from src.leaderboard_utils import filter_search, get_github_data | |
from src.plot_utils import split_models, compare_plots | |
#COMPONENTS TO RELOAD EVERY TIME | |
# leaderboard_table, dummy_leaderboard_table, | |
# open_models_selection, closed_models_selection, show_all, show_names, show_legend | |
#version_select, prev_table, dummy_prev_table | |
class ReloadData(): | |
''' | |
A class containing methods to reload the leaderboard data and plot the results | |
The methods return individual component values directly to use 'every' arg in the component | |
''' | |
def __init__(self): | |
print("Initializing Reload...........") | |
self.primary_leaderboard_df, self.version_dfs, self.version_names = get_github_data() | |
self.plot_df = self.primary_leaderboard_df[0] | |
self.models = list(self.plot_df[list(self.plot_df.columns)[0]].unique()) | |
print("Reload completed ....... Here's a reloaded dataframe for latest version") | |
print(self.primary_leaderboard_df) | |
def get_primary_leaderboard(self): | |
''' | |
Returns | |
self.primary_leaderboard_df[0]: Dataframe containing the primary leaderboard (laterst version of the benchmark results) | |
''' | |
return self.primary_leaderboard_df[0] | |
def get_open_models(self): | |
''' | |
Returns | |
open_models: Checkbox group containing the open models | |
''' | |
self.open_models, _ = split_models(self.models) | |
return self.open_models | |
def get_closed_models(self): | |
''' | |
Returns | |
closed_models: Checkbox group containing the closed models | |
''' | |
_, self.closed_models = split_models(self.models) | |
return self.closed_models | |
def get_plot_df(self): | |
''' | |
Returns | |
plot_df: Dataframe containing the results of latest version for plotting | |
''' | |
return self.plot_df | |
def get_version_names(self): | |
''' | |
Returns | |
version_names: List containing the versions of the benchmark results for dropdown selection | |
''' | |
return self.version_names | |
def get_version_df(self): | |
''' | |
Returns | |
version_dfs: Dataframe containing the benchmark results for all versions | |
''' | |
return self.version_dfs | |
def get_prev_df(self, name='initial'): | |
''' | |
Returns | |
prev_df: Dataframe containing the benchmark results for the previous versions (default = latest version) | |
''' | |
if name == 'initial': | |
name = self.version_names[0] | |
ind = self.version_names.index(name) | |
self.prev_df = self.version_dfs[ind] | |
return self.prev_df | |