Skip to content

Commit

Permalink
Fix
Browse files Browse the repository at this point in the history
  • Loading branch information
HCookie committed Oct 10, 2024
1 parent 2702881 commit 2f66219
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 29 deletions.
4 changes: 3 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -42,8 +42,10 @@ classifiers = [
dynamic = [ "version" ]
dependencies = [
"aniso8601",
"importlib-metadata; python_version<'3.10'",

"pyyaml",
"tomli", # Only needed before 3.11
"tomli", # Only needed before 3.11
"tqdm",
]

Expand Down
31 changes: 3 additions & 28 deletions src/anemoi/utils/provenance.py
Original file line number Diff line number Diff line change
Expand Up @@ -148,38 +148,13 @@ def _module_versions(full):
def package_distributions() -> dict[str, list[str]]:
# Takes a significant amount of time to run
# so cache the result
import collections
from importlib import metadata

# For python 3.9 support
# Copy of importlib.metadata.packages_distributions()

def packages_distributions():
"""Return a mapping of top-level packages to their
distributions.
>>> import collections.abc
>>> pkgs = packages_distributions()
>>> all(isinstance(dist, collections.abc.Sequence) for dist in pkgs.values())
True
"""
pkg_to_dist = collections.defaultdict(list)
for dist in metadata.distributions():
for pkg in _top_level_declared(dist) or _top_level_inferred(dist):
pkg_to_dist[pkg].append(dist.metadata["Name"])
return dict(pkg_to_dist)

def _top_level_declared(dist):
return (dist.read_text("top_level.txt") or "").split()

def _top_level_inferred(dist):
return {
f.parts[0] if len(f.parts) > 1 else f.with_suffix("").name
for f in (dist.files if isinstance(dist.files, (tuple, list)) else [dist.files])
if f.suffix == ".py"
}
if not hasattr(metadata, "packages_distributions"):
import importlib_metadata as metadata

return packages_distributions()
return metadata.packages_distributions()


def import_name_to_distribution_name(packages: list):
Expand Down

0 comments on commit 2f66219

Please sign in to comment.