From c3c415fb73e536aa649edab9dc69d0df6902ddb0 Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Mon, 11 Jul 2022 09:27:45 +0800
Subject: [PATCH] List torch as a installation requirements.

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index d69fb87..b822a11 100644
--- a/setup.py
+++ b/setup.py
@@ -41,6 +41,7 @@ setuptools.setup(
     long_description_content_type="text/markdown",
     ext_modules=[cmake_extension("_kaldifeat")],
     cmdclass={"build_ext": BuildExtension, "bdist_wheel": bdist_wheel},
+    install_requires=["torch"],
     zip_safe=False,
     classifiers=[
         "Programming Language :: C++",