From 9fe006142889faf74b25cdfc5eb7bd41bb32cab3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Val=C3=A9rian=20Rey?= <valerian.rey@gmail.com>
Date: Mon, 2 Mar 2026 23:56:57 +0100
Subject: [PATCH] Fix typing error in setup_context

---
 src/torchjd/autogram/_jacobian_computer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/torchjd/autogram/_jacobian_computer.py b/src/torchjd/autogram/_jacobian_computer.py
index 6bddbd60..a93bd5aa 100644
--- a/src/torchjd/autogram/_jacobian_computer.py
+++ b/src/torchjd/autogram/_jacobian_computer.py
@@ -195,5 +195,5 @@ def vmap(
         return cast(Matrix, jacobian), None
 
     @staticmethod
-    def setup_context(*_) -> None:
+    def setup_context(*_, **__) -> None:
         pass