fix(landsat QA_pixel) : force loading dask

earthdaily · Dec 6, 2023 · 3328902 · 3328902
1 parent 38cbe4b
commit 3328902
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/earthdaily/earthdatastore/__init__.py b/earthdaily/earthdatastore/__init__.py
@@ -429,9 +429,7 @@ def datacube(
                 raise ValueError(
                     f"Specified mask '{mask_with}' is not available. Currently available masks provider are : {mask._available_masks}"
                 )
-                collection = collections[0]
-            else:
-                collection = collections
+            collection = collections[0] if isinstance(collections,list) else collections
 
             if mask_with == "ag_cloud_mask":
                 search_kwargs = self._update_search_kwargs_for_ag_cloud_mask(
@@ -510,7 +508,7 @@ def datacube(
                     acm_datacube = acm_datacube.load()
                 mask_kwargs.update(acm_datacube=acm_datacube)
             else:
-                mask_assets = mask._native_mask_asset_mapping[collections]
+                mask_assets = mask._native_mask_asset_mapping[collection]
                 if "groupby_date" in kwargs:
                     kwargs["groupby_date"] = "max"
                 if "resolution" not in kwargs:

diff --git a/earthdaily/earthdatastore/mask/__init__.py b/earthdaily/earthdatastore/mask/__init__.py
@@ -224,7 +224,7 @@ def landsat_qa_pixel(self, add_mask_var=False, mask_statistics=False):
 
     def _landsat_qa_pixel_convert(self):
         for time in self._obj.time:
-            data = self._obj["qa_pixel"].loc[dict(time=time)].data
+            data = self._obj["qa_pixel"].loc[dict(time=time)].load().data
             data_f = data.flatten()
             clm = QA_PIXEL_cloud_detection(data_f[~np.isnan(data_f)])
             clm = np.where(clm == 0, np.nan, clm)