Skip to content

Commit ca6fb10

Browse files
authored
Merge pull request #216 from mrava87/fix-VStackhasgrad
fix: ensure VStack has hasgrad attribute
2 parents 3fdeb8e + e27c7e3 commit ca6fb10

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

pyproximal/proximal/VStack.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@ def __init__(
5454
nn: Optional[list[ShapeLike]] = None,
5555
restr: Optional[list["LinearOperator"]] = None,
5656
) -> None:
57+
super().__init__(None, any(op.hasgrad for op in ops))
5758
self.ops = ops
5859

5960
if nn is not None:

0 commit comments

Comments
 (0)