Changing default alloc func to only give gen_specs['in'] if it's a fi…

…eld with length
Libensemble · Oct 8, 2019 · 29eabfe · 29eabfe
1 parent 926d0fa
commit 29eabfe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/libensemble/alloc_funcs/give_sim_work_first.py b/libensemble/alloc_funcs/give_sim_work_first.py
@@ -78,6 +78,9 @@ def give_sim_work_first(W, H, sim_specs, gen_specs, alloc_specs, persis_info):
 
             # Give gen work
             gen_count += 1
-            gen_work(Work, i, gen_specs['in'], range(len(H)), persis_info[i])
+            if 'in' in gen_specs and len(gen_specs['in']):
+                gen_work(Work, i, gen_specs['in'], range(len(H)), persis_info[i])
+            else:
+                gen_work(Work, i, [], [], persis_info[i])
 
     return Work, persis_info
diff --git a/libensemble/tests/regression_tests/test_1d_uniform_sampling.py b/libensemble/tests/regression_tests/test_1d_uniform_sampling.py
@@ -26,7 +26,6 @@
 sim_specs = {'sim_f': sim_f, 'in': ['x'], 'out': [('f', float)]}
 
 gen_specs = {'gen_f': gen_f,
-             'in': ['sim_id'],
              'out': [('x', float, (1,))],
              'lb': np.array([-3]),
              'ub': np.array([3]),