Skip to content

Commit 8bca5ef

Browse files
authored
Fix minimum tests by installing torch < 2.9.0 (#803)
1 parent 7858d8f commit 8bca5ef

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tasks.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,9 @@
1414

1515
COMPARISONS = {'>=': operator.ge, '>': operator.gt, '<': operator.lt, '<=': operator.le}
1616

17+
EXTERNAL_DEPENDENCY_CAPS = {
18+
'torch': '2.9.0'
19+
}
1720

1821
if not hasattr(inspect, 'getargspec'):
1922
inspect.getargspec = inspect.getfullargspec
@@ -87,7 +90,8 @@ def install_minimum(c):
8790
if minimum_versions:
8891
install_deps = ' '.join(minimum_versions)
8992
c.run(f'python -m pip install {install_deps}')
90-
93+
for dep, cap in EXTERNAL_DEPENDENCY_CAPS.items():
94+
c.run(f'python -m pip install "{dep}<{cap}"')
9195

9296
@task
9397
def minimum(c):

0 commit comments

Comments
 (0)