Edit text for swarmplot legend

Author: JAnns98

dabest/plotter.py

@@ -398,7 +398,7 @@ def effectsize_df_plotter(effectsize_df, **plot_kwargs):
             legend_kwargs=legend_kwargs
             )
 
-    ########## WIP LEGENDS
+    # Add legend for swarmplot
     if not show_pairs and not proportional and color_col is not None and not show_delta2:
         if len(np.unique(swarm_legend_kwargs['index'])) > 1:
             legend_elements = []
@@ -407,8 +407,6 @@ def effectsize_df_plotter(effectsize_df, **plot_kwargs):
                                                 markerfacecolor=color, markersize=10))
             rawdata_axes.legend(handles=legend_elements, frameon=False)
 
-    ########## WIP LEGENDS
-
     # Plot aesthetic adjustments.
     og_ylim_raw = rawdata_axes.get_ylim()
     og_xlim_raw = rawdata_axes.get_xlim()

nbs/API/plotter.ipynb

@@ -457,7 +457,7 @@
     "            legend_kwargs=legend_kwargs\n",
     "            )\n",
     "\n",
-    "    ########## WIP LEGENDS\n",
+    "    # Add legend for swarmplot\n",
     "    if not show_pairs and not proportional and color_col is not None and not show_delta2:\n",
     "        if len(np.unique(swarm_legend_kwargs['index'])) > 1:\n",
     "            legend_elements = []\n",
@@ -466,8 +466,6 @@
     "                                                markerfacecolor=color, markersize=10))\n",
     "            rawdata_axes.legend(handles=legend_elements, frameon=False)\n",
     "\n",
-    "    ########## WIP LEGENDS\n",
-    "\n",
     "    # Plot aesthetic adjustments.\n",
     "    og_ylim_raw = rawdata_axes.get_ylim()\n",
     "    og_xlim_raw = rawdata_axes.get_xlim()\n",