Apply suggestions from code review

Co-authored-by: Joe <joe@preferred.jp>
linshokaku · Dec 6, 2023 · f1fd387 · f1fd387
1 parent 2903273
commit f1fd387
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pytorch_pfn_extras/ops/register.py b/pytorch_pfn_extras/ops/register.py
@@ -13,7 +13,7 @@
 class OpDesc:
     """Metadata to register an op to torch.library.
 
-    Attributed:
+    Attributes:
         op (callable): code to be executed in the forward/backward of the op.
         meta (callable): function to perform shape inference for forward/backward
             passes.

diff --git a/tests/pytorch_pfn_extras_tests/test_ops/test_register.py b/tests/pytorch_pfn_extras_tests/test_ops/test_register.py
@@ -13,7 +13,7 @@ def _get_function_nodes(fx_module):
 
 @pytest.mark.skipif(
     not ppe.requires("2.1.0") or sys.platform == "win32",
-    reason="torch custom ops only works for PyTorch>2.1 and linux",
+    reason="torch custom ops only works for PyTorch>=2.1 and linux",
 )
 def test_register():
     def test(a):