tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

removing 2.3.1 (#2294)

Browse Source
This commit is contained in:
salman
2025-01-29 04:23:44 +00:00
committed by GitHub
parent c015a76a23
commit c071a530f7
12 changed files with 10 additions and 114 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/e2e/utils.py
View File

@@ -42,18 +42,6 @@ def most_recent_subdir(path):
return subdir
def require_torch_2_3_1(test_case):
"""
Decorator marking a test that requires torch >= 2.3.1
"""
def is_min_2_3_1():
torch_version = version.parse(torch.__version__)
return torch_version >= version.parse("2.3.1")
return unittest.skipUnless(is_min_2_3_1(), "test requires torch>=2.3.1")(test_case)
def require_torch_2_4_1(test_case):
"""
Decorator marking a test that requires torch >= 2.5.1