rslearn 0.0.26__py3-none-any.whl → 0.0.28__py3-none-any.whl

rslearn/utils/m2m_api.py

@@ -0,0 +1,251 @@
+"""API client for the USGS M2M API."""
+
+import json
+import os
+import time
+import uuid
+from datetime import datetime, timedelta
+from typing import Any
+
+import requests
+
+from rslearn.log_utils import get_logger
+
+logger = get_logger(__name__)
+
+
+class APIException(Exception):
+    """Exception raised for M2M API errors."""
+
+    pass
+
+
+class M2MAPIClient:
+    """An API client for interacting with the USGS M2M API."""
+
+    api_url = "https://m2m.cr.usgs.gov/api/api/json/stable/"
+    pagination_size = 1000
+
+    def __init__(
+        self,
+        username: str | None = None,
+        token: str | None = None,
+        timeout: timedelta = timedelta(seconds=120),
+        session: requests.Session | None = None,
+    ) -> None:
+        """Initialize a new M2MAPIClient.
+
+        Args:
+            username: the EROS username. If None, uses M2M_USERNAME environment variable.
+            token: the application token. If None, uses M2M_TOKEN environment variable.
+            timeout: timeout for requests.
+            session: optional requests session to use. If None, a default session is
+                created.
+        """
+        if username is None:
+            username = os.environ.get("M2M_USERNAME")
+            if username is None:
+                raise ValueError(
+                    "username must be provided or M2M_USERNAME environment variable must be set"
+                )
+        if token is None:
+            token = os.environ.get("M2M_TOKEN")
+            if token is None:
+                raise ValueError(
+                    "token must be provided or M2M_TOKEN environment variable must be set"
+                )
+
+        self.username = username
+        self.timeout = timeout
+        self.session = session if session is not None else requests.Session()
+
+        json_data = json.dumps({"username": username, "token": token})
+        response = self.session.post(
+            self.api_url + "login-token",
+            data=json_data,
+            timeout=self.timeout.total_seconds(),
+        )
+
+        response.raise_for_status()
+        self.auth_token = response.json()["data"]
+
+    def request(
+        self, endpoint: str, data: dict[str, Any] | None = None
+    ) -> dict[str, Any] | None:
+        """Make a request to the API.
+
+        Args:
+            endpoint: the endpoint to call
+            data: POST data to pass
+
+        Returns:
+            JSON response data if any
+        """
+        response = self.session.post(
+            self.api_url + endpoint,
+            headers={"X-Auth-Token": self.auth_token},
+            data=json.dumps(data),
+            timeout=self.timeout.total_seconds(),
+        )
+        response.raise_for_status()
+        if response.text:
+            response_dict = response.json()
+
+            if response_dict["errorMessage"]:
+                raise APIException(response_dict["errorMessage"])
+            return response_dict
+        return None
+
+    def get_filters(self, dataset_name: str) -> list[dict[str, Any]]:
+        """Returns filters available for the given dataset.
+
+        Args:
+            dataset_name: the dataset name e.g. landsat_ot_c2_l1
+
+        Returns:
+            list of filter objects
+        """
+        response_dict = self.request("dataset-filters", {"datasetName": dataset_name})
+        if response_dict is None:
+            raise APIException("No response from API")
+        return response_dict["data"]
+
+    def scene_search(
+        self,
+        dataset_name: str,
+        acquisition_time_range: tuple[datetime, datetime] | None = None,
+        cloud_cover_range: tuple[int, int] | None = None,
+        bbox: tuple[float, float, float, float] | None = None,
+        metadata_filter: dict[str, Any] | None = None,
+    ) -> list[dict[str, Any]]:
+        """Search for scenes matching the arguments.
+
+        Args:
+            dataset_name: the dataset name e.g. landsat_ot_c2_l1
+            acquisition_time_range: optional filter on the acquisition time
+            cloud_cover_range: optional filter on the cloud cover
+            bbox: optional spatial filter
+            metadata_filter: optional metadata filter dict
+        """
+        base_data: dict[str, Any] = {"datasetName": dataset_name, "sceneFilter": {}}
+        if acquisition_time_range:
+            base_data["sceneFilter"]["acquisitionFilter"] = {
+                "start": acquisition_time_range[0].isoformat(),
+                "end": acquisition_time_range[1].isoformat(),
+            }
+        if cloud_cover_range:
+            base_data["sceneFilter"]["cloudCoverFilter"] = {
+                "min": cloud_cover_range[0],
+                "max": cloud_cover_range[1],
+                "includeUnknown": False,
+            }
+        if bbox:
+            base_data["sceneFilter"]["spatialFilter"] = {
+                "filterType": "mbr",
+                "lowerLeft": {"longitude": bbox[0], "latitude": bbox[1]},
+                "upperRight": {"longitude": bbox[2], "latitude": bbox[3]},
+            }
+        if metadata_filter:
+            base_data["sceneFilter"]["metadataFilter"] = metadata_filter
+
+        starting_number = 1
+        results = []
+        while True:
+            cur_data = base_data.copy()
+            cur_data["startingNumber"] = starting_number
+            cur_data["maxResults"] = self.pagination_size
+            response_dict = self.request("scene-search", cur_data)
+            if response_dict is None:
+                raise APIException("No response from API")
+            data = response_dict["data"]
+            results.extend(data["results"])
+            if data["recordsReturned"] < self.pagination_size:
+                break
+            starting_number += self.pagination_size
+
+        return results
+
+    def get_scene_metadata(self, dataset_name: str, entity_id: str) -> dict[str, Any]:
+        """Get detailed metadata for a scene.
+
+        Args:
+            dataset_name: the dataset name in which to search
+            entity_id: the entity ID of the scene
+
+        Returns:
+            full scene metadata
+        """
+        response_dict = self.request(
+            "scene-metadata",
+            {
+                "datasetName": dataset_name,
+                "entityId": entity_id,
+                "metadataType": "full",
+            },
+        )
+        if response_dict is None:
+            raise APIException("No response from API")
+        return response_dict["data"]
+
+    def get_downloadable_products(
+        self, dataset_name: str, entity_id: str
+    ) -> list[dict[str, Any]]:
+        """Get the downloadable products for a given scene.
+
+        Args:
+            dataset_name: the dataset name
+            entity_id: the entity ID of the scene
+
+        Returns:
+            list of downloadable products
+        """
+        data = {"datasetName": dataset_name, "entityIds": [entity_id]}
+        response_dict = self.request("download-options", data)
+        if response_dict is None:
+            raise APIException("No response from API")
+        return response_dict["data"]
+
+    def get_download_url(self, entity_id: str, product_id: str) -> str:
+        """Get the download URL for a given product.
+
+        Args:
+            entity_id: the entity ID of the product
+            product_id: the product ID of the product
+
+        Returns:
+            the download URL
+        """
+        label = str(uuid.uuid4())
+        data = {
+            "downloads": [
+                {"label": label, "entityId": entity_id, "productId": product_id}
+            ]
+        }
+        response_dict = self.request("download-request", data)
+        if response_dict is None:
+            raise APIException("No response from API")
+        response = response_dict["data"]
+
+        # Check if URL is immediately available in the response
+        if response.get("availableDownloads"):
+            return response["availableDownloads"][0]["url"]
+
+        # Otherwise poll download-retrieve until the URL is ready
+        while True:
+            # We sleep first because we have observed that if we immediately call
+            # download-retrieve it sometimes has a response that includes neither
+            # available nor requested list.
+            time.sleep(10)
+            response_dict = self.request("download-retrieve", {"label": label})
+            if response_dict is None:
+                raise APIException("No response from API")
+            response = response_dict["data"]
+            if len(response["available"]) > 0:
+                return response["available"][0]["url"]
+            if len(response["requested"]) == 0:
+                logger.debug(
+                    f"Response to download-retrieve did not include our requested download under label {label}: {response}"
+                )
+                raise APIException("Did not get download URL")
+            if response["requested"][0].get("url"):
+                return response["requested"][0]["url"]

