Updated errorbar handling of xspans to properly account for when hue column is being used to plot for color_col is not None

Author: sunroofgod

nbs/API/plotter.ipynb

@@ -347,7 +347,6 @@
     "        for i in range(len(idx)):\n",
     "            filled.append(False)\n",
     "            filled.extend([True] * (len(idx[i]) - 1))\n",
-    "        print(f\"filled is {filled}\")\n",
     "\n",
     "    names = color_groups if not color_by_subgroups else idx\n",
     "    n_groups = len(color_groups)\n",
@@ -360,7 +359,6 @@
     "        unsat_colors = sns.color_palette(n_colors=n_groups)\n",
     "        if empty_circle and not color_by_subgroups:\n",
     "            unsat_colors = [sns.color_palette(\"gray\")[3]] + unsat_colors\n",
-    "            print(unsat_colors)\n",
     "    else:\n",
     "        if isinstance(custom_pal, dict):\n",
     "            groups_in_palette = {\n",
@@ -823,13 +821,13 @@
     "                if bootstraps_color_by_group:\n",
     "                    line_colors.append(plot_palette_raw[all_plot_groups[jj]])\n",
     "\n",
-    "                # Break the loop since hue in Seaborn adds collections to axes and it will result in index out of range\n",
-    "                if jj >= n_groups - 1 and color_col is None:\n",
-    "                    break\n",
-    "\n",
     "            if len(line_colors) != len(all_plot_groups):\n",
     "                line_colors = ytick_color\n",
     "\n",
+    "            # hue in swarmplot would add collections to axes which will result in len(xspans) = len(all_plot_groups) + len(unique groups in hue)\n",
+    "            if len(xspans) > len(all_plot_groups):\n",
+    "                xspans = xspans[:len(all_plot_groups)]\n",
+    "\n",
     "            error_bar(\n",
     "                plot_data,\n",
     "                x=xvar,\n",