small

BlueBrain · Nov 21, 2023 · 301455f · 301455f
1 parent 2cd4c23
commit 301455f
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/neurots/generate/orientations.py b/neurots/generate/orientations.py
@@ -312,8 +312,13 @@ def prob(proposal):
             )
             params = self._parameters[tree_type]["orientation"]["values"]["params"]
             p = _prob(val, *params)
-            if self._context is not None and "trunk_prob" in self._context:  # pragma: no cover
-                p *= self._context["trunk_prob"](proposal, self._soma.center)
+
+            if self._context is not None and self._context.get(
+                "constraints", []
+            ):  # pragma: no cover
+                for constraint in self._context["constraints"]:
+                    if "trunk_prob" in constraint:
+                        p *= constraint["trunk_prob"](proposal, self._soma.center)
             return p
 
         def default_propose():