black formatting

dirac-institute · Feb 12, 2024 · 93fcf1b · 93fcf1b
1 parent 27990aa
commit 93fcf1b
Showing 1 changed file with 19 additions and 18 deletions.
diff --git a/notebooks/reprojection/reproject_demo.ipynb b/notebooks/reprojection/reproject_demo.ipynb
@@ -26,32 +26,33 @@
     "import numpy as np\n",
     "from os import path\n",
     "\n",
+    "\n",
     "# visualization utility.\n",
     "def plot_images(obstimes, image1=None, image2=None, image3=None, image4=None):\n",
-    "    '''Plot four images side by side'''\n",
-    "    \n",
-    "    min_val = -1.\n",
-    "    fig = plt.figure(figsize=(10,10))\n",
+    "    \"\"\"Plot four images side by side\"\"\"\n",
+    "\n",
+    "    min_val = -1.0\n",
+    "    fig = plt.figure(figsize=(10, 10))\n",
     "    if image1:\n",
-    "        f1 = fig.add_subplot(2,2,1, projection=image1.wcs)\n",
+    "        f1 = fig.add_subplot(2, 2, 1, projection=image1.wcs)\n",
     "        f1.title.set_text(str(obstimes[0]))\n",
-    "        plt.imshow(image1.data, vmin=min_val, origin='lower')\n",
-    "        plt.grid(color='white', ls='solid')\n",
+    "        plt.imshow(image1.data, vmin=min_val, origin=\"lower\")\n",
+    "        plt.grid(color=\"white\", ls=\"solid\")\n",
     "    if image2:\n",
-    "        f2 = fig.add_subplot(2,2,2, projection=image2.wcs)\n",
+    "        f2 = fig.add_subplot(2, 2, 2, projection=image2.wcs)\n",
     "        f2.title.set_text(str(obstimes[1]))\n",
-    "        plt.imshow(image2.data, vmin=min_val, origin='lower')\n",
-    "        plt.grid(color='white', ls='solid')\n",
+    "        plt.imshow(image2.data, vmin=min_val, origin=\"lower\")\n",
+    "        plt.grid(color=\"white\", ls=\"solid\")\n",
     "    if image3:\n",
-    "        f3 = fig.add_subplot(2,2,3, projection=image3.wcs)\n",
+    "        f3 = fig.add_subplot(2, 2, 3, projection=image3.wcs)\n",
     "        f3.title.set_text(str(obstimes[2]))\n",
-    "        plt.imshow(image3.data, vmin=min_val, origin='lower')\n",
-    "        plt.grid(color='white', ls='solid')\n",
+    "        plt.imshow(image3.data, vmin=min_val, origin=\"lower\")\n",
+    "        plt.grid(color=\"white\", ls=\"solid\")\n",
     "    if image4:\n",
-    "        f4 = fig.add_subplot(2,2,4, projection=image4.wcs)\n",
+    "        f4 = fig.add_subplot(2, 2, 4, projection=image4.wcs)\n",
     "        f4.title.set_text(str(obstimes[3]))\n",
-    "        plt.imshow(image4.data, vmin=min_val, origin='lower')\n",
-    "        plt.grid(color='white', ls='solid')"
+    "        plt.imshow(image4.data, vmin=min_val, origin=\"lower\")\n",
+    "        plt.grid(color=\"white\", ls=\"solid\")"
    ]
   },
   {
@@ -76,7 +77,7 @@
     "project_root_dir = path.abspath((path.dirname(path.dirname(path.abspath(\"\")))))\n",
     "fake_data_loc = \"tests/data/shifted_wcs_diff_dimms_tiled.fits\"\n",
     "\n",
-    "wunit = WorkUnit.from_fits(path.join(project_root_dir,fake_data_loc))\n",
+    "wunit = WorkUnit.from_fits(path.join(project_root_dir, fake_data_loc))\n",
     "wunit"
    ]
   },
@@ -199,7 +200,7 @@
     "image2_mask = CCDData(images[2].get_mask().image, unit=\"adu\")\n",
     "image2_mask.wcs = uwunit.wcs\n",
     "\n",
-    "plot_images([uwunit.get_all_obstimes()[-1]]*2, original_image2_mask, image2_mask)"
+    "plot_images([uwunit.get_all_obstimes()[-1]] * 2, original_image2_mask, image2_mask)"
    ]
   },
   {