Skip to content

Commit

Permalink
allow python 3.7
Browse files Browse the repository at this point in the history
  • Loading branch information
teddykoker committed Mar 24, 2021
1 parent 220ad12 commit f7cb735
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ def cuda_toolkit_available():
# https://github.com/idiap/fast-transformers/blob/master/setup.py
try:
call(["nvcc"], stdout=DEVNULL, stderr=DEVNULL)
return True and torch.cuda.is_available()
return True
except FileNotFoundError:
return False

Expand Down Expand Up @@ -51,7 +51,7 @@ def ext_modules():

setup(
name="torchsort",
version="0.1.1",
version="0.1.2",
description="Differentiable sorting and ranking in PyTorch",
author="Teddy Koker",
url="https://github.com/teddykoker/torchsort",
Expand All @@ -63,7 +63,7 @@ def ext_modules():
"Programming Language :: Python :: 3",
],
install_requires=["torch"],
python_requires=">=3.8",
python_requires=">=3.7",
extras_require={
"testing": [
"pytest",
Expand Down

0 comments on commit f7cb735

Please sign in to comment.