diff --git a/src/cryo_challenge/_preprocessing/preprocessing_pipeline.py b/src/cryo_challenge/_preprocessing/preprocessing_pipeline.py index 589239c..d0de164 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