From f19153453cc1563d3d1291cebd4af1b6369982d8 Mon Sep 17 00:00:00 2001
From: Rambaud Pierrick <12rambau@users.noreply.github.com>
Date: Thu, 21 Apr 2022 23:04:55 +0200
Subject: [PATCH] mask nan in single band raster

---
 xarray_leaflet/xarray_leaflet.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/xarray_leaflet/xarray_leaflet.py b/xarray_leaflet/xarray_leaflet.py
index b26f034..e32985e 100644
--- a/xarray_leaflet/xarray_leaflet.py
+++ b/xarray_leaflet/xarray_leaflet.py
@@ -392,7 +392,9 @@ def _get_tiles(self, change=None):
                         da_tile = np.stack(das, axis=2)
                         write_image(path, da_tile, self.persist)
                     else:
-                        da_tile = self.colormap(das[0])
+                        alpha = np.where(np.isnan(das[0]), 0, 1) 
+                        da_tile = self.colormap(das[0])                        
+                        da_tile[:,:,3:] = np.expand_dims(alpha, axis=2)
                         write_image(path, da_tile*255, self.persist)
 
         if self.dynamic: