提交 b0873dd5 编写于 作者: T tangwei

add mpi cluster

上级 e1f9013a
......@@ -39,10 +39,14 @@ def build(dirname):
packages = find_packages(dirname, include=('fleetrec.*'))
package_dir = {'': dirname}
package_data = {}
need_copy = ['data/*.txt', 'data/*/*.txt', '*.yaml', 'tree/*.npy','tree/*.txt']
models_copy = ['data/*.txt', 'data/*/*.txt', '*.yaml', '*.sh', 'tree/*.npy', 'tree/*.txt']
engine_copy = ['*/*.sh']
for package in packages:
if package.startswith("fleetrec.models."):
package_data[package] = need_copy
package_data[package] = models_copy
if package.startswith("fleetrec.core.engine"):
package_data[package] = engine_copy
setup(
name=about["__title__"],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册