setup.py 1.2 KB
Newer Older
T
for whl  
tangwei 已提交
1 2 3 4 5 6
"""
setup for fleet-rec.
"""

from setuptools import setup

T
tangwei 已提交
7 8
packages = ["fleetrec", "fleetrec.models", "fleetrec.examples", "fleetrec.core", "fleetrec.core.engine",
            "fleetrec.core.metrics", "fleetrec.core.models", "fleetrec.core.trainers", "fleetrec.core.utils"]
T
for whl  
tangwei 已提交
9 10

requires = [
T
tangwei12 已提交
11
    "paddlepaddle"
T
for whl  
tangwei 已提交
12 13 14 15 16 17
]

about = {}
about["__title__"] = "fleet-rec"
about["__version__"] = "0.0.2"
about["__description__"] = "fleet-rec"
T
tangwei12 已提交
18 19 20 21
about["__author__"] = "seiriosPlus"
about["__author_email__"] = "tangwei12@baidu.com"
about["__url__"] = "https://github.com/seiriosPlus/FleetRec"

T
for whl  
tangwei 已提交
22 23 24 25 26 27 28 29 30 31 32
readme = "..."

setup(
    name=about["__title__"],
    version=about["__version__"],
    description=about["__description__"],
    long_description=readme,
    author=about["__author__"],
    author_email=about["__author_email__"],
    url=about["__url__"],
    packages=packages,
T
tangwei12 已提交
33
    python_requires=">=2.7",
T
for whl  
tangwei 已提交
34 35 36 37 38 39
    install_requires=requires,
    zip_safe=False
)

print('''
\033[32m
T
tangwei12 已提交
40 41 42 43
  _   _   _   _   _   _   _   _   _  
 / \ / \ / \ / \ / \ / \ / \ / \ / \ 
( F | L | E | E | T | - | R | E | C )
 \_/ \_/ \_/ \_/ \_/ \_/ \_/ \_/ \_/ 
T
for whl  
tangwei 已提交
44 45 46 47 48 49
\033[0m
\033[34m
Installation Complete. Congratulations!
How to use it ? Please visit our webside: https://github.com/seiriosPlus/FleetRec
\033[0m
''')