Skip to content

Commit c7b6956

Browse files
committed
Fix cuda_python version specifier
1 parent 68ace96 commit c7b6956

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

cuda_python/setup.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,16 +17,24 @@
1717
)
1818

1919

20-
version = Version(version).base_version
20+
base_version = Version(version).base_version
21+
22+
23+
if base_version == version:
24+
# Tagged release
25+
matcher = "~="
26+
else:
27+
# Pre-release version
28+
matcher = "=="
2129

2230

2331
setup(
2432
version=version,
2533
install_requires=[
26-
f"cuda-bindings~={version}",
34+
f"cuda-bindings{matcher}{version}",
2735
"cuda-pathfinder~=1.1",
2836
],
2937
extras_require={
30-
"all": [f"cuda-bindings[all]~={version}"],
38+
"all": [f"cuda-bindings[all]{matcher}{version}"],
3139
},
3240
)

0 commit comments

Comments
 (0)