From 1454f92189642cb0fbb7f9842654be252b962644 Mon Sep 17 00:00:00 2001 From: Jacob Bieker Date: Thu, 29 Jul 2021 09:19:46 +0100 Subject: [PATCH] Fix last test --- satflow/tests/test_dataset.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/satflow/tests/test_dataset.py b/satflow/tests/test_dataset.py index bae0446f..46f3766c 100644 --- a/satflow/tests/test_dataset.py +++ b/satflow/tests/test_dataset.py @@ -149,11 +149,9 @@ def test_satflow_all_deterministic_validation_restart(): config = load_config("satflow/tests/configs/satflow_all.yaml") cloudflow = SatFlowDataset([dataset], config, train=False) data = next(iter(cloudflow)) - x, image, y = data + x, image = data data = next(iter(cloudflow)) - x2, image2, y2 = data + x2, image2 = data np.testing.assert_almost_equal(x, x2) np.testing.assert_almost_equal(image, image2) - np.testing.assert_almost_equal(y, y2) assert not np.allclose(image[0], image[-1]) - assert not np.allclose(y[0], y[-1])