Fix code style.

keras/backend/jax/trainer.py

@@ -7,7 +7,6 @@
 
 from keras import backend
 from keras import callbacks as callbacks_module
-from keras import ops
 from keras import optimizers as optimizers_module
 from keras.backend import distribution_lib as jax_distribution_lib
 from keras.distribution import distribution_lib

keras/trainers/trainer.py

@@ -967,6 +967,7 @@ def to_symbolic_input(v):
                 y,
                 sample_weight,
             ) = data_adapter_utils.unpack_x_y_sample_weight(data_batch)
+
             # Build all model state with `backend.compute_output_spec`.
             try:
                 y_pred = backend.compute_output_spec(self, x)