From c547f2fa052f84e9fc93f0a8b4af80c43ab351cd Mon Sep 17 00:00:00 2001
From: Nathaniel Starkman <nstarman@users.noreply.github.com>
Date: Mon, 10 Feb 2025 13:17:04 -0500
Subject: [PATCH] =?UTF-8?q?=F0=9F=A9=B9=20fix-simple:=20skip=20applying=20?=
 =?UTF-8?q?to=20solution=20if=20already=20vectorized?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/diffraxtra/interp.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/diffraxtra/interp.py b/src/diffraxtra/interp.py
index 77b51ad..4329807 100644
--- a/src/diffraxtra/interp.py
+++ b/src/diffraxtra/interp.py
@@ -420,7 +420,7 @@ def apply_to_solution(cls, soln: dfx.Solution, /) -> dfx.Solution:
         in a `VectorizedDenseInterpolation`.
 
         """
-        if soln.interpolation is None:
+        if soln.interpolation is None or isinstance(soln.interpolation, cls):  # type: ignore[unreachable]
             return soln
 
         vec_soln: dfx.Solution = eqx.tree_at(