Fix check to be more general

spacetelescope · Dec 24, 2024 · 79ee99a · 79ee99a
1 parent 4e1a694
commit 79ee99a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jdaviz/configs/default/plugins/subset_tools/subset_tools.py b/jdaviz/configs/default/plugins/subset_tools/subset_tools.py
@@ -241,7 +241,7 @@ def get_regions(self, region_type=None, list_of_subset_labels=None,
                 subset_data = lyr.layer
                 subset_label = subset_data.label
 
-                if type(subset_data.subset_state) not in (RoiSubsetState, CompositeSubsetState):
+                if isinstance(subset_data.subset_state, MaskSubsetState):
                     # Ignore MaskSubsetState here
                     continue