diff --git a/src/acom_music_box/plot_output.py b/src/acom_music_box/plot_output.py index 8b51dc5..58b6849 100644 --- a/src/acom_music_box/plot_output.py +++ b/src/acom_music_box/plot_output.py @@ -94,7 +94,7 @@ def _plot_with_gnuplot(self): """ # Prepare columns and data for plotting if not self.species_list: - return; + return for species_group in self.species_list: columns = ['time'] + species_group data_to_plot = self.df[columns] @@ -133,7 +133,7 @@ def _plot_with_matplotlib(self): Plot the specified species using matplotlib. """ if not self.species_list: - return; + return for species_group in self.species_list: indexed = self.df.set_index('time') fig, ax = plt.subplots()