Fix decorator

qubvel · qubvel · commit 81818885d268 · 2024-12-23T14:16:12.000Z
diff --git a/tests/utils.py b/tests/utils.py
@@ -42,6 +42,6 @@ def requires_torch_greater_or_equal(version: str):
     torch_version = Version(torch.__version__)
     provided_version = Version(version)
     return unittest.skipUnless(
-        torch_version < provided_version,
+        torch_version >= provided_version,
         f"torch version {torch_version} is less than {provided_version}",
     )

Original file line number	Diff line number	Diff line change
`@@ -42,6 +42,6 @@ def requires_torch_greater_or_equal(version: str):`
`42`	`42`	`torch_version = Version(torch.__version__)`
`43`	`43`	`provided_version = Version(version)`
`44`	`44`	`return unittest.skipUnless(`
`45`		`- torch_version < provided_version,`
	`45`	`+ torch_version >= provided_version,`
`46`	`46`	`f"torch version {torch_version} is less than {provided_version}",`
`47`	`47`	`)`