diff --git a/examples/main.py b/examples/main.py index 0b3bf63e..859c655b 100755 --- a/examples/main.py +++ b/examples/main.py @@ -23,7 +23,7 @@ import models.simple_models as md from black_it.calibrator import Calibrator -from black_it.loss_functions.minkowski import MinkowskiLoss +from black_it.loss_functions.msm import MethodOfMomentsLoss from black_it.samplers.best_batch import BestBatchSampler from black_it.samplers.halton import HaltonSampler from black_it.samplers.random_forest import RandomForestSampler @@ -53,7 +53,7 @@ print("Real data:", real_data) # define a loss - loss = MinkowskiLoss() + loss = MethodOfMomentsLoss() # define the calibration seed calibration_seed = 1 diff --git a/tests/test_examples/test_main.py b/tests/test_examples/test_main.py index 6c43d2c3..9cbb41e2 100644 --- a/tests/test_examples/test_main.py +++ b/tests/test_examples/test_main.py @@ -24,7 +24,7 @@ TRUE_PARAMETERS_STR = "True parameters: [0.2, 0.2, 0.75]" -BEST_PARAMETERS_STR = "Best parameters found: [0.11 0.13 0.87]" +BEST_PARAMETERS_STR = "Best parameters found: [0.19 0.21 0.68]" class TestMainExample(BaseMainExampleTestClass):