Oops, forget linting

Author: ProGamerGov

captum/optim/_utils/typing.py

@@ -38,4 +38,7 @@ def cleanup(self):
 TransformValList = Union[Sequence[int], Sequence[float], Tensor]
 TransformVal = Union[int, float, Tensor]
 TransformSize = Union[List[int], Tuple[int], int]
-HookFuncArg = Callable[[Module, Union[Tuple[Tensor, ...], Tensor], Union[Tuple[Tensor, ...], Tensor]], Optional[Tensor]]
+HookFuncArg = Callable[
+    [Module, Union[Tuple[Tensor, ...], Tensor], Union[Tuple[Tensor, ...], Tensor]],
+    Optional[Tensor],
+]