File size: 2,207 Bytes
b34c6c3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 | from typing import List
from setuptools import find_packages, setup
from setuptools.command.develop import develop
from setuptools.command.egg_info import egg_info
from setuptools.command.install import install
def fetch_requirements(path) -> List[str]:
"""
This function reads the requirements file.
Args:
path (str): the path to the requirements file.
Returns:
The lines in the requirements file.
"""
with open(path, "r") as fd:
requirements = [r.strip() for r in fd.readlines()]
# requirements.remove("colossalai")
return requirements
def fetch_readme() -> str:
"""
This function reads the README.md file in the current directory.
Returns:
The lines in the README file.
"""
with open("README.md", encoding="utf-8") as f:
return f.read()
def custom_install():
return ["pip", "install", "colossalai", "--no-deps"]
class CustomInstallCommand(install):
def run(self):
install.run(self)
self.spawn(custom_install())
class CustomDevelopCommand(develop):
def run(self):
develop.run(self)
self.spawn(custom_install())
class CustomEggInfoCommand(egg_info):
def run(self):
egg_info.run(self)
self.spawn(custom_install())
setup(
name="videosys",
version="2.0.0",
packages=find_packages(
exclude=(
"videos",
"tests",
"figure",
"*.egg-info",
)
),
description="TeaCache",
long_description=fetch_readme(),
long_description_content_type="text/markdown",
license="Apache Software License 2.0",
install_requires=fetch_requirements("requirements.txt"),
python_requires=">=3.7",
# cmdclass={
# "install": CustomInstallCommand,
# "develop": CustomDevelopCommand,
# "egg_info": CustomEggInfoCommand,
# },
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License",
"Environment :: GPU :: NVIDIA CUDA",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: System :: Distributed Computing",
],
)
|