fix/wrong key

plot.py

@@ -92,7 +92,7 @@ def make_plot(proj_dir, type_sequence, time_sequence, w, difficulty_distribution
         df2 = df2.append(
             pd.DataFrame({'retention': retention, 'expected time': time, 'd': d, 'r': optimal_retention}))
 
-    fig = px.line(df2, x="retention", y="expected time (second)", color='d', log_y=True)
+    fig = px.line(df2, x="retention", y="expected time", color='d', log_y=True)
 
     # print(f"\n-----suggested retention: {np.inner(difficulty_distribution_padding, optimal_retention_list):.2f}-----")
     suggested_retention_markdown = f"""# Suggested Retention: `{np.inner(difficulty_distribution_padding, optimal_retention_list):.2f}`"""