[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

Untitled.ipynb

@@ -20,21 +20,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import math\n",
-    "import geopandas as gpd\n",
-    "import numpy as np\n",
-    "import pandas as pd\n",
-    "import pystac_client\n",
-    "import stackstac\n",
-    "import torch\n",
-    "import yaml\n",
-    "from box import Box\n",
-    "import matplotlib.pyplot as plt\n",
-    "from rasterio.enums import Resampling\n",
-    "from shapely import Point\n",
-    "from sklearn import decomposition, svm\n",
-    "#from torchvision.transformers import v2\n",
-    "#from claymodule.module import ClayMAEModule"
+    "# from torchvision.transformers import v2\n",
+    "# from claymodule.module import ClayMAEModule"
    ]
   },
   {
@@ -54,7 +41,7 @@
    "source": [
     "from torchvision import transforms\n",
     "\n",
-    "print(hasattr(transforms, \"v2\"))\n"
+    "print(hasattr(transforms, \"v2\"))"
    ]
   },
   {
@@ -75,9 +62,7 @@
      ]
     }
    ],
-   "source": [
-    "from torchvision.transformers import v2"
-   ]
+   "source": []
   },
   {
    "cell_type": "code",

claymodel/callbacks_wandb.py

@@ -265,7 +265,7 @@ def on_validation_end(
                         batch["pixels"][j + n_cols][0],
                         cmap="viridis",
                     )
-                    axs[2, j].set_title(f"Actual {j+n_cols}")
+                    axs[2, j].set_title(f"Actual {j + n_cols}")
                     axs[2, j].axis("off")
 
                     # Plot predicted images in rows 1 and 3
@@ -274,7 +274,7 @@ def on_validation_end(
                     axs[1, j].axis("off")
 
                     axs[3, j].imshow(pixels[j + n_cols][0], cmap="viridis")
-                    axs[3, j].set_title(f"Pred {j+n_cols}")
+                    axs[3, j].set_title(f"Pred {j + n_cols}")
                     axs[3, j].axis("off")
 
                 self.logger.experiment.log({f"{platform}": wandb.Image(fig)})

claymodel/datamodule.py

@@ -258,7 +258,7 @@ def setup(self, stage: Literal["fit", "predict"] | None = None) -> None:
         print(f"Total number of chips: {len(chips_path)}")
 
         if stage == "fit":
-        # Check how many unique platforms we have
+            # Check how many unique platforms we have
             unique_platforms = set(chips_platform)
 
             if len(unique_platforms) > 1:

claymodel/utils.py

@@ -35,9 +35,9 @@ def posemb_sincos_2d_with_gsd(
 
 
 def posemb_sincos_1d(waves, dim, temperature: int = 10000, dtype=torch.float32):
-    assert (
-        dim % 2 == 0
-    ), "Feature dimension must be a multiple of 2 for sincos embedding"
+    assert dim % 2 == 0, (
+        "Feature dimension must be a multiple of 2 for sincos embedding"
+    )
     waves = torch.arange(waves) if isinstance(waves, int) else waves
 
     omega = torch.arange(dim // 2, device=waves.device) / (dim // 2 - 1)

docs/clay-v0/clay-v0-location-embeddings.ipynb

@@ -331,7 +331,7 @@
     "plt.scatter(latlon[:, 0], latlon[:, 1], c=latlon[:, 2], label=\"Actual\", alpha=0.3)\n",
     "\n",
     "for i in range(100):\n",
-    "    txt = f\"{latlon[:,0][i]:.2f},{latlon[:, 1][i]:.2f}\"\n",
+    "    txt = f\"{latlon[:, 0][i]:.2f},{latlon[:, 1][i]:.2f}\"\n",
     "    plt.annotate(txt, (latlon[:, 0][i] + 1e-5, latlon[:, 1][i] + 1e-5))"
    ]
   },

docs/references.bib

@@ -1,2 +1,2 @@
 % This file contains bibliography references for the Clay Foundation Model documentation
-% Currently empty but required by the Jupyter Book configuration
+% Currently empty but required by the Jupyter Book configuration

docs/tutorials/inference.ipynb

@@ -136,7 +136,7 @@
     "items = catalog.search(\n",
     "    collections=[PLATFORM_NAME],\n",
     "    bbox=[-122.6, 37.6, -122.35, 37.85],\n",
-    "    datetime=f\"{YEAR}-01-01T00:00:00Z/{YEAR+1}-01-01T00:00:00Z\",\n",
+    "    datetime=f\"{YEAR}-01-01T00:00:00Z/{YEAR + 1}-01-01T00:00:00Z\",\n",
     "    max_items=100,\n",
     ")\n",
     "\n",