Add validation for gradient names in Gradient.py

mesoscope · Jul 8, 2024 · 04898ee · 04898ee
1 parent f4c28ae
commit 04898ee
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cellpack/autopack/Gradient.py b/cellpack/autopack/Gradient.py
@@ -200,6 +200,10 @@ def pick_point_for_ingredient(ingr, allIngrPts, all_gradients):
                 if not hasattr(ingr, "combined_weight"):
                     gradient_list = []
                     for gradient_name in ingr.gradient:
+                        if gradient_name not in all_gradients:
+                            raise ValueError(
+                                f"Gradient {gradient_name} not found in gradient list"
+                            )
                         gradient_list.append(all_gradients[gradient_name])
 
                     combined_weight = Gradient.get_combined_gradient_weight(