Skip to content

Commit

Permalink
chore: Address flakey online retrieval test (feast-dev#4698)
Browse files Browse the repository at this point in the history
  • Loading branch information
robhowley authored and lokeshrangineni committed Oct 29, 2024
1 parent 7cb3a4c commit bd6a291
Showing 1 changed file with 17 additions and 5 deletions.
22 changes: 17 additions & 5 deletions sdk/python/tests/integration/online_store/test_universal_online.py
Original file line number Diff line number Diff line change
Expand Up @@ -485,11 +485,7 @@ def test_online_retrieval_with_event_timestamps(environment, universal_data_sour
assert_feature_store_universal_feature_views_response(df)


@pytest.mark.integration
@pytest.mark.universal_online_stores(only=["redis", "dynamodb", "postgres"])
async def test_async_online_retrieval_with_event_timestamps(
environment, universal_data_sources
):
async def _do_async_retrieval_test(environment, universal_data_sources):
fs = setup_feature_store_universal_feature_views(
environment, universal_data_sources
)
Expand All @@ -510,6 +506,22 @@ async def test_async_online_retrieval_with_event_timestamps(
await fs.close()


@pytest.mark.integration
@pytest.mark.universal_online_stores(only=["redis", "postgres"])
async def test_async_online_retrieval_with_event_timestamps(
environment, universal_data_sources
):
await _do_async_retrieval_test(environment, universal_data_sources)


@pytest.mark.integration
@pytest.mark.universal_online_stores(only=["dynamodb"])
async def test_async_online_retrieval_with_event_timestamps_dynamo(
environment, universal_data_sources
):
await _do_async_retrieval_test(environment, universal_data_sources)


@pytest.mark.integration
@pytest.mark.universal_online_stores
def test_online_list_retrieval(environment, universal_data_sources):
Expand Down

0 comments on commit bd6a291

Please sign in to comment.