Update 'mask' from Spectrum1D.mask if present

glue-viz · Feb 24, 2023 · f53c556 · f53c556
1 parent 1950d70
commit f53c556
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/glue_astronomy/translators/spectrum1d.py b/glue_astronomy/translators/spectrum1d.py
@@ -246,7 +246,10 @@ def parse_attributes(attributes):
 
                 # Get mask if there is one defined, or if this is a subset
                 if subset_state is None:
-                    mask = None
+                    if 'mask' in data.component_ids():
+                        mask = data['mask']
+                    else:
+                        mask = None
                 else:
                     mask = data.get_mask(subset_state=subset_state)
                     mask = ~mask