From 179c90ff31261c4dedeeda103a85e839ceef20f2 Mon Sep 17 00:00:00 2001
From: Alex <94073946+Alex-NRCan@users.noreply.github.com>
Date: Thu, 3 Oct 2024 09:21:05 -0400
Subject: [PATCH] Fixes a memoryfile issue with rasterio. When using f=json, it
 doesn't need to use the MemoryFile. (#1824)

---
 pygeoapi/provider/rasterio_.py | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/pygeoapi/provider/rasterio_.py b/pygeoapi/provider/rasterio_.py
index 69d2dbc..261754c 100644
--- a/pygeoapi/provider/rasterio_.py
+++ b/pygeoapi/provider/rasterio_.py
@@ -240,16 +240,15 @@ class RasterioProvider(BaseProvider):
             out_meta['units'] = _data.units
 
             LOGGER.debug('Serializing data in memory')
-            with MemoryFile() as memfile:
-                with memfile.open(**out_meta) as dest:
-                    dest.write(out_image)
+            if format_ == 'json':
+                LOGGER.debug('Creating output in CoverageJSON')
+                out_meta['bands'] = args['indexes']
+                return self.gen_covjson(out_meta, out_image)
 
-                if format_ == 'json':
-                    LOGGER.debug('Creating output in CoverageJSON')
-                    out_meta['bands'] = args['indexes']
-                    return self.gen_covjson(out_meta, out_image)
-
-                else:  # return data in native format
+            else:  # return data in native format
+                with MemoryFile() as memfile:
+                    with memfile.open(**out_meta) as dest:
+                        dest.write(out_image)
                     LOGGER.debug('Returning data in native format')
                     return memfile.read()