Add pyproject.toml with build-requires torch

parlance · Nov 8, 2022 · ddf16e6 · ddf16e6
1 parent 930672d
commit ddf16e6
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -12,9 +12,7 @@ KenLM language modeling support is also optionally included, and enabled by defa
 The below installation also works for Google Colab.
 
 ```bash
-# get the code
-git clone --recursive https://github.com/parlance/ctcdecode.git
-cd ctcdecode && pip install .
+pip install git+https://github.com/parlance/ctcdecode.git
 ```
 
 ## How to Use

diff --git a/pyproject.toml b/pyproject.toml
@@ -0,0 +1,2 @@
+[build-system]
+requires = ["setuptools", "wheel", "torch"]
diff --git a/requirements.txt b/requirements.txt
diff --git a/setup.py b/setup.py
@@ -140,4 +140,5 @@ def _single_compile(obj):
     packages=find_packages(exclude=["build"]),
     ext_modules=[extension],
     cmdclass={"build_ext": BuildExtension},
+    install_requires=["torch"],
 )