diff --git a/aircox/models/episode.py b/aircox/models/episode.py index 0d8d956..7b51032 100644 --- a/aircox/models/episode.py +++ b/aircox/models/episode.py @@ -222,7 +222,7 @@ class Diffusion(BaseRerun): self.episode = self.initial.episode self.program = self.episode.program - def save_initial(self, *args, **kwargs): + def save_initial(self): self.program = self.episode.program @property diff --git a/aircox/models/program.py b/aircox/models/program.py index e39c1f5..0efc2e5 100644 --- a/aircox/models/program.py +++ b/aircox/models/program.py @@ -198,7 +198,7 @@ class BaseRerun(models.Model): if self.is_rerun: self.save_rerun() else: - self.save_initial(*args, **kwargs) + self.save_initial() super().save(*args, **kwargs) def save_rerun(self):