[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

thunder/core/transforms.py

@@ -1108,8 +1108,8 @@ def _sum_prim_grad(a: TensorProxy, /, dims: Sequence[int]) -> TensorProxy:
 register_grad(pids.SUM, _sum_prim_grad)
 
 
-#@torchctx
-#def _topk_prim_grad(a: TensorProxy, /, k: int, dim: None | int = None, largest: bool = True, sorted: bool = True, *, out=None):
+# @torchctx
+# def _topk_prim_grad(a: TensorProxy, /, k: int, dim: None | int = None, largest: bool = True, sorted: bool = True, *, out=None):
 #    fwd = prims.topk(a, k, dim, largest, sorted, out=out)
 #    val, idx = fwd
 #
@@ -1124,7 +1124,7 @@ def _sum_prim_grad(a: TensorProxy, /, dims: Sequence[int]) -> TensorProxy:
 #    return fwd
 #
 #
-#register_grad(pids.TOPK, _topk_prim_grad)
+# register_grad(pids.TOPK, _topk_prim_grad)
 
 
 # TODO Fix division by zero when n_elem_reduced == 0 or when mean.numel == 0

thunder/tests/opinfos.py

@@ -4771,13 +4771,15 @@ def topk_thunder_ref(*args, **kwargs):
 
 def topk_torch_ref(*args, **kwargs):
     return torch.topk(*args, **kwargs)[0]
+
+
 # }
 
 
 topk_opinfo = OpInfo(
     topk_thunder_ref,
     name="topk",
-    #supports_grad=True,
+    # supports_grad=True,
     # Without the fixed seed this generator does not guarantee
     # to produce inputs at which topk is differentiable
     # (i.e. when topk(x, ...).indices == topk(x + dx, ...).indices).