We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5d084fc commit ee3540eCopy full SHA for ee3540e
1 file changed
src/torchjd/autogram/_jacobian_computer.py
@@ -195,5 +195,5 @@ def vmap(
195
return cast(Matrix, jacobian), None
196
197
@staticmethod
198
- def setup_context(*_) -> None:
+ def setup_context(*_, **__) -> None:
199
pass
0 commit comments