Mic-4920/draw-max-value-500 (#212)

* Limit draws to 500 * Make number of draws constant
ihmeuw · Mar 13, 2024 · 7ceb308 · 7ceb308
1 parent 5a754c5
commit 7ceb308
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -1,3 +1,7 @@
+**1.6.0 - 03/12/24**
+
+ - Limit max number of draws to 500.
+
 **1.5.6 - 02/27/24**
 
  - Give user option to continue psimulate restart if environment has changed

diff --git a/src/vivarium_cluster_tools/psimulate/branches.py b/src/vivarium_cluster_tools/psimulate/branches.py
@@ -16,6 +16,8 @@
 
 from vivarium_cluster_tools.psimulate.model_specification import FULL_ARTIFACT_PATH_KEY
 
+NUMBER_OF_DRAWS = 500
+
 
 class Keyspace:
     """A representation of a collection of simulation configurations."""
@@ -136,7 +138,7 @@ def calculate_input_draws(
         existing draw numbers.
 
     """
-    max_draw_count = 1000
+    max_draw_count = NUMBER_OF_DRAWS
     if input_draw_count > max_draw_count:
         raise ValueError(f"Input draw count must be less than {max_draw_count}.")