fix units

lincc-frameworks · Aug 8, 2024 · 2782bbc · 2782bbc
1 parent cf7ee62
commit 2782bbc
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/docs/tutorials/nested_spectra.ipynb b/docs/tutorials/nested_spectra.ipynb
@@ -81,9 +81,7 @@
     "index = np.array([])\n",
     "# Loop over each spectrum, adding it's data to the arrays\n",
     "for i, hdu in enumerate(sp):\n",
-    "    wave = np.append(\n",
-    "        wave, 10 ** hdu[\"COADD\"].data.loglam\n",
-    "    )  # * u.angstrom # only one entry because we only search for one xid at a time. Could change that?\n",
+    "    wave = np.append(wave, 10 ** hdu[\"COADD\"].data.loglam)  # * u.angstrom\n",
     "    flux = np.append(flux, hdu[\"COADD\"].data.flux * 1e-17)  # * u.erg/u.second/u.centimeter**2/u.angstrom\n",
     "    err = np.append(err, 1 / hdu[\"COADD\"].data.ivar * 1e-17)  # * flux.unit\n",
     "\n",
@@ -149,8 +147,8 @@
     "spec = spec_ndf.iloc[1].coadd_spectrum\n",
     "\n",
     "plt.plot(spec[\"wave\"], spec[\"flux\"])\n",
-    "plt.xlabel(\"Wavelength (Angstroms)\")\n",
-    "plt.ylabel(\"Flux (ergs/s)\")"
+    "plt.xlabel(\"Wavelength (Å)\")\n",
+    "plt.ylabel(r\"Flux ($ergs/s/cm^2/Å$)\")"
    ]
   },
   {