Skip to content

Commit

Permalink
Move import to function
Browse files Browse the repository at this point in the history
  • Loading branch information
KyleFromNVIDIA committed Jun 14, 2024
1 parent 43a9441 commit d1dc75e
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/rapids_metadata/metadata.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@
from dataclasses import dataclass, field
from os import PathLike

from packaging.version import Version
from .rapids_version import get_rapids_version

__all__ = [
Expand Down Expand Up @@ -73,6 +72,8 @@ class RAPIDSMetadata:
versions: dict[str, RAPIDSVersion] = field(default_factory=dict)

def get_current_version(self, directory: PathLike) -> RAPIDSVersion:
from packaging.version import Version

current_version = get_rapids_version(directory)
try:
return self.versions[current_version]
Expand Down

0 comments on commit d1dc75e

Please sign in to comment.