fixes

Author: jlnav

libensemble/alloc_funcs/fast_alloc_and_pausing.py

@@ -35,8 +35,8 @@ def give_sim_work_first(W, H, sim_specs, gen_specs, alloc_specs, persis_info, li
     gen_count = support.count_gens()
 
     if gen_specs["user"].get("single_component_at_a_time"):
-        assert (
-            alloc_specs["user"]["batch_mode"] or gen_specs["batch_mode"]
+        assert alloc_specs["user"].get("batch_mode", False) or gen_specs.get(
+            "batch_mode", False
         ), "Must be in batch mode when using 'single_component_at_a_time'"
     if len(H) != persis_info["H_len"]:
         # Something new is in the history.

libensemble/tests/functionality_tests/test_mpi_gpu_settings.py

@@ -88,8 +88,8 @@
         "out": [("priority", float), ("resource_sets", int), ("x", float, n)],
         "give_all_with_same_priority": False,
         "async_return": False,
+        "initial_batch_size": nworkers - 1,
         "user": {
-            "initial_batch_size": nworkers - 1,
             "max_resource_sets": nworkers - 1,  # Any sim created can req. 1 worker up to all.
             "lb": np.array([-3, -2]),
             "ub": np.array([3, 2]),