Adjoint: annotate __itruediv__

firedrakeproject · Jul 12, 2024 · 1d560a6 · 1d560a6
1 parent 7372172
commit 1d560a6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/firedrake/adjoint_utils/function.py b/firedrake/adjoint_utils/function.py
@@ -196,11 +196,11 @@ def wrapper(self, other, **kwargs):
         return wrapper
 
     @staticmethod
-    def _ad_annotate_idiv(__idiv__):
-        @wraps(__idiv__)
+    def _ad_annotate_itruediv(__itruediv__):
+        @wraps(__itruediv__)
         def wrapper(self, other, **kwargs):
             with stop_annotating():
-                func = __idiv__(self, other, **kwargs)
+                func = __itruediv__(self, other, **kwargs)
 
             ad_block_tag = kwargs.pop("ad_block_tag", None)
             annotate = annotate_tape(kwargs)

diff --git a/firedrake/function.py b/firedrake/function.py
@@ -521,7 +521,7 @@ def __imul__(self, expr):
         IMulAssigner(self, expr).assign()
         return self
 
-    @FunctionMixin._ad_annotate_idiv
+    @FunctionMixin._ad_annotate_itruediv
     def __itruediv__(self, expr):
         from firedrake.assign import IDivAssigner
         IDivAssigner(self, expr).assign()