ex 2.9.1, 2.10.1: reconfigured to run a sweep in Modal and pick the best run

Author: z0u

docs/m2-control/ex-2.9.1-redux.ipynb

@@ -922,7 +922,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 28,
+   "execution_count": 29,
    "id": "cd20e021",
    "metadata": {},
    "outputs": [
@@ -937,9 +937,9 @@
      "data": {
       "text/html": [
        "<picture>\n",
-       "  <source srcset=\"large-assets/ex-2.9.1-results-baseline.dark.png?v=sXixHLMYNQQs3poGuJmZRl3OpBCic_K_bV1M0m3MEY0\" media=\"(prefers-color-scheme: dark)\" />\n",
-       "  <source srcset=\"large-assets/ex-2.9.1-results-baseline.png?v=sXixHLMYNQQs3poGuJmZRl3OpBCic_K_bV1M0m3MEY0\" media=\"(prefers-color-scheme: light)\" />\n",
-       "  <img src=\"large-assets/ex-2.9.1-results-baseline.png?v=sXixHLMYNQQs3poGuJmZRl3OpBCic_K_bV1M0m3MEY0\" alt=\"Composite figure with two latent panels (top), a color slice (middle), and a loss chart (bottom).\" />\n",
+       "  <source srcset=\"large-assets/ex-2.9.1-results-baseline.dark.png?v=zi3OpaE5qc-z6e4HxYjzRpwfHPz1_mb60yCqcVRwQOs\" media=\"(prefers-color-scheme: dark)\" />\n",
+       "  <source srcset=\"large-assets/ex-2.9.1-results-baseline.png?v=zi3OpaE5qc-z6e4HxYjzRpwfHPz1_mb60yCqcVRwQOs\" media=\"(prefers-color-scheme: light)\" />\n",
+       "  <img src=\"large-assets/ex-2.9.1-results-baseline.png?v=zi3OpaE5qc-z6e4HxYjzRpwfHPz1_mb60yCqcVRwQOs\" alt=\"Composite figure with two latent panels (top), a color slice (middle), and a loss chart (bottom).\" />\n",
        "</picture>"
       ],
       "text/plain": [
@@ -960,9 +960,9 @@
      "data": {
       "text/html": [
        "<picture>\n",
-       "  <source srcset=\"large-assets/ex-2.9.1-results-ablated.dark.png?v=Uddyo7HUrfUvFKOSYoFS1pEOuVjlVnwUZ7ZVZDonwhw\" media=\"(prefers-color-scheme: dark)\" />\n",
-       "  <source srcset=\"large-assets/ex-2.9.1-results-ablated.png?v=Uddyo7HUrfUvFKOSYoFS1pEOuVjlVnwUZ7ZVZDonwhw\" media=\"(prefers-color-scheme: light)\" />\n",
-       "  <img src=\"large-assets/ex-2.9.1-results-ablated.png?v=Uddyo7HUrfUvFKOSYoFS1pEOuVjlVnwUZ7ZVZDonwhw\" alt=\"Composite figure with two latent panels (top), a color slice (middle), and a loss chart (bottom).\" />\n",
+       "  <source srcset=\"large-assets/ex-2.9.1-results-ablated.dark.png?v=9lD5_z_DrfyE9SsX95I2oW7M1MklkH4fFeqkMjJjOFo\" media=\"(prefers-color-scheme: dark)\" />\n",
+       "  <source srcset=\"large-assets/ex-2.9.1-results-ablated.png?v=9lD5_z_DrfyE9SsX95I2oW7M1MklkH4fFeqkMjJjOFo\" media=\"(prefers-color-scheme: light)\" />\n",
+       "  <img src=\"large-assets/ex-2.9.1-results-ablated.png?v=9lD5_z_DrfyE9SsX95I2oW7M1MklkH4fFeqkMjJjOFo\" alt=\"Composite figure with two latent panels (top), a color slice (middle), and a loss chart (bottom).\" />\n",
        "</picture>"
       ],
       "text/plain": [
@@ -983,9 +983,9 @@
      "data": {
       "text/html": [
        "<picture>\n",
-       "  <source srcset=\"large-assets/ex-2.9.1-results-pruned.dark.png?v=3Oqgh5B76V4S5VQlURFDcNQ9LZkRSDr1GhL4HUNkblc\" media=\"(prefers-color-scheme: dark)\" />\n",
-       "  <source srcset=\"large-assets/ex-2.9.1-results-pruned.png?v=3Oqgh5B76V4S5VQlURFDcNQ9LZkRSDr1GhL4HUNkblc\" media=\"(prefers-color-scheme: light)\" />\n",
-       "  <img src=\"large-assets/ex-2.9.1-results-pruned.png?v=3Oqgh5B76V4S5VQlURFDcNQ9LZkRSDr1GhL4HUNkblc\" alt=\"Composite figure with two latent panels (top), a color slice (middle), and a loss chart (bottom).\" />\n",
+       "  <source srcset=\"large-assets/ex-2.9.1-results-pruned.dark.png?v=KLNBSjmKm189e0UZFLRQyrZM6IJp6VuVOoYRQFGRwQM\" media=\"(prefers-color-scheme: dark)\" />\n",
+       "  <source srcset=\"large-assets/ex-2.9.1-results-pruned.png?v=KLNBSjmKm189e0UZFLRQyrZM6IJp6VuVOoYRQFGRwQM\" media=\"(prefers-color-scheme: light)\" />\n",
+       "  <img src=\"large-assets/ex-2.9.1-results-pruned.png?v=KLNBSjmKm189e0UZFLRQyrZM6IJp6VuVOoYRQFGRwQM\" alt=\"Composite figure with two latent panels (top), a color slice (middle), and a loss chart (bottom).\" />\n",
        "</picture>"
       ],
       "text/plain": [
@@ -1006,9 +1006,9 @@
      "data": {
       "text/html": [
        "<picture>\n",
-       "  <source srcset=\"large-assets/ex-2.9.1-results-suppression.dark.png?v=jsgx3KqVhASpLk3J6pS6nPgH8Qb7Odwk1jwaaKa6xuw\" media=\"(prefers-color-scheme: dark)\" />\n",
-       "  <source srcset=\"large-assets/ex-2.9.1-results-suppression.png?v=jsgx3KqVhASpLk3J6pS6nPgH8Qb7Odwk1jwaaKa6xuw\" media=\"(prefers-color-scheme: light)\" />\n",
-       "  <img src=\"large-assets/ex-2.9.1-results-suppression.png?v=jsgx3KqVhASpLk3J6pS6nPgH8Qb7Odwk1jwaaKa6xuw\" alt=\"Composite figure with two latent panels (top), a color slice (middle), and a loss chart (bottom).\" />\n",
+       "  <source srcset=\"large-assets/ex-2.9.1-results-suppression.dark.png?v=PMXbPhfJ-WOpUj87s5NyDyjjU82W0wfVQuXtRwMyBCU\" media=\"(prefers-color-scheme: dark)\" />\n",
+       "  <source srcset=\"large-assets/ex-2.9.1-results-suppression.png?v=PMXbPhfJ-WOpUj87s5NyDyjjU82W0wfVQuXtRwMyBCU\" media=\"(prefers-color-scheme: light)\" />\n",
+       "  <img src=\"large-assets/ex-2.9.1-results-suppression.png?v=PMXbPhfJ-WOpUj87s5NyDyjjU82W0wfVQuXtRwMyBCU\" alt=\"Composite figure with two latent panels (top), a color slice (middle), and a loss chart (bottom).\" />\n",
        "</picture>"
       ],
       "text/plain": [
@@ -1033,7 +1033,7 @@
     ")\n",
     "\n",
     "dims = ((3, 0, 1), (1, 2, 0))\n",
-    "pruned_dims = tuple(d - 1 if d > 0 else None for d in dims[0]), tuple(d - 1 if d > 0 else None for d in dims[1])\n",
+    "pruned_dims = ((2, None, 0), (0, 1, None))\n",
     "\n",
     "print('Baseline')\n",
     "viz.plot_stacked_results(\n",
@@ -1060,7 +1060,6 @@
     "viz.plot_stacked_results(\n",
     "    suppression_results,\n",
     "    latent_dims=dims,\n",
-    "    # latent_dims=((1, 0, 2), (1, 2, 0)),\n",
     "    max_error=max_error,\n",
     "    latent_annotations=[\n",
     "        ThemedAnnotation(direction=RED, angle=2 * (np.pi / 2 - falloff.a), dashed=True),\n",

src/ex_color/vis/helpers.py

@@ -251,7 +251,7 @@ def draw_boxplot(
     # else:
     #     ax.ticklabel_format(style='sci', axis='x', scilimits=(0, 0), useMathText=True)
     if xlim is not None:
-        ax.set_xlim(xlim)
+        ax.set_xlim(*xlim)
     return fig