sqme: mypy Context fixes

opendatacube · Jun 24, 2024 · 3cb07d3 · 3cb07d3
1 parent 0d0544d
commit 3cb07d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/odc/loader/testing/mem_reader.py b/odc/loader/testing/mem_reader.py
@@ -11,7 +11,7 @@
 import fsspec
 import numpy as np
 import xarray as xr
-from odc.geo.geobox import GeoBox
+from odc.geo.geobox import GeoBox, GeoBoxBase
 from odc.geo.xr import ODCExtensionDa, ODCExtensionDs, xr_coords, xr_reproject
 
 from ..types import (
@@ -148,7 +148,7 @@ class Context:
 
     def __init__(
         self,
-        geobox: GeoBox,
+        geobox: None | GeoBoxBase,
         chunks: None | dict[str, int],
     ) -> None:
         self.geobox = geobox