def _sample(self, graph): sampled_graph = GraphLearnSampler._sample(self, graph) if self.num_scores> self.MIN_SCORES_2_PREDICT: if self.num_scores - self.num_scores_fitted > self.MIN_BATCH_SIZE: self.create_features() self.cip_fit() print("FITTED ", self.num_scores_fitted) return sampled_graph