Add GDAL initialization to RasterMap and file generators

LabSid-USP · Mar 18, 2024 · 695b995 · 695b995
1 parent e01a7f2
commit 695b995
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/rubem/configuration/raster_map.py b/rubem/configuration/raster_map.py
@@ -67,6 +67,9 @@ def __init__(
     ) -> None:
         self.logger = logging.getLogger(__name__)
 
+        gdal.UseExceptions()
+        gdal.AllRegister()
+
         self.logger.debug("Opening raster file: %s", file_path)
 
         self.__validate_file(file_path)

diff --git a/rubem/file/_file_generators.py b/rubem/file/_file_generators.py
@@ -12,10 +12,6 @@
 logger = logging.getLogger(__name__)
 
 
-gdal.UseExceptions()
-gdal.AllRegister()
-
-
 def report(
     variable: Field,
     name: str,
@@ -79,6 +75,9 @@ def __report(
     else:
         out_tif = os.path.join(str(outpath), f"{name}.{extension}")
 
+    gdal.UseExceptions()
+    gdal.AllRegister()
+
     with gdal.GetDriverByName(driver_short_name).Create(
         out_tif,
         base_raster_info.cols,