Fix output widget bug

opengeos · Apr 22, 2024 · be6face · be6face
1 parent d734ffa
commit be6face
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/hypercoast/hypercoast.py b/hypercoast/hypercoast.py
@@ -2,7 +2,7 @@
 
 import ipyleaflet
 import leafmap
-
+import xarray as xr
 from .emit import read_emit, plot_emit, viz_emit, emit_to_netcdf, emit_to_image
 
 
@@ -38,6 +38,7 @@ def add(self, obj, position="topright", **kwargs):
                 from .ui import SpectralWidget
 
                 SpectralWidget(self, position=position)
+                self.set_plot_options(add_marker_cluster=True)
 
         else:
             super().add(obj, **kwargs)
@@ -159,6 +160,9 @@ def add_emit(
 
             xds = read_emit(source)
             source = emit_to_image(xds, wavelengths=wavelengths)
+        elif isinstance(source, xr.Dataset):
+            xds = source
+            source = emit_to_image(xds, wavelengths=wavelengths)
 
         self.add_raster(
             source,

diff --git a/hypercoast/ui.py b/hypercoast/ui.py
@@ -205,7 +205,8 @@ def cleanup(self):
             if hasattr(self._host_map, "_spectral_data"):
                 self._host_map._spectral_data = {}
 
-            self._output_widget.clear_output()
+            if hasattr(self, "_output_widget") and self._output_widget is not None:
+                self._output_widget.clear_output()
 
         if self.on_close is not None:
             self.on_close()