b9dc15ecef0393a55a953acff3c1d0278b19d9f5,src/fonduer/candidates/candidates.py,CandidateExtractor,clear,#CandidateExtractor#Any#Any#,62

Before Change

    def clear(self, session, split, **kwargs):
        Delete Candidates of each class from given split the database.
        for candidate_class in self.candidate_classes:
            logger.info(
                "Clearing table {} (split {})".format(
                    candidate_class.__tablename__, split
                )
            )
            session.query(Candidate).filter(
                Candidate.type == candidate_class.__tablename__
            ).filter(Candidate.split == split).delete()

    def clear_all(self, split, **kwargs):
        Delete all Candidates from given split the database.
        logger.info("Clearing ALL Candidates.")
        self.session.query(Candidate).filter(Candidate.split == split).delete()

After Change

            logger.info(
                "Clearing table {} (split {})".format(
                    candidate_class.__tablename__, split
                )
            )
            self.session.query(Candidate).filter(
                Candidate.type == candidate_class.__tablename__
            ).filter(Candidate.split == split).delete()

Instances