Add pyproject.toml with build-requires torch

anth2o · anth2o · commit ddf16e66ff59 · 2022-11-08T17:47:52.000+01:00
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"],
 )

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+[build-system]`
	`2`	`+requires = ["setuptools", "wheel", "torch"]`
Original file line number	Diff line number	Diff line change
`@@ -140,4 +140,5 @@ def _single_compile(obj):`
`140`	`140`	`packages=find_packages(exclude=["build"]),`
`141`	`141`	`ext_modules=[extension],`
`142`	`142`	`cmdclass={"build_ext": BuildExtension},`
	`143`	`+ install_requires=["torch"],`
`143`	`144`	`)`