improve formatting

invrs-io · Jul 11, 2024 · 681ee89 · 681ee89
1 parent c33da9b
commit 681ee89
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 7 deletions.
diff --git a/docs/notebooks/ceviche_challenges.ipynb b/docs/notebooks/ceviche_challenges.ipynb
@@ -25,10 +25,15 @@
    },
    "outputs": [],
    "source": [
-    "from invrs_gym import challenges\n",
+    "import os\n",
+    "import contextlib\n",
+    "\n",
+    "with open(os.devnull, \"w\") as f, contextlib.redirect_stdout(f):\n",
+    "    # Avoid printing by `refractiveindex` module on first run.\n",
+    "    from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.ceviche_beam_splitter().eval_metric.__doc__\n",
-    "print(\"\\n\".join(docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]))"
+    "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"
    ]
   },
   {

diff --git a/docs/notebooks/diffractive_splitter.ipynb b/docs/notebooks/diffractive_splitter.ipynb
@@ -28,7 +28,7 @@
     "from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.diffractive_splitter().eval_metric.__doc__\n",
-    "print(\"\\n\".join(docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]))"
+    "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"
    ]
   },
   {

diff --git a/docs/notebooks/meta_atom_library.ipynb b/docs/notebooks/meta_atom_library.ipynb
@@ -28,7 +28,7 @@
     "from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.meta_atom_library().eval_metric.__doc__\n",
-    "print(\"\\n\".join(docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]))"
+    "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"
    ]
   },
   {

diff --git a/docs/notebooks/metagrating.ipynb b/docs/notebooks/metagrating.ipynb
@@ -28,7 +28,7 @@
     "from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.metagrating().eval_metric.__doc__\n",
-    "print(\"\\n\".join(docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]))"
+    "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"
    ]
   },
   {

diff --git a/docs/notebooks/metalens.ipynb b/docs/notebooks/metalens.ipynb
@@ -28,7 +28,7 @@
     "from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.metalens().eval_metric.__doc__\n",
-    "print(\"\\n\".join(docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]))"
+    "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"
    ]
   },
   {

diff --git a/docs/notebooks/photon_extractor.ipynb b/docs/notebooks/photon_extractor.ipynb
@@ -28,7 +28,7 @@
     "from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.photon_extractor().eval_metric.__doc__\n",
-    "print(\"\\n\".join(docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]))"
+    "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"
    ]
   },
   {