diff --git a/tests/test_ax.py b/tests/test_ax.py index 0dc196f..e59ed70 100644 --- a/tests/test_ax.py +++ b/tests/test_ax.py @@ -1,14 +1,13 @@ from ax import optimize -import logging from logging import CRITICAL -from ax.utils.common.logger import get_root_logger -rt = get_root_logger() -rt.setLevel(CRITICAL) +from ax.utils.common.logger import get_logger import warnings +rt = get_logger(name='ax') +rt.setLevel(CRITICAL) warnings.filterwarnings("ignore", category=UserWarning) -def dont_test_intro_example(): +def test_intro_example(): """ https://ax.dev/ """ best_parameters, best_values, experiment, model = optimize( parameters=[ @@ -31,7 +30,7 @@ def dont_test_intro_example(): if __name__ == '__main__': - dont_test_intro_example() + test_intro_example() warning_0_1_16 = """ diff --git a/tuneup/tangential_examples/ndimboxobjectives.py b/tuneup/tangential_examples/ndimboxobjectives.py deleted file mode 100644 index 4318fef..0000000 --- a/tuneup/tangential_examples/ndimboxobjectives.py +++ /dev/null @@ -1,22 +0,0 @@ -# A master list of objective functions defined on cubes [-scale,scale]^3 - -from deap import benchmarks - -# We'll use DEAP's set of groovy benchmarks. -# See pretty pictures at https://deap.readthedocs.io/en/master/api/benchmarks.html#deap.benchmarks -# In the format {function:scale} - -DEAP_OBJECTIVES = {benchmarks.schaffer:100, - benchmarks.sphere:1, - benchmarks.bohachevsky:100, - benchmarks.griewank:600, - benchmarks.rastrigin:5.12, - benchmarks.rastrigin_scaled:5.12, - benchmarks.rastrigin_skew:5.12, - benchmarks.schwefel:500, - benchmarks.rosenbrock:5} - -AN_OBJECTIVE = (benchmarks.schwefel,5.12) - -# But also some from a "real" filtering problem -OBJECTIVES = DEAP_OBJECTIVES