fix historical test flakiness

Signed-off-by: Jacob Klegar <jacob@tecton.ai>
feast-dev · Jan 15, 2021 · 3a7e871 · 3a7e871
1 parent 8cd480f
commit 3a7e871
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)