File tree Expand file tree Collapse file tree 1 file changed +10
-5
lines changed Expand file tree Collapse file tree 1 file changed +10
-5
lines changed Original file line number Diff line number Diff line change @@ -25,8 +25,6 @@ def test_entrypoints():
2525
2626
2727def 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 :
You can’t perform that action at this time.
0 commit comments