allowing for two bands to be used for catalogues

cta-observatory · Oct 23, 2024 · 4c34757 · 4c34757
1 parent a682e55
commit 4c34757
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/ctapipe/utils/astro.py b/src/ctapipe/utils/astro.py
@@ -20,13 +20,13 @@
 class StarCatalogues(Enum):
     Yale = {
         "directory": "V/50/catalog",
-        "band": "V",
+        "band": ["V"],
         "frame": "J2000",
         "ID_type": "HR",
     }  #: Yale bright star catalogue
     Hipparcos = {
         "directory": "I/239/hip_main",
-        "band": "V",
+        "band": ["V", "B"],
         "frame": "ICRS",
         "ID_type": "HIP",
     }  #: hipparcos catalogue
@@ -115,10 +115,10 @@ def get_star_catalog(catalog, min_magnitude=0.0, max_magnitude=10.0, row_limit=1
     catalog_dict = StarCatalogues[catalog].value
 
     columns = ["pmRA", "pmDE", catalog_dict["ID_type"]]
-    if catalog_dict["band"] == "B":
+    if "B" in catalog_dict["band"]:
         columns.append("Bmag")
         columns.append("BTmag")
-    elif catalog_dict["band"] == "V":
+    elif "V" in catalog_dict["band"]:
         columns.append("Vmag")
         columns.append("VTmag")
     if catalog_dict["frame"] == "J2000":