Skip to content

Commit 72f2bd0

Browse files
committed
[CHORE] Fix flaky default_ef test
1 parent a7c785a commit 72f2bd0

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

chromadb/test/ef/test_default_ef.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,7 @@ def unique_by(x: Hashable) -> Hashable:
2121
@settings(deadline=None)
2222
@given(
2323
providers=st.lists(
24-
st.sampled_from(onnxruntime.get_all_providers()).filter(
25-
lambda x: x not in onnxruntime.get_available_providers()
26-
),
24+
st.sampled_from(onnxruntime.get_available_providers()),
2725
unique_by=unique_by,
2826
min_size=1,
2927
)
@@ -37,9 +35,7 @@ def test_unavailable_provider_multiple(providers: List[str]) -> None:
3735

3836
@given(
3937
providers=st.lists(
40-
st.sampled_from(onnxruntime.get_all_providers()).filter(
41-
lambda x: x in onnxruntime.get_available_providers()
42-
),
38+
st.sampled_from(onnxruntime.get_available_providers()),
4339
min_size=1,
4440
unique_by=unique_by,
4541
)
@@ -56,9 +52,7 @@ def test_warning_no_providers_supplied() -> None:
5652

5753
@given(
5854
providers=st.lists(
59-
st.sampled_from(onnxruntime.get_all_providers()).filter(
60-
lambda x: x in onnxruntime.get_available_providers()
61-
),
55+
st.sampled_from(onnxruntime.get_available_providers()),
6256
min_size=1,
6357
).filter(lambda x: len(x) > len(set(x)))
6458
)

0 commit comments

Comments
 (0)