From 6a5fd2b87249501f857adc08cadd29696d7d6958 Mon Sep 17 00:00:00 2001 From: "Mats E. Mollestad" Date: Mon, 6 Nov 2023 22:28:35 +0100 Subject: [PATCH] fix: flaky test --- aligned/tests/test_model_target.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aligned/tests/test_model_target.py b/aligned/tests/test_model_target.py index 889e76a3..f378fbe9 100644 --- a/aligned/tests/test_model_target.py +++ b/aligned/tests/test_model_target.py @@ -124,7 +124,7 @@ class TestModel: await store.insert_into(FeatureLocation.model('test_model'), {'id': [1, 2, 3], 'a': [10, 14, 20]}) - stored_data = pl.read_parquet(path) + stored_data = pl.read_parquet(path).select(expected_frame.columns) assert stored_data.frame_equal(expected_frame)