TypeError: unsupported operand type

[adam-hartshorne]

When attempting to run gaussian-process-interpolation example, I get the following error, where it is complaining about the else if check of ts being an empty list.

Traceback (most recent call last):
  File "/media/adam/shared_drive/PycharmProjects/neural_diffusion_processes-main/notebooks/gaussian-process-interpolation.py", line 214, in <module>
    ys = jax.vmap(solve)(y0s, subkeys)
  File "/media/adam/shared_drive/PycharmProjects/neural_diffusion_processes-main/notebooks/gaussian-process-interpolation.py", line 199, in <lambda>
    lambda y, key: ndp.sde.sde_solve(
  File "/media/adam/shared_drive/PycharmProjects/neural_diffusion_processes-main/neural_diffusion_processes/sde.py", line 439, in sde_solve
    elif ts == []:
  File "/home/adam/anaconda3/envs/geo_np/lib/python3.10/site-packages/jax/_src/numpy/array_methods.py", line 260, in deferring_binary_op
    raise TypeError(f"unsupported operand type(s) for {opchar}: "
TypeError: unsupported operand type(s) for ==: 'ArrayImpl' and 'list'