From 59b950094481030f3715ef5954332d5258a88eeb Mon Sep 17 00:00:00 2001 From: nvieira-mcgill Date: Mon, 27 Jan 2025 18:27:41 -0500 Subject: [PATCH] Updated packet_source to fix bug with BlackBodySimpleSource.create_packets() not taking args or kwargs --- tardis/transport/montecarlo/packet_source.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tardis/transport/montecarlo/packet_source.py b/tardis/transport/montecarlo/packet_source.py index 9e9410e3a4c..ef63587721d 100644 --- a/tardis/transport/montecarlo/packet_source.py +++ b/tardis/transport/montecarlo/packet_source.py @@ -313,7 +313,7 @@ def __init__(self, time_explosion=None, **kwargs): self.time_explosion = time_explosion super().__init__(**kwargs) - def create_packets(self, no_of_packets): + def create_packets(self, no_of_packets, *args, **kwargs): """Generate relativistic black-body packet properties as arrays Parameters @@ -335,7 +335,7 @@ def create_packets(self, no_of_packets): if self.radius is None or self.time_explosion is None: raise ValueError("Black body Radius or Time of Explosion isn't set") self.beta = (self.radius / self.time_explosion) / const.c - return super().create_packets(no_of_packets) + return super().create_packets(no_of_packets, *args, **kwargs) def create_packet_mus(self, no_of_packets): """