Add dependency in pyproject.toml, don't require it for hook

rapidsai · Jun 14, 2024 · 43a9441 · 43a9441
1 parent 88646a5
commit 43a9441
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,6 +26,9 @@ classifiers = [
     "Programming Language :: Python :: 3",
 ]
 requires-python = ">=3.9"
+dependencies = [
+    "packaging",
+]
 
 [project.scripts]
 rapids-metadata-json = "rapids_metadata.json:main"

diff --git a/src/rapids_metadata/rapids_version.py b/src/rapids_metadata/rapids_version.py
@@ -15,13 +15,13 @@
 import os.path
 from os import PathLike
 
-from packaging.version import InvalidVersion, Version
-
 
 __all__ = ["get_rapids_version"]
 
 
 def get_rapids_version(directory: PathLike) -> str:
+    from packaging.version import InvalidVersion, Version
+
     while not os.path.samefile(directory, os.path.dirname(directory)):
         try:
             with open(os.path.join(directory, "VERSION")) as f: