|
- # Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- import os
- import setuptools
- import sys
- import paddlenlp
- long_description = "PaddlePaddle NLP Model Core Library"
-
- with open("requirements.txt") as fin:
- REQUIRED_PACKAGES = fin.read()
-
-
- def get_package_data_files(package, data, package_dir=None):
- """
- Helps to list all specified files in package including files in directories
- since `package_data` ignores directories.
- """
- if package_dir is None:
- package_dir = os.path.join(*package.split('.'))
- all_files = []
- for f in data:
- path = os.path.join(package_dir, f)
- if os.path.isfile(path):
- all_files.append(f)
- continue
- for root, _dirs, files in os.walk(path, followlinks=True):
- root = os.path.relpath(root, package_dir)
- for file in files:
- file = os.path.join(root, file)
- if file not in all_files:
- all_files.append(file)
- return all_files
-
-
- setuptools.setup(
- name="paddlenlp",
- version=paddlenlp.__version__,
- author="PaddlePaddle Speech and Language Team",
- author_email="paddlesl@baidu.com",
- description=long_description,
- long_description=long_description,
- long_description_content_type="text/plain",
- url="https://github.com/PaddlePaddle/PaddleNLP",
- packages=setuptools.find_packages(
- where='.', exclude=('examples*', 'tests*')),
- package_data={
- 'paddlenlp.ops': get_package_data_files('paddlenlp.ops', [
- 'CMakeLists.txt', 'README.md', 'cmake', 'faster_transformer',
- 'patches', 'optimizer'
- ])
- },
- setup_requires=['cython', 'numpy'],
- install_requires=REQUIRED_PACKAGES,
- python_requires='>=3.6',
- classifiers=[
- 'Programming Language :: Python :: 3',
- 'Programming Language :: Python :: 3.6',
- 'Programming Language :: Python :: 3.7',
- 'Programming Language :: Python :: 3.8',
- 'License :: OSI Approved :: Apache Software License',
- 'Operating System :: OS Independent',
- ],
- license='Apache 2.0')
|