From bdda3618d7766b7ec6d2076e8af2d7b1958a2dba Mon Sep 17 00:00:00 2001
From: Evgenii Zheltonozhskii <zheltonozhskiy@gmail.com>
Date: Tue, 31 Oct 2023 20:17:25 +0200
Subject: [PATCH] Nit fix

---
 diffrax/integrate.py   | 7 -------
 test/test_integrate.py | 3 ---
 2 files changed, 10 deletions(-)

diff --git a/diffrax/integrate.py b/diffrax/integrate.py
index 548e05f3..15959c7d 100644
--- a/diffrax/integrate.py
+++ b/diffrax/integrate.py
@@ -19,7 +19,6 @@
 from .saveat import SaveAt, SubSaveAt
 from .solution import is_okay, is_successful, RESULTS, Solution
 from .solver import (
-    AbstractImplicitSolver,
     AbstractItoSolver,
     AbstractSolver,
     AbstractStratonovichSolver,
@@ -608,11 +607,6 @@ def diffeqsolve(
 
     # Error checking and warning for complex dtypes
     if any(jtu.tree_leaves(jtu.tree_map(jnp.iscomplexobj, y0))):
-        if isinstance(solver, AbstractImplicitSolver):
-            raise ValueError(
-                "Implicit solvers in conjunction with complex dtypes is currently not "
-                "supported."
-            )
         warnings.warn(
             "Complex dtype support is work in progress, please read "
             "https://github.com/patrick-kidger/diffrax/pull/197 and proceed carefully."
@@ -667,7 +661,6 @@ def diffeqsolve(
     t1 = jnp.asarray(t1, dtype=dtype)
     if dt0 is not None:
         dt0 = jnp.asarray(dt0, dtype=dtype)
-    timelikes.append(dtype)
 
     def _get_subsaveat_ts(saveat):
         out = [s.ts for s in jtu.tree_leaves(saveat.subs, is_leaf=_is_subsaveat)]
diff --git a/test/test_integrate.py b/test/test_integrate.py
index e547105f..e8da0865 100644
--- a/test/test_integrate.py
+++ b/test/test_integrate.py
@@ -77,9 +77,6 @@ def test_basic(solver, t_dtype, y_dtype, treedef, stepsize_controller, getkey):
         def f(t, y, args):
             return jtu.tree_map(lambda _y: operator.mul(-1j, _y), y)
 
-        if isinstance(solver, diffrax.AbstractImplicitSolver):
-            return
-
     else:
 
         def f(t, y, args):