Shapely >= 1.2.9 required for shapely.geometry.box

environment.yml

@@ -8,8 +8,9 @@ dependencies:
   - pip
   - python
   - pytorch::pytorch>=1.7
-  - pytorch::torchvision<0.10
+  - pytorch::torchvision!=0.10.0
   - setuptools>=42
+  - shapely>=1.2.9
   - sphinx
   - rasterio
   - pip:

requirements.txt

@@ -14,6 +14,7 @@ radiant-mlhub>=0.2.1
 rarfile
 rasterio
 setuptools>=42
+shapely>=1.2.9
 sphinx
 torch>=1.7
-torchvision<0.10
+torchvision!=0.10.0

setup.cfg

@@ -29,8 +29,9 @@ install_requires =
     numpy
     pillow
     rasterio
+    shapely>=1.2.9
     torch>=1.7
-    torchvision
+    torchvision!=0.10.0
 python_requires = >= 3.6
 packages = find:
 

spack.yaml

@@ -17,7 +17,8 @@ spack:
   - py-rarfile
   - py-rasterio
   - "py-setuptools@42:"
+  - "py-shapely@1.2.9:"
   - py-sphinx
   - "py-torch@1.7:"
-  - py-torchvision
+  - "py-torchvision@:0.9,0.10.1:"
   concretization: together

torchgeo/samplers/samplers.py

@@ -1,6 +1,8 @@
 import abc
+from datetime import datetime
 from typing import Any, Iterator, Tuple
 
+from shapely.geometry import box
 from torch.utils.data import Sampler
 
 # https://github.com/pytorch/pytorch/issues/60979
@@ -35,6 +37,9 @@ class RandomGeoSampler(GeoSampler):
     the dataset and return as many random :term:`chips <chip>` as possible.
     """
 
+    def __init__(self, roi: box, start_time: datetime, end_time: datetime) -> None:
+        pass
+
 
 class GridGeoSampler(GeoSampler):
     """Samples elements in a grid-like fashion.