Fix last test

openclimatefix · Jul 29, 2021 · 1454f92 · 1454f92
1 parent 4b79071
commit 1454f92
Showing 1 changed file with 2 additions and 4 deletions.
diff --git 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])