rslearn/utils/retry_session.py

@@ -0,0 +1,43 @@
+"""Utility for creating requests sessions with retry logic."""
+
+from datetime import timedelta
+from http import HTTPStatus
+
+import requests
+from requests.adapters import HTTPAdapter
+from urllib3.util.retry import Retry
+
+
+def create_retry_session(
+    total_retries: int = 5,
+    backoff_factor: float = 1.0,
+    backoff_max: timedelta = timedelta(seconds=60),
+) -> requests.Session:
+    """Create a requests session with retry logic.
+
+    Args:
+        total_retries: maximum number of retries
+        backoff_factor: factor for exponential backoff
+        backoff_max: maximum backoff time
+
+    Returns:
+        configured requests session
+    """
+    session = requests.Session()
+    retry = Retry(
+        total=total_retries,
+        backoff_factor=backoff_factor,
+        backoff_max=backoff_max.total_seconds(),
+        allowed_methods=["GET", "POST"],
+        status_forcelist=[
+            HTTPStatus.TOO_MANY_REQUESTS,
+            HTTPStatus.INTERNAL_SERVER_ERROR,
+            HTTPStatus.BAD_GATEWAY,
+            HTTPStatus.SERVICE_UNAVAILABLE,
+            HTTPStatus.GATEWAY_TIMEOUT,
+        ],
+        raise_on_status=False,
+    )
+    session.mount("http://", HTTPAdapter(max_retries=retry))
+    session.mount("https://", HTTPAdapter(max_retries=retry))
+    return session

