Skip to content

Commit ee3540e

Browse files
authored
typing(autogram): Fix typing error in setup_context (#602)
1 parent 5d084fc commit ee3540e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/torchjd/autogram/_jacobian_computer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -195,5 +195,5 @@ def vmap(
195195
return cast(Matrix, jacobian), None
196196

197197
@staticmethod
198-
def setup_context(*_) -> None:
198+
def setup_context(*_, **__) -> None:
199199
pass

0 commit comments

Comments
 (0)