Added 'default' and 'none' to IO Module hierarchy parsing

icecube · Jul 1, 2020 · 159ea3d · 159ea3d
1 parent 9efc342
commit 159ea3d
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/python/asteria/IO.py b/python/asteria/IO.py
@@ -157,8 +157,17 @@ def find(group, config):
     condition = '&'.join( statements )
     pass_interactions = set( row.nrow for row in tab_interactions.where(condition))
 
-    # Find Simulations that have the requested Hierarchy 
-    pass_hierarchy = set( row.nrow for row in tab_hierarchy.where(config.simulation.hierarchy))
+    # Find Simulations that have the requested Hierarchy
+    pass_hierarchy = set()
+    if config.simulation.hierarchy is None:
+        pass_hierarchy = set(row.nrow for row in tab_hierarchy.where('none'))
+    else:
+        if config.simulation.hierarchy.lower() in ['none', 'no']:
+            pass_hierarchy = set(row.nrow for row in tab_hierarchy.where('none'))
+        elif config.simulation.hierarchy.lower() in ['normal', 'default']:
+            pass_hierarchy = set(row.nrow for row in tab_hierarchy.where('normal'))
+        elif config.simulation.hierarchy.lower() in ['inverted']:
+            pass_hierarchy = set(row.nrow for row in tab_hierarchy.where('inverted'))
 
     # Find Simulations that have the requested flavors
     statements = []