From 6e210034c57755673087d123a0b8b587d001b314 Mon Sep 17 00:00:00 2001
From: zigzagcai <caizheng@pjlab.org.cn>
Date: Fri, 19 Jul 2024 13:21:25 +0800
Subject: [PATCH] add torch as torch_scatter requirements

---
 setup.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index fba55fc8..97a8612c 100644
--- a/setup.py
+++ b/setup.py
@@ -107,7 +107,8 @@ def get_extensions():
     return extensions
 
 
-install_requires = []
+install_requires = ["torch>=1.8.0"]
+extra_index_url = ["https://download.pytorch.org/whl/"]
 
 test_requires = [
     'pytest',
@@ -130,6 +131,7 @@ def get_extensions():
     keywords=['pytorch', 'scatter', 'segment', 'gather'],
     python_requires='>=3.8',
     install_requires=install_requires,
+    extra_index_url=extra_index_url,
     extras_require={
         'test': test_requires,
     },