chore: cleanup

phil65 · Nov 14, 2024 · 7a24c68 · 7a24c68
1 parent 9704210
commit 7a24c68
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mknodes/info/packageregistry.py b/mknodes/info/packageregistry.py
@@ -3,8 +3,10 @@
 from abc import ABCMeta
 from collections.abc import MutableMapping
 
+import epregistry
+
 from mknodes.info import packageinfo
-from mknodes.utils import log, packagehelpers
+from mknodes.utils import log
 
 
 logger = log.get_logger(__name__)
@@ -66,7 +68,7 @@ def get_info(self, mod_name: str) -> packageinfo.PackageInfo:
         Args:
             mod_name: Name of the module
         """
-        mapping = packagehelpers.get_package_map()
+        mapping = epregistry.get_packages_distributions()
         pkg_name = mapping[mod_name][0] if mod_name in mapping else mod_name
         pkg_name = pkg_name.lower()
         if pkg_name not in self._packages: