0

patrick-kidger · Aug 7, 2024 · b183abd · b183abd
1 parent 3aee2f5
commit b183abd
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/diffrax/_integrate.py b/diffrax/_integrate.py
@@ -1014,10 +1014,15 @@ def _promote(yi):
     del timelikes
 
     # Backward compatibility
-    if isinstance(
-        solver, (EulerHeun, ItoMilstein, StratonovichMilstein)
-    ) and _term_compatible(
-        y0, args, terms, (ODETerm, AbstractTerm), solver.term_compatible_contr_kwargs
+    if (
+        isinstance(solver, (EulerHeun, ItoMilstein, StratonovichMilstein))
+        and _term_compatible(
+            y0,
+            args,
+            terms,
+            (ODETerm, AbstractTerm),
+            solver.term_compatible_contr_kwargs,
+        )[0]
     ):
         warnings.warn(
             "Passing `terms=(ODETerm(...), SomeOtherTerm(...))` to "