Skip to content

Commit

Permalink
add torch as torch_scatter requirements
Browse files Browse the repository at this point in the history
  • Loading branch information
season0528 committed Jul 19, 2024
1 parent 96aa2e3 commit 6e21003
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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',
Expand All @@ -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,
},
Expand Down

0 comments on commit 6e21003

Please sign in to comment.