From e022e8f5c7d660ede8d9d13342c3e2b44d11ce31 Mon Sep 17 00:00:00 2001 From: Antoine Hoorelbeke Date: Tue, 8 Nov 2022 17:30:51 +0100 Subject: [PATCH] Add pyproject.toml with build-requires torch --- README.md | 4 +--- pyproject.toml | 2 ++ requirements.txt | 1 - setup.py | 1 + 4 files changed, 4 insertions(+), 4 deletions(-) create mode 100644 pyproject.toml delete mode 100644 requirements.txt diff --git a/README.md b/README.md index e7a91929..1bc62412 100644 --- 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 new file mode 100644 index 00000000..db93852a --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,2 @@ +[build-system] +requires = ["setuptools", "wheel", "torch"] diff --git a/requirements.txt b/requirements.txt deleted file mode 100644 index 08ed5eeb..00000000 --- a/requirements.txt +++ /dev/null @@ -1 +0,0 @@ -torch \ No newline at end of file diff --git a/setup.py b/setup.py index a89f24ab..e759b686 100644 --- 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"], )