d3bfdfbfc4609dd786b18140b78fa84bb1d78731,snorkel/viewer.py,Viewer,save_labels,#Viewer#,159

Before Change

    def save_labels(self):
        for cid, v in self.get_labels():
            existing_annotation = self.session.query(Annotation) \
                .filter(Annotation.annotator == self.annotator) \
                .filter(Annotation.candidate == self.candidates[cid]) \
                .first()
            if existing_annotation is not None:
                existing_annotation.value = v
            else:
                self.session.add(Annotation(annotator=self.annotator, candidate=self.candidates[cid], value=v))

After Change

        
        for cid, v in self._get_labels():
            if self.annotations[cid] is not None:
                if self.annotations[cid].value != v:
                    self.annotations[cid].value = v
            else:
                self.annotations[cid] = Annotation(annotator=self.annotator, candidate=self.candidates[cid], value=v)
                self.session.add(self.annotations[cid])
        self.session.commit()

Instances