Added Fornax_2022

icecube · Nov 3, 2023 · 44d380a · 44d380a
1 parent 48914b1
commit 44d380a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/asteria/source.py b/python/asteria/source.py
@@ -56,7 +56,8 @@ def __init__(self, model, model_params=None):
             logging.warning(f"Model '{self.model.__class__.__name__}' lacks one or more of the following attributes: "
                              "'luminosity', 'meanE', 'pinch'. Some ASTERIA methods may not function fully")
 
-        if self.model.__class__.__name__ in ('Fornax_2019', 'Fornax_2021'):
+        # TODO: change condition to see if class inherits from Fornax baseclasses
+        if self.model.__class__.__name__ in ('Fornax_2019', 'Fornax_2021', 'Fornax_2022'):
             logging.warning(f"Model '{self.model.__class__.__name__}' detected. Special compatibility mode enabled.\n"
                             "Expect a reduction in performance and increase in simulation run times.", )
             self._special_compat_mode = True