lint: appease mypy

metaopt · Dec 9, 2023 · 26903cb · 26903cb
1 parent 99c5569
commit 26903cb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/torchopt/diff/implicit/decorator.py b/torchopt/diff/implicit/decorator.py
@@ -279,7 +279,7 @@ def make_custom_vjp_solver_fn(
         # pylint: disable-next=missing-class-docstring,abstract-method
         class ImplicitMetaGradient(Function):
             @staticmethod
-            def forward(  # type: ignore[override] # pylint: disable=arguments-differ
+            def forward(  # pylint: disable=arguments-differ
                 ctx: Any,
                 *flat_args: Any,
             ) -> tuple[Any, ...]:

diff --git a/torchopt/distributed/api.py b/torchopt/distributed/api.py
@@ -324,7 +324,7 @@ def remote_async_call(
     if reducer is not None:
         return cast(
             Future[U],
-            future.then(lambda fut: cast(Callable[[Iterable[T]], U], reducer)(fut.wait())),
+            future.then(lambda fut: reducer(fut.wait())),
         )
     return future