setup.py

#!/usr/bin/env python3
import os
import site
import sys
from distutils.core import setup

import setuptools

# Editable install in user site directory can be allowed with this hack:
# https://github.com/pypa/pip/issues/7953.
site.ENABLE_USER_SITE = "--user" in sys.argv[1:]

with open("README.md") as f:
    long_description = f.read()

with open(os.path.join("robust_speech", "version.txt")) as f:
    version = f.read().strip()

setup(
    name="robust_speech",
    version=version,
    description="Adversarially Robust Speech Recognition toolkit built over SpeechBrain",
    long_description=long_description,
    long_description_content_type="text/markdown",
    author="Raphael Olivier",
    author_email="rolivier@cs.cmu.edu",
    packages=setuptools.find_packages(),
    package_data={"robust_speech": ["version.txt", "log-config.yaml"]},
    install_requires=[
        "audlib",
        "adversarial-robustness-toolbox==1.10.0",
        "tensorflow==2.8.0",
        "torch>=1.7,<=1.11",
        "speechbrain @ git+https://github.com/RaphaelOlivier/speechbrain.git",
        "transformers>=4.15.0",
        "torchvision",
    ],
    python_requires=">=3.7",
    url="https://github.com/RaphaelOlivier/robust_speech",
)