setup.py 1.3 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 9
packages = ["fleetrec", "fleetrec.models",
            "fleetrec.examples", "fleetrec.examples.user_define", "fleetrec.examples.build_in"
            "fleetrec.core", "fleetrec.core.engine",
T
tangwei 已提交
10
            "fleetrec.core.metrics", "fleetrec.core.models", "fleetrec.core.trainers", "fleetrec.core.utils"]
T
for whl  
tangwei 已提交
11 12

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

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

T
for whl  
tangwei 已提交
24 25 26 27 28 29 30 31 32 33 34
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 已提交
35
    python_requires=">=2.7",
T
for whl  
tangwei 已提交
36 37 38 39 40 41
    install_requires=requires,
    zip_safe=False
)

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