{rslearn-0.0.26.dist-info → rslearn-0.0.28.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rslearn
-Version: 0.0.26
+Version: 0.0.28
 Summary: A library for developing remote sensing datasets and models
 Author: OlmoEarth Team
 License:                                  Apache License
@@ -248,6 +248,9 @@ Requires-Dist: pycocotools>=2.0; extra == "extra"
 Requires-Dist: pystac_client>=0.9; extra == "extra"
 Requires-Dist: rtree>=1.4; extra == "extra"
 Requires-Dist: termcolor>=3.0; extra == "extra"
+Requires-Dist: xarray>=2024.1; extra == "extra"
+Requires-Dist: zarr>=3.1.2; extra == "extra"
+Requires-Dist: numcodecs>=0.16; extra == "extra"
 Requires-Dist: satlaspretrain_models>=0.3; extra == "extra"
 Requires-Dist: scipy>=1.16; extra == "extra"
 Requires-Dist: terratorch>=1.0.2; extra == "extra"
@@ -587,7 +590,7 @@ data:
       groups: ["default"]
     predict_config:
       groups: ["predict"]
-      load_all_patches: true
+      load_all_crops: true
       skip_targets: true
       patch_size: 512
 trainer:
@@ -750,7 +753,7 @@ test_config:
     split: val
 predict_config:
   groups: ["predict"]
-  load_all_patches: true
+  load_all_crops: true
   skip_targets: true
   patch_size: 512
 ```

{rslearn-0.0.26.dist-info → rslearn-0.0.28.dist-info}/RECORD

@@ -8,34 +8,36 @@ rslearn/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rslearn/template_params.py,sha256=Vop0Ha-S44ctCa9lvSZRjrMETznJZlR5y_gJrVIwrPg,791
 rslearn/config/__init__.py,sha256=n1qpZ0ImshTtLYl5mC73BORYyUcjPJyHiyZkqUY1hiY,474
 rslearn/config/dataset.py,sha256=abxIUFDAYmCd4pzGnkPnW_pYyws1yhXFWJ5HnVU4WHo,23942
-rslearn/data_sources/__init__.py,sha256=zzuZUxrlEIw84YpD2I0HJvCoLDB29LbmnKTXiJykzGU,660
-rslearn/data_sources/aws_landsat.py,sha256=0ZQtmd2NCnvLy4vFSB1AlmoguJbiQB_e_T4eS1tnW9Q,20443
-rslearn/data_sources/aws_open_data.py,sha256=rDE9VZXXA669fD9jfOizxpN41lmvvJc-skZz3LettJU,29197
-rslearn/data_sources/aws_sentinel1.py,sha256=knBg2ZdwzGRIibUPAqnhYR-DbHqFL4tLsuRVrucTWU4,4745
-rslearn/data_sources/aws_sentinel2_element84.py,sha256=VkIAjCqzqi_m6StSQ-aY_AMtesFhkYYgwiUNkpBe7AA,13984
-rslearn/data_sources/climate_data_store.py,sha256=gwlbugwAT0YoOINOKNVlZuzpxUO5G0tfzwy4XVlURFg,18360
-rslearn/data_sources/copernicus.py,sha256=DLyXwnkFo2LzbxfLKHkOvHyZcNYFg5w-yXQPeBL67_w,35049
-rslearn/data_sources/data_source.py,sha256=lxdcxefETX5ui5uUdZn-ACSBhy6YVo4XDfk368EMD40,4862
-rslearn/data_sources/earthdaily.py,sha256=BNP7BK_vb9yQw6LaIaJ80vjCfBZ9TeALGkjHYIet_W0,18205
-rslearn/data_sources/earthdata_srtm.py,sha256=bwo8e_y9fFPliZ411tTWOiAUlEcb3AWBkeZxnkFY5SI,10633
-rslearn/data_sources/eurocrops.py,sha256=FQqdBcNl43fArq6rd_-iffVJyliIDbB0lIHvNdmtQBU,8663
-rslearn/data_sources/gcp_public_data.py,sha256=qXWFQ4y0cCVBla68H4bpolG0FKW9_vLyTlA-0mJc1Y0,37349
-rslearn/data_sources/google_earth_engine.py,sha256=xdoSDjSVp6lPVPMv4UJZ6BRUozUA2hFbSTl1707TBoM,23523
-rslearn/data_sources/local_files.py,sha256=xg3IqMVXi6s_Kpnp91QGF7hSiTw4y-vgoe2bQNglWcs,19064
-rslearn/data_sources/openstreetmap.py,sha256=TzZfouc2Z4_xjx2v_uv7aPn4tVW3flRVQN4qBfl507E,18161
-rslearn/data_sources/planet.py,sha256=6FWQ0bl1k3jwvwp4EVGi2qs3OD1QhnKOKP36mN4HELI,9446
-rslearn/data_sources/planet_basemap.py,sha256=e9R6FlagJjg8Z6Rc1dC6zK3xMkCohz8eohXqXmd29xg,9670
-rslearn/data_sources/planetary_computer.py,sha256=k6CO5Yim2I-frlD8r2_uBo0CQFw89mN3_5mrv0Xk2WU,26449
-rslearn/data_sources/soilgrids.py,sha256=rwO4goFPQ7lx420FvYBHYFXdihnZqn_-IjdqtxQ9j2g,12455
-rslearn/data_sources/stac.py,sha256=Xty1JDueAAonNVLRo8vfNBhlHrVLjhmZ-uRBYbrGvtA,10753
-rslearn/data_sources/usda_cdl.py,sha256=_WvxZkm0fbXfniRs6NT8iVCbTTmVPflDhsFT2ci6_Dk,6879
-rslearn/data_sources/usgs_landsat.py,sha256=kPOb3hsZe5-guUcFZZkwzcRpYZ3Zo7Bk4E829q_xiyU,18516
+rslearn/data_sources/__init__.py,sha256=FZQckYwsnnLokMeYmi0ktUyQd9bAHyLN1_-Xc3qYLag,767
+rslearn/data_sources/aws_landsat.py,sha256=bJmwBbUV4vjKBNp1MHt4sHhnIjhMis_jOI3FpksQc6w,16435
+rslearn/data_sources/aws_open_data.py,sha256=fum34DqqDiuiiYBfZtGFrNNOLylE9o3o7Cyb2e0Eo0g,29101
+rslearn/data_sources/aws_sentinel1.py,sha256=LfJLDhsd_6h_JinD8PbiiAyxajkIdvAc--5BJryUKlo,4674
+rslearn/data_sources/aws_sentinel2_element84.py,sha256=qeCuiSlvhWChSY3AwYsKT6nZUsjNNWoOBPEUxkV8kJI,10112
+rslearn/data_sources/climate_data_store.py,sha256=mqLJfYubD6m9VwxpLunoIv_MNFN6Ue1hBBVj552e8uQ,18289
+rslearn/data_sources/copernicus.py,sha256=ushAgYGxU2MzPcUNnEvEfPgO0RCC9Rbjzi189xq0jgc,35001
+rslearn/data_sources/data_source.py,sha256=xojlCoAnGTCHKbEx98JkW0oYzAKBbgGMNc0kicEjHWk,4863
+rslearn/data_sources/direct_materialize_data_source.py,sha256=PBmxsLBNakJeX1s92pc4FCuHANfhkYHS2vt60RGdkj0,11276
+rslearn/data_sources/earthdaily.py,sha256=qUtHUG1oV5IlCWXVovUcYxQhqdNDKWaEe-BKnooWX88,14623
+rslearn/data_sources/earthdatahub.py,sha256=KRf1VnxPI9jsT0utEkeYvsCwu7LXo9t-RvMi8gXehag,15889
+rslearn/data_sources/eurocrops.py,sha256=dJ4d0xvt-rID_HuAchyucFJBuAQL-Kk1h_qm6GOH-mE,8641
+rslearn/data_sources/gcp_public_data.py,sha256=9M_tqGUG9740TC3ZRJ8-sekWkKFRp7_XdYHQEAloJpw,37301
+rslearn/data_sources/google_earth_engine.py,sha256=7fytWecT7oWCx9u4nXidb4ycdOoX2hV3qCeVvOPleC8,23475
+rslearn/data_sources/hf_srtm.py,sha256=tIKPdPt2vpRscsAmbsKbn_UlEIgJYua6lXRU2wwyXq4,21250
+rslearn/data_sources/local_files.py,sha256=1m2fz5d5ZLDtx-qMxqVNGzVeGSL8RVYmrNe4NiQS5_Y,19065
+rslearn/data_sources/openstreetmap.py,sha256=KZe52ZYwzetjsyCaWN43-WUcH1Xd8j6AocaOPidG6Yo,18162
+rslearn/data_sources/planet.py,sha256=K4Lpm9UyUJjIiSISe3jDlV_o0UipLRJ2Snw7rmJfz2g,9398
+rslearn/data_sources/planet_basemap.py,sha256=oColM4udr23e8c9dK-gc08a2q4Iil23tun3P9zjPOqc,9622
+rslearn/data_sources/planetary_computer.py,sha256=hiYdYgU5gP2mJuDvF4tS4Sk4vWOndUcwfFUpQ7x-lZY,27231
+rslearn/data_sources/soilgrids.py,sha256=qbnnCIOa6tlN8wxmNCzAj60-pghKEbRxa7lVIgM95Dk,12484
+rslearn/data_sources/stac.py,sha256=Gj8TZ5pifVzWPCuzgphrle2ekQ02OET54rj-02sR2nw,10705
+rslearn/data_sources/usda_cdl.py,sha256=3GhcgTB50T7GA44nB9WwItqDJliELquw_YbiAVxh6kc,6808
+rslearn/data_sources/usgs_landsat.py,sha256=IsQOhWY8nwmgixJu1uMSR4CqsC3igcP3TArdBXkETd8,10178
 rslearn/data_sources/utils.py,sha256=NOC0qOyYVS6f8EUrSeP4mH0XZbSrtTLV-gKGbCC6ccg,16586
 rslearn/data_sources/vector_source.py,sha256=NCa7CxIrGKe9yRT0NyyFKFQboDGDZ1h7663PV9OfMOM,44
-rslearn/data_sources/worldcereal.py,sha256=rGGxwJAC3pAASVBQoUhwS3qL-qcmF6W__Tb53qWCEmE,21501
-rslearn/data_sources/worldcover.py,sha256=n7gi-JRytxkvkUhKT--dVziMcWSSyMbZA7ZCzLT2MJY,6037
-rslearn/data_sources/worldpop.py,sha256=S3RSc5kTwSs2bcREjNarBsqf3MBX5CN0eHj7Qkx4K74,5625
-rslearn/data_sources/xyz_tiles.py,sha256=P601CvUmoVDC_ZRVhPKaIoPYYCq5UhZ-v7DaGlN5y_0,13797
+rslearn/data_sources/worldcereal.py,sha256=OWZA0pvQQiKvuA5AVAc0lw8JStMEeF4DYOh0n2vdg6I,21521
+rslearn/data_sources/worldcover.py,sha256=ahyrGoXMAGWsIUDHSrqPywiK7ycwUD3E3BruNMxpo90,6057
+rslearn/data_sources/worldpop.py,sha256=lUI7syzNs4FiGVLz2KIiFo63MafFSyWg__YdrSOXtWA,5645
+rslearn/data_sources/xyz_tiles.py,sha256=RKn3q19FbgOR8a-qQ4ijnNPjmYPaVimIaSL7lsWlYqM,13729
 rslearn/dataset/__init__.py,sha256=bHtBlEEBCekO-gaJqiww0-VjvZTE5ahx0llleo8bfP8,289
 rslearn/dataset/add_windows.py,sha256=NwIvku6zxCJ9kgVFa5phJc0Gj1Y1bCzh6TLb9nEGl0s,8462
 rslearn/dataset/dataset.py,sha256=YyGFy_VGUaOPfrEQqBl0Fp5JAsH3KNCqo6ZTu3TW4yY,3223
@@ -45,14 +47,14 @@ rslearn/dataset/materialize.py,sha256=VoL5Qf5pGcQV4QMlO5vrcu7w0Sl1NdIRLUVk0kSCMO
 rslearn/dataset/remap.py,sha256=6MaImsY02GNACpvRM81RvWmjZWRfAHxo_R3Ox6XLF6A,2723
 rslearn/dataset/window.py,sha256=X4q8YzcSOTtwKxCPf71QLMoyKUtYMSnZu0kPnmVSUx4,10644
 rslearn/dataset/storage/__init__.py,sha256=R50AVV5LH2g7ol0-jyvGcB390VsclXGbJXz4fmkn9as,52
-rslearn/dataset/storage/file.py,sha256=g9HZ3CD4QcgyVNsBaXhjIKQgDOAeZ4R08sJ7ntx4wo8,6815
+rslearn/dataset/storage/file.py,sha256=GJJgH_eHknLbMQwoC3mOoXJKl6Ha3oNXzz62FIEMWlg,7130
 rslearn/dataset/storage/storage.py,sha256=DxZ7iwV938PiLwdQzb5EXSb4Mj8bRGmOTmA9fzq_Ge8,4840
 rslearn/models/__init__.py,sha256=_vWoF9d2Slah8-6XhYhdU4SRsy_CNxXjCGQTD2yvu3Q,22
 rslearn/models/anysat.py,sha256=nzk6hB83ltNFNXYRNA1rTvq2AQcAhwyvgBaZui1M37o,8107
 rslearn/models/attention_pooling.py,sha256=Kss1W_HYNzfX78RVDi18m0-cG2BoPsC240u0oEqJ0d4,7187
 rslearn/models/clip.py,sha256=QG1oUFqcVuNEhx7BNfJ1FnxIOMNUwRNBwXCe3CR6wFI,2415
 rslearn/models/component.py,sha256=uikFDzPYaW_LSXsrSsES1aup4IDIuWHsitWLpKgF7zU,3432
-rslearn/models/concatenate_features.py,sha256=Attemr5KurxlOojpclD0Pd5Cu2KHpNdpXe8jCSjpJ9U,3818
+rslearn/models/concatenate_features.py,sha256=0gZPkUBpyhzq1HNhN7mKl44u2vsL2p-o3gy5gpAy2go,3947
 rslearn/models/conv.py,sha256=dEAAfhPo4bFlZPSAQjzqZTpP-hdJ394TytYssVK-fDA,2001
 rslearn/models/croma.py,sha256=BcKV4-D4uira6f9zvW73aslF_XitAhObnyrE_3iTcTs,11008
 rslearn/models/dinov3.py,sha256=Q9X7VTwzjllLSEvc235C9BY_jMnIoSybsiOkeA58uHo,6472
@@ -110,17 +112,18 @@ rslearn/models/presto/__init__.py,sha256=eZrB-XKi_vYqZhpyAOwppJi4dRuMtYVAdbq7KRy
 rslearn/models/presto/presto.py,sha256=fkyHB85Hfx5L-4yejSFAFv83gk9VFqAR1GTgggtq0EA,11049
 rslearn/models/presto/single_file_presto.py,sha256=-P00xjhj9dx3O6HqWpQmG9dPk_i6bT_t8vhX4uQm5tA,30242
 rslearn/tile_stores/__init__.py,sha256=-cW1J7So60SEP5ZLHCPdaFBV5CxvV3QlOhaFnUkhTJ0,1675
-rslearn/tile_stores/default.py,sha256=PYaDNvBxhJTDKJGw0EjDTSE1OKajR7_iJpMbOjj-mE8,15054
+rslearn/tile_stores/default.py,sha256=AG2j0FCNi_4cnXqLjRIef5wMqMJ5_YtSkTIhk7qJQVQ,15134
 rslearn/tile_stores/tile_store.py,sha256=9AeYduDYPp_Ia2NMlq6osptpz_AFGIOQcLJrqZ_m-z0,10469
 rslearn/train/__init__.py,sha256=fnJyY4aHs5zQqbDKSfXsJZXY_M9fbTsf7dRYaPwZr2M,30
-rslearn/train/all_patches_dataset.py,sha256=EVoYCmS3g4OfWPt5CZzwHVx9isbnWh5HIGA0RBqPFeA,21145
-rslearn/train/data_module.py,sha256=gar7RsyurdBLtbd1y8llkbMrRU7_RTX2tonp45qPIns,23784
-rslearn/train/dataset.py,sha256=a4ehgAiPbEWJQ8T8JJuhfF62CN9KKr6iOW67U6WLPmY,41380
+rslearn/train/all_crops_dataset.py,sha256=CWnqbSjRXJZQsudljvpA07oldiP4fZTmjwrT0sjVnq4,21399
+rslearn/train/data_module.py,sha256=G1TRhXg8SPewYy0BTZN5KpeLPK72qIaH15ePfUwrxgM,23865
+rslearn/train/dataset.py,sha256=vCmm6yrW2bAc5A94aBwQe-SOIGdVcZYMM2oBYRq2_sw,45253
 rslearn/train/dataset_index.py,sha256=S5iXhQga5gnnkDqThXXlyjIwkJBPVWiUfDPx3iVs-pw,5306
-rslearn/train/lightning_module.py,sha256=V4YoEg9PrwrgG4q9Dmv_9OBrSIK-SRPzjWtZRIfmPFg,15366
-rslearn/train/model_context.py,sha256=FCHPHLrQ57RjJpsWi3u1HV0tZc-gf0tzvx4lpdEf6gU,3204
+rslearn/train/lightning_module.py,sha256=n4hasJBVlAmMhvf2yaFo0gy1vGz5haQkJpZdCSKlJ8A,17482
+rslearn/train/metrics.py,sha256=RknMf2n09D5XBCf0YM4Zmm0XI-pFbRtsmbY51ipVMPk,4799
+rslearn/train/model_context.py,sha256=8DMWGj5xCRmRDo_38lkhkUMHfK_yg3XZrUJQIz5a1vA,3200
 rslearn/train/optimizer.py,sha256=EKSqkmERalDA0bF32Gey7n6z69KLyaUWKlRsGJfKBmE,927
-rslearn/train/prediction_writer.py,sha256=rW0BUaYT_F1QqmpnQlbrLiLya1iBfC5Pb78G_NlF-vA,15956
+rslearn/train/prediction_writer.py,sha256=cRFehEtr0iBuVqzE69a0B4Lvb8ywxLeyon34KWI86H0,16961
 rslearn/train/scheduler.py,sha256=Yyq2fMHH08OyGt9qWD7iLc92XNIBbZHLtzlLR-f732s,2728
 rslearn/train/callbacks/__init__.py,sha256=VNV0ArZyYMvl3dGK2wl6F046khYJ1dEBlJS6G_SYNm0,47
 rslearn/train/callbacks/adapters.py,sha256=yfv8nyCj3jmo2_dNkFrjukKxh0MHsf2xKqWwMF0QUtY,1869
@@ -128,13 +131,13 @@ rslearn/train/callbacks/freeze_unfreeze.py,sha256=8fIzBMhCKKjpTffIeAdhdSjsBd8NjT
 rslearn/train/callbacks/gradients.py,sha256=4YqCf0tBb6E5FnyFYbveXfQFlgNPyxIXb2FCWX4-6qs,5075
 rslearn/train/callbacks/peft.py,sha256=wEOKsS3RhsRaZTXn_Kz2wdsZdIiIaZPdCJWtdJBurT8,4156
 rslearn/train/tasks/__init__.py,sha256=dag1u72x1-me6y0YcOubUo5MYZ0Tjf6-dOir9UeFNMs,75
-rslearn/train/tasks/classification.py,sha256=72ZBcbunMsdPYQN53S-4GfiLIDrr1X3Hni07dBJ0pu0,14261
-rslearn/train/tasks/detection.py,sha256=8nxa90ht0-xlcO_gsI-zR02nL82-wJXq4LpXNFvg5Lw,21730
+rslearn/train/tasks/classification.py,sha256=_3cRa8ojd9sG2ELRW_BvByZh2YFdCBaklR8Kv9LAgOY,14864
+rslearn/train/tasks/detection.py,sha256=uDMGtsCMSk9OGXn-vpFKBAyHyVN0ji2NCfqBgg1BQyw,21725
 rslearn/train/tasks/embedding.py,sha256=NdJEAaDWlWYzvOBVf7eIHfFOzqTgavfFH1J1gMbAMVo,3891
 rslearn/train/tasks/multi_task.py,sha256=32hvwyVsHqt7N_M3zXsTErK1K7-0-BPHzt7iGNehyaI,6314
-rslearn/train/tasks/per_pixel_regression.py,sha256=njShN-U9fx3SPcCxGgbDlZAp3DT_GlTt0BRZS416gnw,10387
-rslearn/train/tasks/regression.py,sha256=bVS_ApZSpbL0NaaM8Mu5Bsu4SBUyLpVtrPslulvvZHs,12695
-rslearn/train/tasks/segmentation.py,sha256=dn1yo1dIArKvW9Giw8-LZyIZ87q76eslL0mk58GyApo,29663
+rslearn/train/tasks/per_pixel_regression.py,sha256=3m_BTP2akadYe3IuAlCG2bd_alfNyom55-pFrI2q4PE,10928
+rslearn/train/tasks/regression.py,sha256=TzHL42gm3aIdev0R7_uz_TSYbAwSvQPjCD42y1p9_7Y,13269
+rslearn/train/tasks/segmentation.py,sha256=b9XS09EQvum89eoW3vWqFMKuCRtznODteKIr1hFnIz4,30531
 rslearn/train/tasks/task.py,sha256=nMPunl9OlnOimr48saeTnwKMQ7Du4syGrwNKVQq4FL4,4110
 rslearn/train/transforms/__init__.py,sha256=BkCAzm4f-8TEhPIuyvCj7eJGh36aMkZFYlq-H_jkSvY,778
 rslearn/train/transforms/concatenate.py,sha256=S8f1svzwb5UmeAgzXe4Af_hFvt5o0tQctIE6t3QYuPI,2625
@@ -147,17 +150,19 @@ rslearn/train/transforms/resize.py,sha256=LPXCD6NS8Xi1uPOtHxRWtT5CAkpEzsVafHkHIQ
 rslearn/train/transforms/select_bands.py,sha256=owxMOvWEp8lRZhj6pQ1qFYebKZCQJZZCW8sBrK_T9-I,1868
 rslearn/train/transforms/sentinel1.py,sha256=zYwT3amv1M7BYZIfkIX3J2Rzh161d-ngCqp3ypO9gBE,2011
 rslearn/train/transforms/transform.py,sha256=n1Qzqix2dVvej-Q7iPzHeOQbqH79IBlvqPoymxhNVpE,4446
-rslearn/utils/__init__.py,sha256=GNvdTUmXakiEMnLdje7k1fe5aC7SFVqP757kbpN6Fzw,558
+rslearn/utils/__init__.py,sha256=GZc1erpEfXTc32yjEDbt5rnMrnXEBY7WVm3v4NlwwWY,620
 rslearn/utils/array.py,sha256=RC7ygtPnQwU6Lb9kwORvNxatJcaJ76JPsykQvndAfes,2444
 rslearn/utils/colors.py,sha256=ELY9_buH06TOVPLrDAyf2S0G--ZiOxnnP8Ujim6_3ig,369
 rslearn/utils/feature.py,sha256=lsg0WThZDJzo1mrbaL04dXYI5G3x-n5FG9aEjj7uUaI,1649
-rslearn/utils/fsspec.py,sha256=h3fER_bkewzR9liEAULXguTIvXLUXA17pC_yZoWN5Tk,5902
-rslearn/utils/geometry.py,sha256=S60uEvCQ3SyR2-3TJ1yUucFmGRM_rcoIT_gH83PN3RE,21485
+rslearn/utils/fsspec.py,sha256=TcEUgXKvsmtKHv5JVOI2Vp4WNfVNeTok0x4JgZaD1iw,7052
+rslearn/utils/geometry.py,sha256=VzLoxtwdV3uC3szowT-bGuCFF6ge8eK0m01lq8q-01Q,22423
 rslearn/utils/get_utm_ups_crs.py,sha256=kUrcyjCK7KWvuP1XR-nURPeRqYeRO-3L8QUJ1QTF9Ps,3599
 rslearn/utils/grid_index.py,sha256=hRmrtgpqN1pLa-djnZtgSXqKJlbgGyttGnCEmPLD0zo,2347
 rslearn/utils/jsonargparse.py,sha256=TRyZA151KzhjJlZczIHYguEA-YxCDYaZ2IwCRgx76nM,4791
+rslearn/utils/m2m_api.py,sha256=YemQkh9ItdU6ojP3XBUf9rj1we_P-8mdy7IHvlMNlKQ,8893
 rslearn/utils/mp.py,sha256=XYmVckI5TOQuCKc49NJyirDJyFgvb4AI-gGypG2j680,1399
 rslearn/utils/raster_format.py,sha256=fwotJBadwqYSdK8UokiKOk1pOF8JMim3kP_VwLWivPI,27382
+rslearn/utils/retry_session.py,sha256=alVwqvZqqG3TZNg7NVbgfpUnSMdWve8T7gYzZmBz1jc,1284
 rslearn/utils/rtree_index.py,sha256=j0Zwrq3pXuAJ-hKpiRFQ7VNtvO3fZYk-Em2uBPAqfx4,6460
 rslearn/utils/spatial_index.py,sha256=eomJAUgzmjir8j9HZnSgQoJHwN9H0wGTjmJkMkLLfsU,762
 rslearn/utils/sqlite_index.py,sha256=YGOJi66544e6JNtfSft6YIlHklFdSJO2duxQ4TJ2iu4,2920
@@ -171,10 +176,10 @@ rslearn/vis/render_sensor_image.py,sha256=D0ynK6ABPV046970lIKwF98klpSCtrsUvZTwtZ
 rslearn/vis/render_vector_label.py,sha256=ncwgRKCYCJCK1-wTpjgksOiDDebku37LpAyq6wsg4jg,14939
 rslearn/vis/utils.py,sha256=Zop3dEmyaXUYhPiGdYzrTO8BRXWscP2dEZy2myQUnNk,2765
 rslearn/vis/vis_server.py,sha256=kIGnhTy-yfu5lBOVCoo8VVG259i974JPszudCePbzfI,20157
-rslearn-0.0.26.dist-info/licenses/LICENSE,sha256=_99ZWPoLdlUbqZoSC5DF4ihiNwl5rTEmBaq2fACecdg,11352
-rslearn-0.0.26.dist-info/licenses/NOTICE,sha256=wLPr6rwV_jCg-xEknNGwhnkfRfuoOE9MZ-lru2yZyLI,5070
-rslearn-0.0.26.dist-info/METADATA,sha256=ZkFWiVzrI_4J1qtC6zSF1dxGDtW4a8wPKdYaZbmGsoQ,38576
-rslearn-0.0.26.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-rslearn-0.0.26.dist-info/entry_points.txt,sha256=doTBQ57NT7nq-dgYGgTTw6mafcGWb_4PWYtYR4rGm50,46
-rslearn-0.0.26.dist-info/top_level.txt,sha256=XDKo90WBH8P9RQumHxo0giLJsoufT4r9odv-WE6Ahk4,8
-rslearn-0.0.26.dist-info/RECORD,,
+rslearn-0.0.28.dist-info/licenses/LICENSE,sha256=_99ZWPoLdlUbqZoSC5DF4ihiNwl5rTEmBaq2fACecdg,11352
+rslearn-0.0.28.dist-info/licenses/NOTICE,sha256=wLPr6rwV_jCg-xEknNGwhnkfRfuoOE9MZ-lru2yZyLI,5070
+rslearn-0.0.28.dist-info/METADATA,sha256=OSGXg3yVyndUAZYL9EwvCl095O_Vfknpxlhx8O5dLmQ,38714
+rslearn-0.0.28.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+rslearn-0.0.28.dist-info/entry_points.txt,sha256=doTBQ57NT7nq-dgYGgTTw6mafcGWb_4PWYtYR4rGm50,46
+rslearn-0.0.28.dist-info/top_level.txt,sha256=XDKo90WBH8P9RQumHxo0giLJsoufT4r9odv-WE6Ahk4,8
+rslearn-0.0.28.dist-info/RECORD,,