forked from rc/aircox
save diffusion error
This commit is contained in:
parent
c1bf1c34f0
commit
4eef440a2a
|
@ -208,6 +208,11 @@ class Diffusion(BaseRerun):
|
|||
str_ += ' ({})'.format(_('rerun'))
|
||||
return str_
|
||||
|
||||
def save(self, *args, **kwargs):
|
||||
super().save(*args, **kwargs)
|
||||
if self.is_initial and self.episode != self._initial['episode']:
|
||||
self.rerun_set.update(episode=self.episode, program=self.program)
|
||||
|
||||
#def save(self, no_check=False, *args, **kwargs):
|
||||
#if self.start != self._initial['start'] or \
|
||||
# self.end != self._initial['end']:
|
||||
|
@ -217,10 +222,8 @@ class Diffusion(BaseRerun):
|
|||
self.episode = self.initial.episode
|
||||
self.program = self.episode.program
|
||||
|
||||
def save_initial(self):
|
||||
def save_initial(self, *args, **kwargs):
|
||||
self.program = self.episode.program
|
||||
if self.episode != self._initial['episode']:
|
||||
self.rerun_set.update(episode=self.episode, program=self.program)
|
||||
|
||||
@property
|
||||
def duration(self):
|
||||
|
|
|
@ -198,7 +198,7 @@ class BaseRerun(models.Model):
|
|||
if self.is_rerun:
|
||||
self.save_rerun()
|
||||
else:
|
||||
self.save_initial()
|
||||
self.save_initial(*args, **kwargs)
|
||||
super().save(*args, **kwargs)
|
||||
|
||||
def save_rerun(self):
|
||||
|
|
Loading…
Reference in New Issue
Block a user