diff --git a/sampling_methods/kcenter_greedy.py b/sampling_methods/kcenter_greedy.py index ff7e548..d9b8a05 100644 --- a/sampling_methods/kcenter_greedy.py +++ b/sampling_methods/kcenter_greedy.py @@ -102,7 +102,7 @@ def select_batch_(self, model, already_selected, N, **kwargs): new_batch = [] for _ in range(N): - if self.already_selected is None: + if self.min_distances is None: # Initialize centers with a randomly selected datapoint ind = np.random.choice(np.arange(self.n_obs)) else: