Fixup linter

Author: mtauraso

docs/notebooks/nested_pandas_time_series_with_hyrax.ipynb

@@ -24,7 +24,7 @@
     "import hyrax\n",
     "\n",
     "rng = np.random.default_rng(42)\n",
-    "output_path = Path('./random_lightcurve.parquet')\n",
+    "output_path = Path(\"./random_lightcurve.parquet\")\n",
     "\n",
     "n_objects = 10\n",
     "n_points = 64\n",
@@ -39,12 +39,12 @@
     "    band = rng.choice([0, 1, 2], size=n_points)\n",
     "    rows.append(\n",
     "        {\n",
-    "            'object_id': f'obj-{idx:04d}',\n",
-    "            'lightcurve': {\n",
-    "                'time': times.tolist(),\n",
-    "                'flux': flux.tolist(),\n",
-    "                'flux_err': flux_err.tolist(),\n",
-    "                'band': band.tolist(),\n",
+    "            \"object_id\": f\"obj-{idx:04d}\",\n",
+    "            \"lightcurve\": {\n",
+    "                \"time\": times.tolist(),\n",
+    "                \"flux\": flux.tolist(),\n",
+    "                \"flux_err\": flux_err.tolist(),\n",
+    "                \"band\": band.tolist(),\n",
     "            },\n",
     "        }\n",
     "    )\n",
@@ -53,20 +53,20 @@
     "pq.write_table(table, output_path)\n",
     "\n",
     "h = hyrax.Hyrax()\n",
-    "h.config['data_request'] = {\n",
-    "    'infer': {\n",
-    "        'lc': {\n",
-    "            'dataset_class': 'NestedPandasDataset',\n",
-    "            'data_location': str(output_path),\n",
-    "            'primary_id_field': 'object_id',\n",
+    "h.config[\"data_request\"] = {\n",
+    "    \"infer\": {\n",
+    "        \"lc\": {\n",
+    "            \"dataset_class\": \"NestedPandasDataset\",\n",
+    "            \"data_location\": str(output_path),\n",
+    "            \"primary_id_field\": \"object_id\",\n",
     "            # Request only nested columns using nested-pandas dot-notation.\n",
-    "            'fields': ['lightcurve.time', 'lightcurve.flux', 'lightcurve.flux_err', 'lightcurve.band'],\n",
+    "            \"fields\": [\"lightcurve.time\", \"lightcurve.flux\", \"lightcurve.flux_err\", \"lightcurve.band\"],\n",
     "        }\n",
     "    }\n",
     "}\n",
     "\n",
     "prepared = h.prepare()\n",
-    "print(f'Loaded rows: {len(prepared[\"infer\"])}')"
+    "print(f\"Loaded rows: {len(prepared['infer'])}\")"
    ]
   },
   {
@@ -84,7 +84,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "prepared['infer'][0]['lc'].keys()"
+    "prepared[\"infer\"][0][\"lc\"].keys()"
    ]
   },
   {
@@ -94,7 +94,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "prepared['infer'][0]['lc']['lightcurve.time'][:5], prepared['infer'][0]['lc']['lightcurve.flux'][:5]"
+    "prepared[\"infer\"][0][\"lc\"][\"lightcurve.time\"][:5], prepared[\"infer\"][0][\"lc\"][\"lightcurve.flux\"][:5]"
    ]
   },
   {
@@ -107,19 +107,19 @@
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "\n",
-    "time = np.asarray(prepared['infer'][0]['lc']['lightcurve.time'])\n",
-    "flux = np.asarray(prepared['infer'][0]['lc']['lightcurve.flux'])\n",
-    "flux_err = np.asarray(prepared['infer'][0]['lc']['lightcurve.flux_err'])\n",
-    "band = np.asarray(prepared['infer'][0]['lc']['lightcurve.band'])\n",
+    "time = np.asarray(prepared[\"infer\"][0][\"lc\"][\"lightcurve.time\"])\n",
+    "flux = np.asarray(prepared[\"infer\"][0][\"lc\"][\"lightcurve.flux\"])\n",
+    "flux_err = np.asarray(prepared[\"infer\"][0][\"lc\"][\"lightcurve.flux_err\"])\n",
+    "band = np.asarray(prepared[\"infer\"][0][\"lc\"][\"lightcurve.band\"])\n",
     "\n",
     "plt.figure(figsize=(10, 5))\n",
     "for b in np.unique(band):\n",
     "    m = band == b\n",
-    "    plt.errorbar(time[m], flux[m], flux_err[m], fmt='o', label=f'band {b}', alpha=0.8)\n",
+    "    plt.errorbar(time[m], flux[m], flux_err[m], fmt=\"o\", label=f\"band {b}\", alpha=0.8)\n",
     "\n",
-    "plt.xlabel('MJD')\n",
-    "plt.ylabel('Flux')\n",
-    "plt.grid(True, which='both', linestyle='--', linewidth=0.5)\n",
+    "plt.xlabel(\"MJD\")\n",
+    "plt.ylabel(\"Flux\")\n",
+    "plt.grid(True, which=\"both\", linestyle=\"--\", linewidth=0.5)\n",
     "plt.legend()\n",
     "plt.show()"
    ]