From 11d6fae28b994eb5c0231cd4ed620c66d62ba472 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?David=20Silva-S=C3=A1nchez?=
 <31875788+DSilva27@users.noreply.github.com>
Date: Tue, 6 Aug 2024 09:03:18 -0400
Subject: [PATCH] Revert "expanded ice cream flavor list"

---
 .../_preprocessing/preprocessing_pipeline.py  | 50 ++++++++-----------
 1 file changed, 20 insertions(+), 30 deletions(-)

diff --git a/src/cryo_challenge/_preprocessing/preprocessing_pipeline.py b/src/cryo_challenge/_preprocessing/preprocessing_pipeline.py
index d0de164..589239c 100644
--- a/src/cryo_challenge/_preprocessing/preprocessing_pipeline.py
+++ b/src/cryo_challenge/_preprocessing/preprocessing_pipeline.py
@@ -42,36 +42,26 @@ def save_submission(volumes, populations, submission_id, submission_index, confi
 def preprocess_submissions(submission_dataset, config):
     np.random.seed(config["seed_flavor_assignment"])
     ice_cream_flavors = [
-    "Chocolate",
-    "Vanilla",
-    "Cookies N' Cream",
-    "Mint Chocolate Chip",
-    "Strawberry",
-    "Butter Pecan",
-    "Salted Caramel",
-    "Pistachio",
-    "Rocky Road",
-    "Coffee",
-    "Cookie Dough",
-    "Chocolate Chip",
-    "Neapolitan",
-    "Cherry",
-    "Rainbow Sherbet",
-    "Peanut Butter",
-    "Cotton Candy",
-    "Lemon Sorbet",
-    "Mango",
-    "Black Raspberry",
-    "Pina Colada",
-    "Birthday Cake",
-    "Maple Walnut",
-    "Tiramisu",
-    "Coconut",
-    "Chocolate Fudge Brownie",
-    "Dulce de Leche",
-    "Green Tea",
-    "Rum Raisin",
-    "Blueberry Cheesecake"
+        "Chocolate",
+        "Vanilla",
+        "Cookies N' Cream",
+        "Mint Chocolate Chip",
+        "Strawberry",
+        "Butter Pecan",
+        "Salted Caramel",
+        "Pistachio",
+        "Rocky Road",
+        "Coffee",
+        "Cookie Dough",
+        "Chocolate Chip",
+        "Neapolitan",
+        "Cherry",
+        "Rainbow Sherbet",
+        "Peanut Butter",
+        "Cotton Candy",
+        "Lemon Sorbet",
+        "Mango",
+        "Black Raspberry",
     ]
 
     n_subs = max(submission_dataset.subs_index) + 1