Skip to content

Commit 93552a6

Browse files
committed
fix: make entrypoint tests compatible with old python versions
1 parent 64630ce commit 93552a6

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

tests/test_entrypoints.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,6 @@ def test_entrypoints():
2525

2626

2727
def test_readers_and_writers_entrypoints():
28-
import importlib.metadata
29-
3028
# Map of expected entry points for readers and writers
3129
expected_readers = {
3230
"spark": "from_spark",
@@ -47,9 +45,16 @@ def test_readers_and_writers_entrypoints():
4745
"polars": "to_polars",
4846
"parquet": "write_parquet",
4947
}
50-
eps = importlib.metadata.entry_points()
51-
readers = {ep.name for ep in eps.select(group="edsnlp_readers")}
52-
writers = {ep.name for ep in eps.select(group="edsnlp_writers")}
48+
eps = entry_points()
49+
if hasattr(eps, "select"):
50+
readers_eps = eps.select(group="edsnlp_readers")
51+
writers_eps = eps.select(group="edsnlp_writers")
52+
else:
53+
readers_eps = eps.get("edsnlp_readers", [])
54+
writers_eps = eps.get("edsnlp_writers", [])
55+
56+
readers = {ep.name for ep in readers_eps}
57+
writers = {ep.name for ep in writers_eps}
5358
for name in expected_readers:
5459
assert name in readers, f"Reader entry point '{name}' is missing"
5560
for name in expected_writers:

0 commit comments

Comments
 (0)