Fix bug, GaiaDR3Vega system not found after adding additional systems

gaia-dpci · Oct 3, 2023 · 4945ecc · 4945ecc
1 parent 5846b6f
commit 4945ecc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gaiaxpy/generator/generator.py b/gaiaxpy/generator/generator.py
@@ -75,7 +75,8 @@ def __is_gaia_initially_in_systems(_internal_photometric_system: list,
         Returns:
             bool: True if Gaia DR3 is in the list, False otherwise.
         """
-        return _gaia_system in _internal_photometric_system
+        gaia_system_name = _gaia_system.get_system_name()
+        return any([item.get_system_name() == gaia_system_name for item in _internal_photometric_system])
     validate_photometric_system(photometric_system)
     validate_save_arguments(generate.__defaults__[1], output_file, generate.__defaults__[2], output_format, save_file)
     # Prepare systems, keep track of original systems (especially required for error_correction)