bugfix for cog

basaks · Aug 3, 2023 · 5c36de1 · 5c36de1
1 parent 2fa4d1d
commit 5c36de1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/configs/ref_rf.yaml b/configs/ref_rf.yaml
@@ -75,9 +75,9 @@ prediction:
   quantiles: 0.95
   outbands: 4
   geotif_options:
-    TILED: YES
-    BLOCKXSIZE: 256
-    BLOCKYSIZE: 256
+#    TILED: YES
+#    BLOCKXSIZE: 256
+#    BLOCKYSIZE: 256
     driver: COG
 
 validation:

diff --git a/uncoverml/geoio.py b/uncoverml/geoio.py
@@ -340,9 +340,9 @@ def __init__(self, shape, bbox, crs, name, n_subchunks, outputdir,
         file_names = []
 
         if mpiops.chunk_index == 0:
+            driver = kwargs.pop('driver') if 'driver' in kwargs else 'GTiff'
             for band in range(self.outbands):
                 output_filename = os.path.join(outputdir, name + "_" + file_tags[band] + ".tif")
-                driver = kwargs.pop('driver') if 'driver' in kwargs else 'GTiff'
                 f = rasterio.open(output_filename, 'w', driver=driver,
                                   width=self.shape[0], height=self.shape[1],
                                   dtype=np.float32, count=1,