8946a24d7c1a21a80ccaf86418c354cf87e84bfa,src/fonduer/supervision/labeler.py,LabelerUDF,apply,#LabelerUDF#Any#Any#Any#Any#Any#,205

Before Change

        )

        label_keys = set()
        for cands in cands_list:
            records = list(get_mapping(cands, self._f_gen, label_keys))
            batch_upsert_records(self.session, Label, records)

        // Insert all Label Keys
        if train or update:
            add_keys(self.session, LabelKey, label_keys)

        // This return + yield makes a completely empty generator

After Change

        self.lfs = lfs
        if docs:
            // Call apply on the specified docs for all splits
            split = ALL_SPLITS
            super(Labeler, self).apply(
                docs, split=split, train=train, lfs=self.lfs, clear=clear, **kwargs
            )
            // Needed to sync the bulk operations
            self.session.commit()
        else:
            // Only grab the docs containing candidates from the given split.
            split_docs = get_docs_from_split(
                self.session, self.candidate_classes, split
            )
            super(Labeler, self).apply(
                split_docs,
                split=split,
                train=train,
                lfs=self.lfs,
                clear=clear,
                **kwargs
            )
            // Needed to sync the bulk operations
            self.session.commit()

    def get_lfs(self):
        Return a list of lists of labeling functions for this Labeler.
        return self.lfs

Instances