obtaining DeprecationWarning and ndim warning fixes

Libensemble · Jul 3, 2023 · e4703c1 · e4703c1
1 parent 3fe2dcf
commit e4703c1
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 3 deletions.
diff --git a/libensemble/gen_funcs/persistent_fd_param_finder.py b/libensemble/gen_funcs/persistent_fd_param_finder.py
@@ -88,7 +88,7 @@ def fd_param_finder(H, persis_info, gen_specs, libE_info):
             for k in range(nf + 1):
                 if k != nf / 2:
                     logical_conds = (calc_in["x_ind"] == i, calc_in["f_ind"] == j, calc_in["n_ind"] == k)
-                    Fhist0[i, j, k] = calc_in["f_val"][np.logical_and.reduce(logical_conds)]
+                    Fhist0[i, j, k] = calc_in["f_val"][np.logical_and.reduce(logical_conds)][0]
 
             # Compute noise for (i, j):
             # [Fnoise(i, j), ~, inform(i, j)] = ECnoise(nf-1, Fhist0(i, j, 2:nf));

diff --git a/libensemble/sim_funcs/borehole.py b/libensemble/sim_funcs/borehole.py
@@ -65,7 +65,7 @@ def borehole_func(x):
     denom1 = 2 * L * Tu / (np.log(r / rw) * rw**2 * Kw)
     denom2 = Tu / Tl
 
-    return (numer / (np.log(r / rw) * (1 + denom1 + denom2))).reshape(-1)
+    return ((numer / (np.log(r / rw) * (1 + denom1 + denom2))).reshape(-1))[0]
 
 
 def gen_borehole_input(n):

diff --git a/libensemble/sim_funcs/rosenbrock.py b/libensemble/sim_funcs/rosenbrock.py
@@ -31,7 +31,7 @@ def EvaluateFunction(x, component):
         x2 = x[2 * i + 1]
         f = 100 * (x1**2 - x2) ** 2 + (x1 - 1) ** 2
 
-    return f
+    return f[0]
 
 
 def EvaluateJacobian(x, component, const):

diff --git a/libensemble/tests/regression_tests/test_evaluate_mixed_sample.py b/libensemble/tests/regression_tests/test_evaluate_mixed_sample.py
@@ -14,6 +14,8 @@
 # TESTSUITE_COMMS: mpi local tcp
 # TESTSUITE_NPROCS: 2 4
 
+import warnings
+
 import numpy as np
 
 from libensemble.alloc_funcs.give_pregenerated_work import give_pregenerated_sim_work as alloc_f
@@ -24,6 +26,8 @@
 from libensemble.sim_funcs.borehole import borehole_func, gen_borehole_input
 from libensemble.tools import parse_args, save_libE_output
 
+warnings.filterwarnings("ignore", category=DeprecationWarning)
+
 # Main block is necessary only when using local comms with spawn start method (default on macOS and Windows).
 if __name__ == "__main__":
     nworkers, is_manager, libE_specs, _ = parse_args()

diff --git a/libensemble/tests/regression_tests/test_ytopt_heffte.py b/libensemble/tests/regression_tests/test_ytopt_heffte.py
@@ -21,6 +21,7 @@
 warnings.filterwarnings("ignore", category=ImportWarning)
 warnings.filterwarnings("ignore", category=FutureWarning)
 warnings.filterwarnings("ignore", category=RuntimeWarning)
+warnings.filterwarnings("ignore", category=DeprecationWarning)
 
 import os
 import secrets