Resolve type checking issues

lanl · Dec 6, 2023 · d0ad4e4 · d0ad4e4
1 parent 4312239
commit d0ad4e4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/scico/linop/_stack.py b/scico/linop/_stack.py
@@ -11,7 +11,7 @@
 
 import operator
 from functools import partial
-from typing import List, Optional, Union
+from typing import Optional, Sequence, Union
 
 import scico.numpy as snp
 from scico.numpy import Array, BlockArray
@@ -48,7 +48,7 @@ class VerticalStack(VStack, LinearOperator):
 
     def __init__(
         self,
-        ops: List[LinearOperator],
+        ops: Sequence[LinearOperator],
         collapse_output: Optional[bool] = True,
         jit: bool = True,
         **kwargs,
@@ -66,7 +66,7 @@ def __init__(
         super().__init__(ops=ops, collapse_output=collapse_output, jit=jit, **kwargs)
 
     def _adj(self, y: Union[Array, BlockArray]) -> Array:  # type: ignore
-        return sum([op.adj(y_block) for y_block, op in zip(y, self.ops)])
+        return sum([op.adj(y_block) for y_block, op in zip(y, self.ops)])  # type: ignore
 
     @partial(_wrap_add_sub, op=operator.add)
     def __add__(self, other):
@@ -126,7 +126,7 @@ class DiagonalStack(DStack, LinearOperator):
 
     def __init__(
         self,
-        ops: List[LinearOperator],
+        ops: Sequence[LinearOperator],
         collapse_input: Optional[bool] = True,
         collapse_output: Optional[bool] = True,
         jit: bool = True,
@@ -151,7 +151,7 @@ def __init__(
         )
 
     def _adj(self, y: Union[Array, BlockArray]) -> Union[Array, BlockArray]:  # type: ignore
-        result = tuple(op.T @ y_n for op, y_n in zip(self.ops, y))
+        result = tuple(op.T @ y_n for op, y_n in zip(self.ops, y))  # type: ignore
         if self.collapse_input:
             return snp.stack(result)
         return snp.blockarray(result)
diff --git a/scico/operator/_stack.py b/scico/operator/_stack.py
@@ -9,7 +9,7 @@
 
 from __future__ import annotations
 
-from typing import List, Optional, Sequence, Tuple, Union
+from typing import Optional, Sequence, Tuple, Union
 
 import numpy as np
 
@@ -78,7 +78,7 @@ class VerticalStack(Operator):
 
     def __init__(
         self,
-        ops: List[Operator],
+        ops: Sequence[Operator],
         collapse_output: Optional[bool] = True,
         jit: bool = True,
         **kwargs,
@@ -116,7 +116,7 @@ def __init__(
         )
 
     @staticmethod
-    def check_if_stackable(ops: List[Operator]):
+    def check_if_stackable(ops: Sequence[Operator]):
         """Check that input ops are suitable for stack creation."""
         if not isinstance(ops, (list, tuple)):
             raise ValueError("Expected a list of Operator.")
@@ -227,7 +227,7 @@ class DiagonalStack(Operator):
 
     def __init__(
         self,
-        ops: List[Operator],
+        ops: Sequence[Operator],
         collapse_input: Optional[bool] = True,
         collapse_output: Optional[bool] = True,
         jit: bool = True,