diff --git a/dpdata/plugins/__init__.py b/dpdata/plugins/__init__.py index 4887501ca..ca097fd75 100644 --- a/dpdata/plugins/__init__.py +++ b/dpdata/plugins/__init__.py @@ -14,6 +14,9 @@ importlib.import_module(module_name, PACKAGE_BASE) # https://setuptools.readthedocs.io/en/latest/userguide/entry_point.html -eps = metadata.entry_points().get('dpdata.plugins', []) +try: + eps = metadata.entry_points(group='dpdata.plugins') +except TypeError: + eps = metadata.entry_points().get('dpdata.plugins', []) for ep in eps: plugin = ep.load()