Refactor

tardis-sn · Jul 16, 2024 · 06e4fe8 · 06e4fe8
1 parent 3a7de52
commit 06e4fe8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tardis/visualization/tools/liv_plot.py b/tardis/visualization/tools/liv_plot.py
@@ -309,9 +309,9 @@ def _prepare_plot_data(
         if num_bins:
             if num_bins < 1:
                 raise ValueError("Number of bins must be positive")
-            elif len(bin_edges) - 1 < num_bins:
-                logger.info(
-                    "Number of bins must be less than or equal to number of shells"
+            elif num_bins > len(bin_edges) - 1:
+                logger.warn(
+                    "Number of bins must be less than or equal to number of shells. Plotting with number of bins equals to number of shells."
                 )
                 new_bin_edges = bin_edges
             else: