diff --git a/python/nx-cugraph/_nx_cugraph/__init__.py b/python/nx-cugraph/_nx_cugraph/__init__.py index f6e5e3aa570..098de46af8e 100644 --- a/python/nx-cugraph/_nx_cugraph/__init__.py +++ b/python/nx-cugraph/_nx_cugraph/__init__.py @@ -23,18 +23,19 @@ $ python _nx_cugraph/__init__.py """ -from packaging.version import Version - from _nx_cugraph._version import __version__ -_nx_cugraph_version = Version(__version__) +# This is normally handled by packaging.version.Version, but instead of adding +# an additional runtime dependency on "packaging", assume __version__ will +# always be in .. format. +(_version_major, _version_minor) = __version__.split(".")[:2] # Entries between BEGIN and END are automatically generated _info = { "backend_name": "cugraph", "project": "nx-cugraph", "package": "nx_cugraph", - "url": f"https://github.com/rapidsai/cugraph/tree/branch-{_nx_cugraph_version.major:02}.{_nx_cugraph_version.minor:02}/python/nx-cugraph", + "url": f"https://github.com/rapidsai/cugraph/tree/branch-{_version_major:0>2}.{_version_minor:0>2}/python/nx-cugraph", "short_summary": "GPU-accelerated backend.", # "description": "TODO", "functions": {