removing baseline results not inplace

app.py

@@ -307,14 +307,14 @@ def get_figure(results, dataset, gender, indie_var_name, include_baseline=True):
     colors = ['b', 'g', 'c', 'm', 'y', 'r', 'k']  # assert no
 
     # Grab then remove baselines from df
-    baseline = results.loc[BASELINE_MARKER]
-    results.drop(index=BASELINE_MARKER, axis=1, inplace=True)
+    results_to_plot = results.drop(index=BASELINE_MARKER, axis=1)
     
     fig, ax = plt.subplots()
     for i, col in enumerate(results.columns):
-        ax.plot(results[col],  color=colors[i])#, color=colors)
+        ax.plot(results_to_plot[col],  color=colors[i])#, color=colors)
 
     if include_baseline == True:
+        baseline = results.loc[BASELINE_MARKER]
         for i, (name, value) in enumerate(baseline.items()):
             if name == indie_var_name:
                 continue
@@ -327,7 +327,7 @@ def get_figure(results, dataset, gender, indie_var_name, include_baseline=True):
         ax.set_xlabel("Date injected into input text")
     ax.set_title(f"Softmax probability of pronouns predicted {gender}\n by model type vs {indie_var_name}.")
     ax.set_ylabel(f"Avg softmax prob for {gender} pronouns")
-    ax.legend(list(results.columns))
+    ax.legend(list(results_to_plot.columns))
     return fig