Dev: remove unused import

Author: BillHuang2001

src/evox/monitors/eval_monitor.py

@@ -1,7 +1,6 @@
 import warnings
 
 import jax
-import jax.experimental.host_callback as hcb
 import jax.numpy as jnp
 import numpy as np
 from jax.experimental import io_callback

src/evox/monitors/evoxvis_monitor.py

@@ -5,7 +5,6 @@
 from typing import Optional
 
 import jax
-import jax.experimental.host_callback as hcb
 import pyarrow as pa
 
 

src/evox/monitors/std_mo_monitor.py

@@ -1,7 +1,6 @@
 import warnings
 
 import jax
-import jax.experimental.host_callback as hcb
 import jax.numpy as jnp
 import numpy as np
 from jax.experimental import io_callback

src/evox/monitors/std_so_monitor.py

@@ -1,7 +1,6 @@
 import warnings
 
 import jax
-import jax.experimental.host_callback as hcb
 import jax.numpy as jnp
 from jax.experimental import io_callback
 from jax.sharding import SingleDeviceSharding