fix tests

2025-11-13 17:03:31 +01:00
parent 0ee63686dd
commit 8440a672a7
5 changed files with 289 additions and 300 deletions
--- a/tests/test_callback/test_adaptive_refinement_callback.py
+++ b/tests/test_callback/test_adaptive_refinement_callback.py
@@ -51,7 +51,7 @@ def test_sample(condition_to_update):
    }
    trainer.train()
    after_n_points = {
-        loc: len(trainer.data_module.train_dataset.input[loc])
+        loc: len(trainer.data_module.train_dataset[loc].input)
        for loc in condition_to_update
    }
    assert before_n_points == trainer.callbacks[0].initial_population_size