(test_case)
| 507 | """Decorator marking a test that requires torch with a specific version or greater.""" |
| 508 | |
| 509 | def decorator(test_case): |
| 510 | correct_torch_version = is_torch_available() and is_torch_version(">=", torch_version) |
| 511 | return pytest.mark.skipif( |
| 512 | not correct_torch_version, |
| 513 | reason=f"test requires torch with the version greater than or equal to {torch_version}", |
| 514 | )(test_case) |
| 515 | |
| 516 | return decorator |
| 517 |
nothing calls this directly
no test coverage detected
searching dependent graphs…