Black and isort

jmlarson1 · jmlarson1 · commit 3a744718605e · 2026-01-13T15:21:42.000-06:00
diff --git a/libensemble/tests/regression_tests/support.py b/libensemble/tests/regression_tests/support.py
@@ -1,7 +1,6 @@
 import copy
 
 import numpy as np
-
 from libensemble.specs import input_fields, output_data
 
 branin_vals_and_minima = np.array(
diff --git a/libensemble/tests/regression_tests/test_1d_sampling.py b/libensemble/tests/regression_tests/test_1d_sampling.py
@@ -14,7 +14,6 @@
 # TESTSUITE_NPROCS: 2 4
 
 import numpy as np
-
 from libensemble import Ensemble
 from libensemble.gen_funcs.sampling import latin_hypercube_sample as gen_f
 
diff --git a/libensemble/tests/regression_tests/test_2d_sampling.py b/libensemble/tests/regression_tests/test_2d_sampling.py
@@ -14,7 +14,6 @@
 # TESTSUITE_NPROCS: 2 4
 
 import numpy as np
-
 from libensemble import Ensemble
 from libensemble.gen_funcs.sampling import latin_hypercube_sample as gen_f
 
diff --git a/libensemble/tests/regression_tests/test_GPU_variable_resources.py b/libensemble/tests/regression_tests/test_GPU_variable_resources.py
@@ -26,7 +26,6 @@
 # TESTSUITE_NPROCS: 6
 
 import numpy as np
-
 from libensemble import Ensemble
 from libensemble.alloc_funcs.start_only_persistent import only_persistent_gens as alloc_f
 from libensemble.executors.mpi_executor import MPIExecutor
diff --git a/libensemble/tests/regression_tests/test_GPU_variable_resources_multi_task.py b/libensemble/tests/regression_tests/test_GPU_variable_resources_multi_task.py
@@ -35,7 +35,6 @@
 # TESTSUITE_NPROCS: 10
 
 import numpy as np
-
 from libensemble import Ensemble
 from libensemble.alloc_funcs.start_only_persistent import only_persistent_gens as alloc_f
 from libensemble.executors.mpi_executor import MPIExecutor
diff --git a/libensemble/tests/regression_tests/test_evaluate_mixed_sample.py b/libensemble/tests/regression_tests/test_evaluate_mixed_sample.py
@@ -17,7 +17,6 @@
 import warnings
 
 import numpy as np
-
 from libensemble import Ensemble
 from libensemble.alloc_funcs.give_pregenerated_work import give_pregenerated_sim_work as alloc_f
 
@@ -33,9 +32,7 @@
     samp = 1000
     n = 8
 
-    H0 = np.zeros(
-        samp, dtype=[("x", float, n), ("f", float), ("sim_id", int), ("sim_started", bool), ("sim_ended", bool)]
-    )
+    H0 = np.zeros(samp, dtype=[("x", float, n), ("f", float), ("sim_id", int), ("sim_started", bool), ("sim_ended", bool)])
     np.random.seed(0)
     H0["x"] = gen_borehole_input(samp)
     for i in range(500):
diff --git a/libensemble/tests/regression_tests/test_gpCAM.py b/libensemble/tests/regression_tests/test_gpCAM.py
@@ -27,7 +27,6 @@
 import warnings
 
 import numpy as np
-
 from libensemble.alloc_funcs.start_only_persistent import only_persistent_gens as alloc_f
 from libensemble.gen_funcs.persistent_gpCAM import persistent_gpCAM, persistent_gpCAM_covar
 
diff --git a/libensemble/tests/regression_tests/test_inverse_bayes_example.py b/libensemble/tests/regression_tests/test_inverse_bayes_example.py
@@ -19,7 +19,6 @@
 # TESTSUITE_NPROCS: 3 4
 
 import numpy as np
-
 from libensemble import Ensemble
 from libensemble.alloc_funcs.inverse_bayes_allocf import only_persistent_gens_for_inverse_bayes as alloc_f
 from libensemble.gen_funcs.persistent_inverse_bayes import persistent_updater_after_likelihood as gen_f
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_dfols.py b/libensemble/tests/regression_tests/test_persistent_aposmm_dfols.py
@@ -21,9 +21,8 @@
 import multiprocessing
 import sys
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 
 # Import libEnsemble items for this test
 from libensemble.libE import libE
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_exception.py b/libensemble/tests/regression_tests/test_persistent_aposmm_exception.py
@@ -18,9 +18,8 @@
 import multiprocessing
 import sys
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 
 # Import libEnsemble items for this test
 from libensemble.libE import libE
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_external_localopt.py b/libensemble/tests/regression_tests/test_persistent_aposmm_external_localopt.py
@@ -26,9 +26,8 @@
 import shutil  # For copying the external_localopt script
 import sys
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 
 # Import libEnsemble items for this test
 from libensemble.libE import libE
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_ibcdfo_manifold_sampling.py b/libensemble/tests/regression_tests/test_persistent_aposmm_ibcdfo_manifold_sampling.py
@@ -27,9 +27,8 @@
 import multiprocessing
 import sys
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 from libensemble.libE import libE
 
 libensemble.gen_funcs.rc.aposmm_optimizers = "ibcdfo_manifold_sampling"
@@ -39,7 +38,7 @@
 from libensemble.tools import add_unique_random_streams, parse_args, save_libE_output
 
 try:
-    import ibcdfo # noqa: F401
+    import ibcdfo  # noqa: F401
 
 except ModuleNotFoundError:
     sys.exit("Please 'pip install ibcdfo'")
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_ibcdfo_pounders.py b/libensemble/tests/regression_tests/test_persistent_aposmm_ibcdfo_pounders.py
@@ -27,9 +27,8 @@
 import multiprocessing
 import sys
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 from libensemble.libE import libE
 from libensemble.sim_funcs.chwirut1 import chwirut_eval
 
@@ -40,7 +39,7 @@
 from libensemble.tools import add_unique_random_streams, parse_args, save_libE_output
 
 try:
-    import ibcdfo # noqa: F401
+    import ibcdfo  # noqa: F401
 except ModuleNotFoundError:
     sys.exit("Please 'pip install ibcdfo'")
 
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_ibcdfo_pounders_jax.py b/libensemble/tests/regression_tests/test_persistent_aposmm_ibcdfo_pounders_jax.py
@@ -27,9 +27,8 @@
 import multiprocessing
 import sys
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 from libensemble.libE import libE
 
 libensemble.gen_funcs.rc.aposmm_optimizers = "ibcdfo_pounders"
@@ -39,8 +38,9 @@
 from libensemble.tools import add_unique_random_streams, parse_args, save_libE_output
 
 try:
-    import ibcdfo # noqa: F401
-    from declare_hfun_and_combine_model_with_jax import hfun, combinemodels_jax
+    import ibcdfo  # noqa: F401
+
+    from declare_hfun_and_combine_model_with_jax import combinemodels_jax, hfun
 
 except ModuleNotFoundError:
     sys.exit("Please 'pip install ibcdfo'")
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_nlopt.py b/libensemble/tests/regression_tests/test_persistent_aposmm_nlopt.py
@@ -19,9 +19,8 @@
 import sys
 from math import gamma, pi, sqrt
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 
 # Import libEnsemble items for this test
 from libensemble.libE import libE
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_periodic.py b/libensemble/tests/regression_tests/test_persistent_aposmm_periodic.py
@@ -19,9 +19,8 @@
 import multiprocessing
 import sys
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 
 libensemble.gen_funcs.rc.aposmm_optimizers = ["nlopt", "scipy"]
 from libensemble.alloc_funcs.persistent_aposmm_alloc import persistent_aposmm_alloc as alloc_f
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_pounders.py b/libensemble/tests/regression_tests/test_persistent_aposmm_pounders.py
@@ -22,9 +22,8 @@
 import sys
 from math import ceil, gamma, pi, sqrt
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 
 # Import libEnsemble items for this test
 from libensemble.libE import libE
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_scipy.py b/libensemble/tests/regression_tests/test_persistent_aposmm_scipy.py
@@ -18,9 +18,8 @@
 import multiprocessing
 import sys
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 
 # Import libEnsemble items for this test
 from libensemble.libE import libE
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_tao_blmvm.py b/libensemble/tests/regression_tests/test_persistent_aposmm_tao_blmvm.py
@@ -20,9 +20,8 @@
 import sys
 from math import gamma, pi, sqrt
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 
 # Import libEnsemble items for this test
 from libensemble.libE import libE
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_tao_nm.py b/libensemble/tests/regression_tests/test_persistent_aposmm_tao_nm.py
@@ -19,9 +19,8 @@
 import multiprocessing
 import sys
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 
 # Import libEnsemble items for this test
 from libensemble.libE import libE
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_timeout.py b/libensemble/tests/regression_tests/test_persistent_aposmm_timeout.py
@@ -20,9 +20,8 @@
 import multiprocessing
 import sys
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 
 libensemble.gen_funcs.rc.aposmm_optimizers = "nlopt"
 
diff --git a/libensemble/tests/regression_tests/test_persistent_aposmm_with_grad.py b/libensemble/tests/regression_tests/test_persistent_aposmm_with_grad.py
@@ -21,9 +21,8 @@
 import sys
 from math import gamma, pi, sqrt
 
-import numpy as np
-
 import libensemble.gen_funcs
+import numpy as np
 
 # Import libEnsemble items for this test
 from libensemble.libE import libE
@@ -122,9 +121,7 @@
 
     if is_manager:
         assert persis_info[1].get("run_order"), "Run_order should have been given back"
-        assert (
-            len(persis_info[1]["run_order"]) >= gen_specs["user"]["stop_after_k_minima"]
-        ), "This test should have many runs started."
+        assert len(persis_info[1]["run_order"]) >= gen_specs["user"]["stop_after_k_minima"], "This test should have many runs started."
         assert len(H) < exit_criteria["sim_max"], "Test should have stopped early due to 'stop_after_k_minima'"
 
         print("[Manager]:", H[np.where(H["local_min"])]["x"])
diff --git a/libensemble/tests/regression_tests/test_persistent_fd_param_finder.py b/libensemble/tests/regression_tests/test_persistent_fd_param_finder.py
@@ -20,7 +20,6 @@
 import shutil  # For ECnoise.m
 
 import numpy as np
-
 from libensemble import Ensemble
 from libensemble.alloc_funcs.start_fd_persistent import finite_diff_alloc as alloc_f
 from libensemble.gen_funcs.persistent_fd_param_finder import fd_param_finder as gen_f
diff --git a/libensemble/tests/regression_tests/test_persistent_gp_multitask_ax.py b/libensemble/tests/regression_tests/test_persistent_gp_multitask_ax.py
@@ -24,7 +24,6 @@
 import warnings
 
 import numpy as np
-
 from libensemble import logger
 from libensemble.alloc_funcs.start_only_persistent import only_persistent_gens
 from libensemble.libE import libE
diff --git a/libensemble/tests/regression_tests/test_persistent_surmise_calib.py b/libensemble/tests/regression_tests/test_persistent_surmise_calib.py
@@ -31,7 +31,6 @@
 #   Install Surmise package
 
 import numpy as np
-
 from libensemble import Ensemble
 from libensemble.alloc_funcs.start_only_persistent import only_persistent_gens as alloc_f
 from libensemble.gen_funcs.persistent_surmise_calib import surmise_calib as gen_f
diff --git a/libensemble/tests/regression_tests/test_persistent_surmise_killsims.py b/libensemble/tests/regression_tests/test_persistent_surmise_killsims.py
@@ -33,7 +33,6 @@
 import os
 
 import numpy as np
-
 from libensemble.alloc_funcs.start_only_persistent import only_persistent_gens as alloc_f
 from libensemble.executors.executor import Executor
 from libensemble.gen_funcs.persistent_surmise_calib import surmise_calib as gen_f
@@ -130,9 +129,7 @@
     exit_criteria = {"sim_max": max_evals}
 
     # Perform the run
-    H, persis_info, flag = libE(
-        sim_specs, gen_specs, exit_criteria, persis_info, alloc_specs=alloc_specs, libE_specs=libE_specs
-    )
+    H, persis_info, flag = libE(sim_specs, gen_specs, exit_criteria, persis_info, alloc_specs=alloc_specs, libE_specs=libE_specs)
 
     if is_manager:
         print("Cancelled sims", H["sim_id"][H["cancel_requested"]])
diff --git a/libensemble/tests/regression_tests/test_persistent_tasmanian.py b/libensemble/tests/regression_tests/test_persistent_tasmanian.py
@@ -21,7 +21,6 @@
 from time import time
 
 import numpy as np
-
 from libensemble.alloc_funcs.start_only_persistent import only_persistent_gens as alloc_f
 from libensemble.gen_funcs.persistent_tasmanian import sparse_grid_batched as gen_f_batched
 
diff --git a/libensemble/tests/regression_tests/test_persistent_tasmanian_async.py b/libensemble/tests/regression_tests/test_persistent_tasmanian_async.py
@@ -22,7 +22,6 @@
 from time import time
 
 import numpy as np
-
 from libensemble.gen_funcs.persistent_tasmanian import get_sparse_grid_specs
 
 # Import libEnsemble items for this test
diff --git a/libensemble/tests/regression_tests/test_with_app_persistent_aposmm_tao_nm.py b/libensemble/tests/regression_tests/test_with_app_persistent_aposmm_tao_nm.py
@@ -24,7 +24,6 @@
 import sys
 
 import numpy as np
-
 from libensemble.alloc_funcs.persistent_aposmm_alloc import persistent_aposmm_alloc as alloc_f
 from libensemble.executors import MPIExecutor
 from libensemble.gen_funcs.persistent_aposmm import aposmm as gen_f
