| from setuptools import setup, find_packages |
|
|
| test_packages = [ |
| "pytest>=5.4.3", |
| "pytest-cov>=2.6.1" |
| ] |
|
|
| docs_packages = [ |
| "mkdocs==1.1", |
| "mkdocs-material==4.6.3", |
| "mkdocstrings==0.8.0", |
| ] |
|
|
| base_packages = [ |
| "numpy>=1.20.0", |
| "hdbscan>=0.8.29", |
| "umap-learn>=0.5.0", |
| "pandas>=1.1.5", |
| "scikit-learn>=0.22.2.post1", |
| "tqdm>=4.41.1", |
| "sentence-transformers>=0.4.1", |
| "plotly>=4.7.0", |
| ] |
|
|
| flair_packages = [ |
| "transformers>=3.5.1", |
| "torch>=1.4.0", |
| "flair>=0.7" |
| ] |
|
|
| spacy_packages = [ |
| "spacy>=3.0.1" |
| ] |
|
|
| use_packages = [ |
| "tensorflow", |
| "tensorflow_hub", |
| "tensorflow_text" |
| ] |
|
|
| gensim_packages = [ |
| "gensim>=4.0.0" |
| ] |
|
|
| vision_packages = [ |
| "Pillow>=9.2.0", |
| "accelerate>=0.19.0" |
| ] |
|
|
| datamap_packages = [ |
| "matplotlib>=3.8", |
| "datamapplot>=0.1" |
| ] |
|
|
| extra_packages = flair_packages + spacy_packages + use_packages + gensim_packages |
|
|
| dev_packages = docs_packages + test_packages |
|
|
| with open("README.md", "r") as fh: |
| long_description = fh.read() |
|
|
| setup( |
| name="bertopic", |
| packages=find_packages(exclude=["notebooks", "docs"]), |
| version="0.16.0", |
| author="Maarten P. Grootendorst", |
| author_email="maartengrootendorst@gmail.com", |
| description="BERTopic performs topic Modeling with state-of-the-art transformer models.", |
| long_description=long_description, |
| long_description_content_type="text/markdown", |
| url="https://github.com/MaartenGr/BERTopic", |
| project_urls={ |
| "Documentation": "https://maartengr.github.io/BERTopic/", |
| "Source Code": "https://github.com/MaartenGr/BERTopic/", |
| "Issue Tracker": "https://github.com/MaartenGr/BERTopic/issues", |
| }, |
| keywords="nlp bert topic modeling embeddings", |
| classifiers=[ |
| "Programming Language :: Python", |
| "Intended Audience :: Science/Research", |
| "Intended Audience :: Developers", |
| "Topic :: Scientific/Engineering :: Artificial Intelligence", |
| "License :: OSI Approved :: MIT License", |
| "Topic :: Scientific/Engineering", |
| "Operating System :: Microsoft :: Windows", |
| "Operating System :: POSIX", |
| "Operating System :: Unix", |
| "Operating System :: MacOS", |
| "Programming Language :: Python :: 3.7", |
| "Programming Language :: Python :: 3.9", |
| "Programming Language :: Python :: 3.8", |
| ], |
| install_requires=base_packages, |
| extras_require={ |
| "test": test_packages, |
| "docs": docs_packages, |
| "dev": dev_packages, |
| "flair": flair_packages, |
| "spacy": spacy_packages, |
| "use": use_packages, |
| "gensim": gensim_packages, |
| "vision": vision_packages, |
| "datamap": datamap_packages, |
| }, |
| python_requires='>=3.7', |
| ) |
|
|
