Max number of arch adjustments

app.py

@@ -19,8 +19,14 @@ def merge_columns(dataset: Dataset, max_number_of_columns: int):
     to_merge = list(sorted_downloads.keys())[max_number_of_columns:]
     to_keep = list(sorted_downloads.keys())[:max_number_of_columns]
 
+    if 'no_arch' in to_merge:
+        to_merge.remove('no_arch')
+
+    if 'no_arch' in to_keep:
+        to_keep.remove('no_arch')
+
     dictionary = dataset.to_dict()
-    dictionary['combined'] = dictionary.pop('no_arch')
+    dictionary['combined'] = dictionary['no_arch']
 
     while len(to_merge):
         current = dictionary['combined']
@@ -42,12 +48,12 @@ def merge_columns(dataset: Dataset, max_number_of_columns: int):
     return dataset
 
 
-def plot(library: str, stacked: bool, number_of_pipelines_to_show: int):
+def plot(library: str, stacked: bool, number_of_architectures_to_show: int):
     dataset = load_dataset(f"open-source-metrics/{library}-checkpoint-downloads")['train']
     n_archs = len(dataset.column_names) - 1  # Remove dates
 
-    if n_archs > number_of_pipelines_to_show:
-        dataset = merge_columns(dataset, number_of_pipelines_to_show)
+    if n_archs > number_of_architectures_to_show:
+        dataset = merge_columns(dataset, number_of_architectures_to_show)
 
     dates = dataset['dates']
     axis = dataset.column_names
@@ -58,8 +64,6 @@ def plot(library: str, stacked: bool, number_of_pipelines_to_show: int):
         fig.add_trace(
             go.Scatter(x=dates, y=dataset[i], mode='lines+markers', name=i, stackgroup='one' if stacked else None)
         )
-
-    fig.show()
     return fig
 
 
@@ -67,7 +71,7 @@ with gr.Blocks() as demo:
     inputs = [
         gr.Dropdown(pipelines),
         gr.Checkbox(label='Stacked'),
-        gr.Slider(minimum=1, maximum=len(pipelines), value=3, step=1, label="Max number of pipelines to show")
+        gr.Slider(minimum=1, maximum=len(pipelines), value=3, step=1, label="Max number of architectures to show")
     ]
     submit = gr.Button('Submit')
     with gr.Row():