VideoSys/CogVideoX

1from typing import List2 3from setuptools import find_packages, setup4 5 6def fetch_requirements(path) -> List[str]:7    """8    This function reads the requirements file.9 10    Args:11        path (str): the path to the requirements file.12 13    Returns:14        The lines in the requirements file.15    """16    with open(path, "r") as fd:17        return [r.strip() for r in fd.readlines()]18 19 20def fetch_readme() -> str:21    """22    This function reads the README.md file in the current directory.23 24    Returns:25        The lines in the README file.26    """27    with open("README.md", encoding="utf-8") as f:28        return f.read()29 30 31setup(32    name="videosys",33    version="2.0.0",34    packages=find_packages(35        exclude=(36            "videos",37            "tests",38            "figure",39            "*.egg-info",40        )41    ),42    description="VideoSys",43    long_description=fetch_readme(),44    long_description_content_type="text/markdown",45    license="Apache Software License 2.0",46    install_requires=fetch_requirements("requirements.txt"),47    python_requires=">=3.6",48    classifiers=[49        "Programming Language :: Python :: 3",50        "License :: OSI Approved :: Apache Software License",51        "Environment :: GPU :: NVIDIA CUDA",52        "Topic :: Scientific/Engineering :: Artificial Intelligence",53        "Topic :: System :: Distributed Computing",54    ],55)56