From c405a7e337039f59da8be8ead0bd15ef728092e1 Mon Sep 17 00:00:00 2001
From: Miro Astore <miro.astore@gmail.com>
Date: Mon, 5 Aug 2024 16:03:06 -0400
Subject: [PATCH] expanded ice cream flavor list

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

diff --git a/src/cryo_challenge/_preprocessing/preprocessing_pipeline.py b/src/cryo_challenge/_preprocessing/preprocessing_pipeline.py
index 926b2c1..3e8da53 100644
--- a/src/cryo_challenge/_preprocessing/preprocessing_pipeline.py
+++ b/src/cryo_challenge/_preprocessing/preprocessing_pipeline.py
@@ -42,26 +42,36 @@ 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",
+    "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"
     ]
 
     n_subs = max(submission_dataset.subs_index) + 1