FeatureExtractionPipeline.py 文件源码

def _extract_for_single_repo_single_extractor(self,
                                                  extractor: FeatureExtractor,
                                                  repo: Repository,
                                                  api_repo: GithubRepository) -> [Feature]:
        cached_features = []
        for feature in extractor.features:
            cached_feature = self._get_cached(repo, feature)
            if self.cache_features and cached_feature:
                cached_features.append(cached_feature)
            else:
                updated_features = extractor.extract(api_repo)
                for updated_feature in updated_features:
                    self._update_cached(repo, updated_feature)
                return updated_features
        return cached_features