From 3a7e8710d9d039f5d5b5bfaa569a61c4a7bac744 Mon Sep 17 00:00:00 2001 From: Jacob Klegar Date: Fri, 15 Jan 2021 18:13:41 -0500 Subject: [PATCH] fix historical test flakiness Signed-off-by: Jacob Klegar --- tests/e2e/test_historical_features.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/e2e/test_historical_features.py b/tests/e2e/test_historical_features.py index 22fa1721c3..38d6ecb47c 100644 --- a/tests/e2e/test_historical_features.py +++ b/tests/e2e/test_historical_features.py @@ -124,7 +124,8 @@ def test_historical_features( feature_refs = ["transactions:daily_transactions"] - job_submission_time = datetime.utcnow() + # remove microseconds because job.get_start_time() does not contain microseconds + job_submission_time = datetime.utcnow().replace(microsecond=0) job = feast_client.get_historical_features(feature_refs, customers_df) assert job.get_start_time() >= job_submission_time assert job.get_start_time() <= job_submission_time + timedelta(hours=1)