add value error to ensure active virtual packet logging

tardis-sn · Jun 2, 2021 · f4cc9bd · f4cc9bd
1 parent bdeb1a6
commit f4cc9bd
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/tardis/visualization/tools/sdec_plot.py b/tardis/visualization/tools/sdec_plot.py
@@ -415,12 +415,18 @@ def from_simulation(cls, sim):
         -------
         SDECPlotter
         """
-        return cls(
-            dict(
-                virtual=SDECData.from_simulation(sim, "virtual"),
-                real=SDECData.from_simulation(sim, "real"),
+        if sim.runner.virt_logging:
+            return cls(
+                dict(
+                    virtual=SDECData.from_simulation(sim, "virtual"),
+                    real=SDECData.from_simulation(sim, "real"),
+                )
+            )
+        else:
+            raise ValueError(
+                "Virtual packet logging is inactive. "
+                "Please set config option virtual_packet_logging: True."
             )
-        )
 
     @classmethod
     def from_hdf(cls, hdf_fpath):