setup.py 754 Bytes
Newer Older
Soumith Chintala's avatar
Soumith Chintala committed
1
2
#!/usr/bin/env python

3
4
5
import os
import sys

Soumith Chintala's avatar
Soumith Chintala committed
6
from setuptools import setup, find_packages
7
8
9
10

import build

this_file = os.path.dirname(__file__)
Soumith Chintala's avatar
Soumith Chintala committed
11
12
13
14

setup(
    name="torchaudio",
    version="0.1",
Soumith Chintala's avatar
Soumith Chintala committed
15
16
    description="An audio package for PyTorch",
    url="https://github.com/pytorch/audio",
17
    author="Soumith Chintala, David Pollack, Sean Naren",
Soumith Chintala's avatar
Soumith Chintala committed
18
    author_email="soumith@pytorch.org",
19
20
21
    # Require cffi.
    install_requires=["cffi>=1.0.0"],
    setup_requires=["cffi>=1.0.0"],
Soumith Chintala's avatar
Soumith Chintala committed
22
23
    # Exclude the build files.
    packages=find_packages(exclude=["build"]),
24
25
26
27
28
    # Package where to put the extensions. Has to be a prefix of build.py.
    ext_package="",
    # Extensions to compile.
    cffi_modules=[
        os.path.join(this_file, "build.py:ffi")
Soumith Chintala's avatar
Soumith Chintala committed
29
    ],
30
)