setup.py

import json

from pathlib import Path
from setuptools import setup, find_packages

with open(Path(__file__).parent.joinpath('small_text/version.json')) as f:
    version = json.load(f)

version_str = '.'.join(map(str, [version['major'], version['minor'], version['micro']]))
if version['pre_release'] != '':
    version_str += '.' + version['pre_release']


PYTORCH_DEPENDENCIES = ['torch>=1.6.0']


setup(name='small-text',
      version=version_str,
      license='MIT License',
      description='Active Learning for Text Classification in Python.',
      long_description=Path('README.md').read_text(encoding='utf-8'),
      long_description_content_type='text/markdown',
      author='Christopher Schröder',
      author_email='small-text@protonmail.com',
      url='https://github.com/webis-de/small-text',
      keywords=['active learning', 'text classification'],
      classifiers=[
            'Development Status :: 5 - Production/Stable',
            'License :: OSI Approved :: MIT License',
            'Operating System :: OS Independent',
            'Programming Language :: Python :: 3',
            'Programming Language :: Python :: 3.8',
            'Programming Language :: Python :: 3.9',
            'Programming Language :: Python :: 3.10',
            'Programming Language :: Python :: 3.11',
            'Topic :: Scientific/Engineering :: Artificial Intelligence',
            'Topic :: Text Processing :: Linguistic',
      ],
      packages=find_packages(),
      include_package_data=True,
      python_requires='>=3.8',
      install_requires=[
            'dill>=0.3.7',
            'scipy',
            'numpy>=1.21.0',
            'scikit-learn>=0.24.1',
            'tqdm',
            'packaging',
            'tokenizers>=0.11.5'
      ],
      extras_require={
            'pytorch': PYTORCH_DEPENDENCIES,
            'transformers': PYTORCH_DEPENDENCIES + ['transformers>=4.0.0']
      })