Update src/reload_utils.py

src/reload_utils.py

@@ -3,80 +3,75 @@
 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
+def get_primary_leaderboard():
+    '''
+    Returns
+        primary_leaderboard_df[0]: Dataframe containing the primary leaderboard (laterst version of the benchmark results)
+    '''
+    primary_leaderboard_df, _, _ = get_github_data()
+    print("Reloading primary df: ")
+    print(primary_leaderboard_df)
+    return primary_leaderboard_df[0]
 
-class ReloadData():
+def get_open_models():
     '''
-    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
+    Returns
+        open_models: Checkbox group containing the open models
     '''
+    primary_leaderboard_df, _, _ = get_github_data()
+    temp_df = primary_leaderboard_df[0]
+    models = list(temp_df[list(temp_df.columns)[0]].unique())
+    open_models, _ = split_models(models)
+    return open_models
 
-    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_closed_models():
+    '''
+    Returns
+        closed_models: Checkbox group containing the closed models
+    '''
+    primary_leaderboard_df, _, _ = get_github_data()
+    temp_df = primary_leaderboard_df[0]
+    models = list(temp_df[list(temp_df.columns)[0]].unique())
+    _, closed_models = split_models(models)
+    return closed_models
 
-    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):      
+def get_plot_df():      
         '''
         Returns
             plot_df: Dataframe containing the results of latest version for plotting
         '''
-        return self.plot_df
+        primary_leaderboard_df, _, _ = get_github_data()
+        plot_df = primary_leaderboard_df[0]
+        return 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_version_names():
+    '''
+    Returns
+        version_names: List containing the versions of the benchmark results for dropdown selection
+    '''
+    _, _, version_names = get_github_data()
+    return version_names
     
-    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]
+def get_version_df():
+    '''
+    Returns
+        version_dfs: Dataframe containing the benchmark results for all versions
+    '''
+    _, version_dfs, _ = get_github_data()
+    return version_dfs
 
-        ind = self.version_names.index(name)
-        self.prev_df = self.version_dfs[ind]
-        return self.prev_df
+def get_prev_df(name='initial'):
+    '''
+    Returns
+        prev_df: Dataframe containing the benchmark results for the previous versions (default = latest version)
+    '''
+    _, version_dfs, version_names = get_github_data()
+    
+    if name == 'initial':
+        name = version_names[0]
 
+    ind = version_names.index(name)
+    prev_df = version_dfs[ind]
+    return prev_df