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