Use entry_points utility in sizeof (#9390)

jrbourbeau · web-flow · commit c1bc90aa5149 · 2022-08-17T12:56:27.000-05:00
diff --git a/dask/compatibility.py b/dask/compatibility.py
@@ -15,10 +15,10 @@ def entry_points(group=None):
     In 3.10 the return type changed from a dict to an ``importlib.metadata.EntryPoints``.
     This compatibility utility can be removed once Python 3.10 is the minimum.
     """
-    eps = importlib.metadata.entry_points()
-    if group:
-        try:
-            return eps.select(group=group)
-        except AttributeError:
+    if _PY_VERSION >= parse_version("3.10"):
+        return importlib.metadata.entry_points(group=group)
+    else:
+        eps = importlib.metadata.entry_points()
+        if group:
             return eps.get(group, [])
-    return eps
+        return eps
diff --git a/dask/sizeof.py b/dask/sizeof.py
@@ -1,10 +1,10 @@
-import importlib.metadata
 import itertools
 import logging
 import random
 import sys
 from array import array
 
+from dask.compatibility import entry_points
 from dask.utils import Dispatch
 
 sizeof = Dispatch(name="sizeof")
@@ -216,12 +216,7 @@ def sizeof_pyarrow_chunked_array(data):
 
 def _register_entry_point_plugins():
     """Register sizeof implementations exposed by the entry_point mechanism."""
-    if sys.version_info >= (3, 10):
-        sizeof_entry_points = importlib.metadata.entry_points(group="dask.sizeof")
-    else:
-        sizeof_entry_points = importlib.metadata.entry_points().get("dask.sizeof", [])
-
-    for entry_point in sizeof_entry_points:
+    for entry_point in entry_points(group="dask.sizeof"):
         registrar = entry_point.load()
         try:
             registrar(sizeof)
