setup.py

#!/usr/bin/env python
import glob
import os
from collections.abc import Iterator
from importlib.util import module_from_spec, spec_from_file_location

from setuptools import find_packages, setup

_PATH_ROOT = os.path.realpath(os.path.dirname(__file__))
_PATH_SOURCE = os.path.join(_PATH_ROOT, "src")
_PATH_REQUIRE = os.path.join(_PATH_ROOT, "requirements")


def _load_py_module(fname: str, pkg: str = "lightning_utilities"):
    spec = spec_from_file_location(os.path.join(pkg, fname), os.path.join(_PATH_SOURCE, pkg, fname))
    py = module_from_spec(spec)
    spec.loader.exec_module(py)
    return py


about = _load_py_module("__about__.py")
requirements_module = _load_py_module(os.path.join("install", "requirements.py"))


# load basic requirements using the central parser from lightning_utilities.install.requirements
def _parse_requirements(lines: list[str]) -> Iterator[str]:
    """Parse requirements from lines using the canonical parser."""
    return (str(req) for req in requirements_module._parse_requirements(lines))


with open(os.path.join(_PATH_REQUIRE, "core.txt")) as fp:
    requirements = list(_parse_requirements(fp.readlines()))


# make extras as automated loading
def _requirement_extras(path_req: str = _PATH_REQUIRE) -> dict:
    extras = {}
    for fpath in glob.glob(os.path.join(path_req, "*.txt")):
        fname = os.path.basename(fpath)
        if fname.startswith(("_", "gha-")):
            continue
        if fname in ("core.txt",):
            continue
        name, _ = os.path.splitext(fname)
        with open(fpath) as fp:
            extras[name] = list(_parse_requirements(fp.readlines()))
    return extras


# loading readme as description
with open(os.path.join(_PATH_ROOT, "README.md")) as fp:
    readme = fp.read()

setup(
    name="lightning-utilities",
    version=about.__version__,
    description=about.__docs__,
    author=about.__author__,
    author_email=about.__author_email__,
    url=about.__homepage__,
    download_url="https://github.com/Lightning-AI/utilities",
    license=about.__license__,  # Should be a license identifier like "MIT"
    license_files=["LICENSE"],  # Path to your license file
    packages=find_packages(where="src"),
    package_dir={"": "src"},
    long_description=readme,
    long_description_content_type="text/markdown",
    include_package_data=True,
    zip_safe=False,
    keywords=["Utilities", "DevOps", "CI/CD"],
    python_requires=">=3.10",
    setup_requires=[],
    install_requires=requirements,
    extras_require=_requirement_extras(),
    project_urls={
        "Bug Tracker": "https://github.com/Lightning-AI/utilities/issues",
        "Documentation": "https://dev-toolbox.rtfd.io/en/latest/",  # TODO: Update domain
        "Source Code": "https://github.com/Lightning-AI/utilities",
    },
    classifiers=[
        "Environment :: Console",
        "Natural Language :: English",
        # How mature is this project? Common values are
        #   3 - Alpha, 4 - Beta, 5 - Production/Stable
        "Development Status :: 3 - Alpha",
        # Indicate who your project is intended for
        "Intended Audience :: Developers",
        # Pick your license as you wish
        # 'License :: OSI Approved :: BSD License',
        "Operating System :: OS Independent",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
        "Programming Language :: Python :: 3.12",
        "Programming Language :: Python :: 3.13",
        "Programming Language :: Python :: 3.14",
    ],
)