Fixes issue #10

data_loading/common.py

@@ -1,12 +1,15 @@
+from __future__ import annotations
 from pathlib import Path
-from typing import Optional, Union
+from typing import Optional, Union, TYPE_CHECKING
 
 import numpy as np
-import numpy.typing as npt
 from PIL import Image
 import tifffile
 
-Patches = npt.NDArray
+if TYPE_CHECKING:
+    import numpy.typing as npt
+
+    Patches = npt.NDArray
 
 
 def load_patch(
@@ -15,7 +18,7 @@ def load_patch(
     *,
     data: Union[str, list[str]] = "all",
     landcover_mapping: Optional[npt.NDArray] = None,
-    return_arrays: bool = True
+    return_arrays: bool = True,
 ) -> list[Patches]:
     """Loads the patch data associated to an observation id
 

data_loading/environmental_raster.py

@@ -1,12 +1,18 @@
+from __future__ import annotations
 import warnings
 from pathlib import Path
-from typing import Any, Optional, Union
+from typing import Any, Optional, Union, TYPE_CHECKING
 
 import matplotlib.pyplot as plt
 import numpy as np
-import numpy.typing as npt
 import rasterio
 
+if TYPE_CHECKING:
+    import numpy.typing as npt
+
+    Coordinates = tuple[float, float]
+    Patch = npt.NDArray[np.float32]
+
 
 # fmt: off
 bioclimatic_raster_names = [
@@ -22,9 +28,6 @@
 raster_names = bioclimatic_raster_names + pedologic_raster_names
 # fmt: on
 
-Coordinates = tuple[float, float]
-Patch = npt.NDArray[np.float32]
-
 
 class Raster(object):
     """

data_loading/pytorch_dataset.py

@@ -3,18 +3,18 @@
 from typing import Callable, Optional, Union, TYPE_CHECKING
 
 import numpy as np
-import numpy.typing as npt
 import pandas as pd
 
 from torch.utils.data import Dataset
 
-from .common import load_patch, Patches
+from .common import load_patch
 
 if TYPE_CHECKING:
+    import numpy.typing as npt
+    from .common import Patches
     from .environmental_raster import PatchExtractor
 
-
-Targets = npt.NDArray[np.int64]
+    Targets = npt.NDArray[np.int64]
 
 
 class GeoLifeCLEF2022Dataset(Dataset):

metrics.py

@@ -1,9 +1,12 @@
+from __future__ import annotations
 import numbers
 import warnings
-from typing import Optional
+from typing import Optional, TYPE_CHECKING
 
 import numpy as np
-import numpy.typing as npt
+
+if TYPE_CHECKING:
+    import numpy.typing as npt
 
 
 def generic_validation(

plotting.py

@@ -3,11 +3,11 @@
 from collections.abc import Collection
 
 import numpy as np
-import numpy.typing as npt
 import matplotlib.pyplot as plt
 from matplotlib.patches import Patch
 
 if TYPE_CHECKING:
+    import numpy.typing as npt
     import pandas as pd
 
 

spatial_split/spatial_split_utils.py

@@ -1,7 +1,12 @@
+from __future__ import annotations
+from typing import TYPE_CHECKING
+
 import numpy as np
-import numpy.typing as npt
 from geopy import distance
 
+if TYPE_CHECKING:
+    import numpy.typing as npt
+
 WGS84_EQUATOR_CIRCUMFERENCE = 40075.017  # km
 WGS84_MERIDIAN_CIRCUMFERENCE = 40007.863  # km
 

submission.py

@@ -1,8 +1,11 @@
-from typing import Iterable
+from __future__ import annotations
+from typing import Iterable, TYPE_CHECKING
 
-import numpy.typing as npt
 import pandas as pd
 
+if TYPE_CHECKING:
+    import numpy.typing as npt
+
 
 def generate_submission_file(
     filename: str,