add missing full_species_list

KasukabeDefenceForce · Jan 10, 2025 · 459b0ae · 459b0ae
1 parent fbbb82f
commit 459b0ae
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/tardis/visualization/plot_util.py b/tardis/visualization/plot_util.py
@@ -221,6 +221,7 @@ def parse_species_list_util(species_list):
     ]
 
     return {
+        "full_species_list": full_species_list,
         "species_mapped": species_mapped,
         "keep_colour": keep_colour,
         "species_list": requested_species_ids,

diff --git a/tardis/visualization/tools/liv_plot.py b/tardis/visualization/tools/liv_plot.py
@@ -92,6 +92,7 @@ def _parse_species_list(self, species_list, packets_mode, nelements=None):
         if parsed_species_data is None:
             self._species_list = None
         else:
+            self._full_species_list = parsed_species_data["full_species_list"]
             self._species_mapped = parsed_species_data["species_mapped"]
             self._keep_colour = parsed_species_data["keep_colour"]
             self._species_list = parsed_species_data["species_list"]

diff --git a/tardis/visualization/tools/sdec_plot.py b/tardis/visualization/tools/sdec_plot.py
@@ -96,6 +96,7 @@ def _parse_species_list(self, species_list):
         if parsed_species_data is None:
             self._species_list = None
         else:
+            self._full_species_list = parsed_species_data["full_species_list"]
             self._species_mapped = parsed_species_data["species_mapped"]
             self._keep_colour = parsed_species_data["keep_colour"]
             self._species_list = parsed_species_data["species_list"]