We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7858d8f commit 8bca5efCopy full SHA for 8bca5ef
tasks.py
@@ -14,6 +14,9 @@
14
15
COMPARISONS = {'>=': operator.ge, '>': operator.gt, '<': operator.lt, '<=': operator.le}
16
17
+EXTERNAL_DEPENDENCY_CAPS = {
18
+ 'torch': '2.9.0'
19
+}
20
21
if not hasattr(inspect, 'getargspec'):
22
inspect.getargspec = inspect.getfullargspec
@@ -87,7 +90,8 @@ def install_minimum(c):
87
90
if minimum_versions:
88
91
install_deps = ' '.join(minimum_versions)
89
92
c.run(f'python -m pip install {install_deps}')
-
93
+ for dep, cap in EXTERNAL_DEPENDENCY_CAPS.items():
94
+ c.run(f'python -m pip install "{dep}<{cap}"')
95
96
@task
97
def minimum(c):
0 commit comments