From 0f0e17c8f36c53aec1e00f377db78dae1c55ea30 Mon Sep 17 00:00:00 2001
From: benleetownsend <benlt@hotmail.co.uk>
Date: Mon, 13 Jul 2020 18:15:19 +0100
Subject: [PATCH] FIX: duplicate examples in train

---
 enso/sample/random_sampler.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/enso/sample/random_sampler.py b/enso/sample/random_sampler.py
index eb6f66e..bbfd114 100644
--- a/enso/sample/random_sampler.py
+++ b/enso/sample/random_sampler.py
@@ -100,7 +100,9 @@ def _choose_starting_points(self, n_points=3):
             for cls in self.classes:
                 indices = [i for i, val in enumerate(self.train_labels) if cls in val]
                 index = random.choice(indices)
-                points.append(self.train_indices[index])
+                train_index = self.train_indices[index]
+                if train_index not in points:
+                    points.append(train_index)
         return points
 
     @property