diff --git a/src/main/java/fr/univlille/sae/classification/model/Iris.java b/src/main/java/fr/univlille/sae/classification/model/Iris.java
index 78b880325e81201eab2ebc0eabdf0b38eb1612d7..20ba0c983f9cee3250b1e2899c0bcb6ba2ffc20b 100644
--- a/src/main/java/fr/univlille/sae/classification/model/Iris.java
+++ b/src/main/java/fr/univlille/sae/classification/model/Iris.java
@@ -137,23 +137,8 @@ public class Iris extends LoadableData {
     }
 
 
-    /**
-     * Renvoie la couleur associée à la variété de l'Iris.
-     * @return couleur correspondant à la variété.
-     */
-    @Override
-    public Color getColor() {
-        switch (this.variety) {
-            case "Setosa":
-                return Color.RED;
-            case "Versicolor":
-                return Color.BLUE;
-            case "Virginica":
-                return Color.GREEN;
-            default:
-                return Color.BLACK; // Couleur par défaut si la variété est inconnue
-        }
-    }
+   
+
 
     /**
      * Renvoie les noms des attributs de l'Iris.
diff --git a/src/main/java/fr/univlille/sae/classification/model/LoadableData.java b/src/main/java/fr/univlille/sae/classification/model/LoadableData.java
index ae3f033e92c77040d04bb0342008e2819e4bd136..04aadaded9d64cec60d8f6df63bcbd2f5b2a3995 100644
--- a/src/main/java/fr/univlille/sae/classification/model/LoadableData.java
+++ b/src/main/java/fr/univlille/sae/classification/model/LoadableData.java
@@ -84,11 +84,11 @@ public abstract class LoadableData {
      */
     public abstract String[] getAttributesName();
 
-    /**
-     * Renvoie la couleur associée à l'objet.
-     * @return couleur correspondant à la classification de l'objet.
-     */
-    public abstract Color getColor();
+
+
+
+
+
 
     /**
      * Renvoie la valeur des données en fonction de l'axe spécifié.
diff --git a/src/main/java/fr/univlille/sae/classification/model/Pokemon.java b/src/main/java/fr/univlille/sae/classification/model/Pokemon.java
index 1311287cdc4219dda769e5a2c94a2311dbddf904..b6d65eed681d452df28b40406416c6f727a00f43 100644
--- a/src/main/java/fr/univlille/sae/classification/model/Pokemon.java
+++ b/src/main/java/fr/univlille/sae/classification/model/Pokemon.java
@@ -106,54 +106,7 @@ public class Pokemon extends LoadableData{
         };
     }
 
-    /**
-     * Renvoie la couleur associée à l'objet.
-     *
-     * @return couleur correspondant à la classification de l'objet.
-     */
-    @Override
-    public Color getColor() {
-        switch (this.type1) {
-            case "normal":
-                return Color.LIGHTGREY;
-            case "grass":
-                return Color.GREEN;
-            case "electric":
-                return Color.YELLOW;
-            case "bug":
-                return Color.GREENYELLOW;
-            case "psychic":
-                return Color.PLUM;
-            case "poison":
-                return Color.PURPLE;
-            case "steel":
-                return Color.SILVER;
-            case "dragon":
-                return Color.WHITE;
-            case "flying":
-                return Color.SKYBLUE;
-            case "water":
-                return Color.BLUE;
-            case "rock":
-                return Color.SIENNA;
-            case "fire":
-                return Color.RED;
-            case "fairy":
-                return Color.PINK;
-            case "fighting":
-                return Color.FIREBRICK;
-            case "ice":
-                return Color.DARKTURQUOISE;
-            case "ghost":
-                return Color.DARKMAGENTA;
-            case "dark":
-                return Color.GREY;
-            case "ground":
-                return Color.KHAKI;
-            default:
-                return Color.BLACK; // Couleur par défaut si la variété est inconnue
-        }
-    }
+
 
     /**
      * Renvoie la valeur des données en fonction de l'axe spécifié.
diff --git a/src/test/java/fr/univlille/sae/classification/model/IrisTest.java b/src/test/java/fr/univlille/sae/classification/model/IrisTest.java
index e0873a5f3a78475d65d874d2e7d40ab6788a9f0a..824be5f4bcf220de6165f03d342622ef2316bb37 100644
--- a/src/test/java/fr/univlille/sae/classification/model/IrisTest.java
+++ b/src/test/java/fr/univlille/sae/classification/model/IrisTest.java
@@ -40,7 +40,7 @@ class IrisTest {
 
     @Test
     void getColor() {
-        assertEquals(Color.RED, iris.getColor());
+        assertEquals(Color.RED, iris.getClassifications().get(iris.getClassification()));
     }
 
     @Test