Update neurots/generate/section.py

Co-authored-by: Adrien Berchet <[email protected]>
BlueBrain · Sep 25, 2024 · 1f082b9 · 1f082b9
1 parent 3bffa0d
commit 1f082b9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/neurots/generate/section.py b/neurots/generate/section.py
@@ -128,8 +128,8 @@ def prob(*args, **kwargs):
                     p *= constraint["section_prob"](*args, **kwargs)
                 return p
 
-            max_tries = DEFAULT_MAX_TRIES
-            randomness_increase = DEFAULT_RANDOMNESS_INCREASE
+            max_tries = -1
+            randomness_increase = -1
             for constraint in self.context["constraints"]:
                 max_tries = max(
                     max_tries,
@@ -141,6 +141,10 @@ def prob(*args, **kwargs):
                         "randomness_increase", DEFAULT_RANDOMNESS_INCREASE
                     ),
                 )
+            if max_tries < 0:
+                max_tries = DEFAULT_MAX_TRIES
+            if randomness_increase < 0:
+                randomness_increase = DEFAULT_RANDOMNESS_INCREASE
 
             direction = accept_reject(
                 self._propose,