setup.py

# -*- coding: utf-8 -*-
import re
import sys
from setuptools import setup
from setuptools.command.test import test as TestCommand

REQUIRES = [
    'docopt', 'invoke', 'tqdm', 'sentencepiece', 'google-api-python-client', 'oauth2client',
    'tensorflow==1.15'
]


class PyTest(TestCommand):
    def finalize_options(self):
        """finalize options"""
        TestCommand.finalize_options(self)
        self.test_args = []
        self.test_suite = True

    def run_tests(self):
        """run tests"""
        import pytest
        errcode = pytest.main(self.test_args)
        sys.exit(errcode)


def find_version(fname):
    """Attempts to find the version number in the file names fname.
    Raises RuntimeError if not found.
    """
    version = ''
    with open(fname) as fp:
        reg = re.compile(r'__version__ = [\'"]([^\'"]*)[\'"]')
        for line in fp:
            m = reg.match(line)
            if m:
                version = m.group(1)
                break
    if not version:
        raise RuntimeError('Cannot find version information')
    return version


__version__ = find_version("biomedbert.py")


def read(fname):
    """read file content"""
    with open(fname) as fp:
        content = fp.read()
    return content


setup(
    name='biomedbert',
    version="0.1.0",
    description='A Natural Language Processing and Understanding (NLP/NLU) tool for biomedical research',
    long_description=read("README.rst"),
    author='AI vs COVID-19 Team',
    author_email='dvdbisong@gmail.com',
    url='https://github.com/ghego/covid-19_research_collaboration',
    install_requires=REQUIRES,
    license=read("LICENSE"),
    zip_safe=False,
    keywords='biomedbert',
    classifiers=[
        'Development Status :: 2 - Pre-Alpha',
        'Intended Audience :: Developers',
        'License :: OSI Approved :: MIT License',
        'Natural Language :: English',
        "Programming Language :: Python :: 2",
        'Programming Language :: Python :: 2.7',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.4',
        'Programming Language :: Python :: 3.5',
        'Programming Language :: Python :: 3.6',
        'Programming Language :: Python :: Implementation :: CPython',
        'Programming Language :: Python :: Implementation :: PyPy'
    ],
    py_modules=["biomedbert"],
    entry_points={
        'console_scripts': [
            "biomedbert = biomedbert:main"
        ]
    },
    tests_require=['pytest'],
    cmdclass={'test': PyTest}
)