From ac255b292ce172a4b674e162f1b325a6f2d788ff Mon Sep 17 00:00:00 2001 From: Lenz Fiedler Date: Wed, 15 Jan 2025 10:31:03 +0100 Subject: [PATCH] Small adjustment in test --- test/on_the_fly_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/on_the_fly_test.py b/test/on_the_fly_test.py index 28d20916..2074ceb4 100644 --- a/test/on_the_fly_test.py +++ b/test/on_the_fly_test.py @@ -20,7 +20,6 @@ class TestOnTheFly: @staticmethod def __setup_training(lazy_loading, checkpoint_name=None): test_parameters = mala.Parameters() - test_parameters.use_lammps = False test_parameters.data.data_splitting_type = "by_snapshot" test_parameters.data.input_rescaling_type = "feature-wise-standard" test_parameters.data.output_rescaling_type = "minmax" @@ -177,4 +176,5 @@ def test_shuffling(self): test_parameters, test_network, data_handler ) test_trainer.train_network() + data_shuffler.delete_temporary_shuffled_snapshots() assert test_trainer.final_validation_loss < np.inf