diff --git a/src/pg_rad/simulator/engine.py b/src/pg_rad/simulator/engine.py index 7b8c950..f22305c 100644 --- a/src/pg_rad/simulator/engine.py +++ b/src/pg_rad/simulator/engine.py @@ -34,6 +34,7 @@ class SimulationEngine: return SimulationOutput( name=self.landscape.name, + size=self.landscape.size, count_rate=count_rate_results, sources=source_results ) diff --git a/src/pg_rad/simulator/outputs.py b/src/pg_rad/simulator/outputs.py index 9aca77e..404e742 100644 --- a/src/pg_rad/simulator/outputs.py +++ b/src/pg_rad/simulator/outputs.py @@ -27,5 +27,6 @@ class SourceOutput: @dataclass class SimulationOutput: name: str + size: tuple count_rate: CountRateOutput sources: List[SourceOutput]