提交 be92338e 编写于 作者: T tangwei

fix forder format

上级 c2ebf30b
""" """
setup for fleet-rec. setup for fleet-rec.
""" """
import os
import sys import sys
from setuptools import setup, find_packages from setuptools import setup, find_packages
import tempfile
import shutil
if sys.version_info.major == 2: if sys.version_info.major == 2:
requires = [ requires = [
"paddlepaddle == 1.7.2", "paddlepaddle == 1.7.2",
"netron >= 5.1.1", "netron >= 0.0.0",
"yaml" "yaml >= 5.1.1"
] ]
else: else:
requires = [ requires = [
"paddlepaddle >= 0.0.0", "paddlepaddle >= 0.0.0",
"netron >= 0.0.0", "netron >= 0.0.0",
"pyyaml" "pyyaml >= 5.1.1"
] ]
about = {} about = {}
...@@ -27,19 +30,44 @@ about["__url__"] = "https://github.com/seiriosPlus/FleetRec" ...@@ -27,19 +30,44 @@ about["__url__"] = "https://github.com/seiriosPlus/FleetRec"
readme = "..." readme = "..."
setup(
name=about["__title__"], def run_cmd(command):
version=about["__version__"], assert command is not None and isinstance(command, str)
description=about["__description__"], return os.popen(command).read().strip()
long_description=readme,
author=about["__author__"],
author_email=about["__author_email__"], def build(dirname):
url=about["__url__"], print(dirname)
packages=find_packages(), package_dir = os.path.dirname(os.path.abspath(__file__))
python_requires=">=2.7", run_cmd("cp -r {}/* {}".format(package_dir, dirname))
install_requires=requires, run_cmd("mv {} {}".format(os.path.join(dirname, "dataset"), os.path.join(dirname, "fleetrec")))
zip_safe=False run_cmd("mv {} {}".format(os.path.join(dirname, "demo"), os.path.join(dirname, "fleetrec")))
) run_cmd("mv {} {}".format(os.path.join(dirname, "doc"), os.path.join(dirname, "fleetrec")))
run_cmd("mv {} {}".format(os.path.join(dirname, "models"), os.path.join(dirname, "fleetrec")))
run_cmd("mv {} {}".format(os.path.join(dirname, "tools"), os.path.join(dirname, "fleetrec")))
packages = find_packages(dirname)
package_dir = {'': dirname}
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,
package_dir=package_dir,
python_requires=">=2.7",
install_requires=requires,
zip_safe=False
)
dirname = tempfile.mkdtemp()
build(dirname)
shutil.rmtree(dirname)
print(''' print('''
\033[32m \033[32m
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册