Merge pull request #65 from flatironinstitute/expand_flavors

expanded ice cream flavor list
flatironinstitute · Aug 6, 2024 · f6ae1e6 · f6ae1e6
2 parents d7fd640 + c405a7e
commit f6ae1e6
Showing 1 changed file with 30 additions and 20 deletions.
diff --git 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