Fix comment

jax_toolkit/losses/utils.py

@@ -51,7 +51,7 @@ def get_haiku_loss_function(
         def loss_function_wrapper(
             params: hk.Params, x: jnp.ndarray, y_true: jnp.ndarray, rng: jnp.ndarray = None
         ) -> jnp.ndarray:
-            # rng argument can be used is net_transform.apply() is non-deterministic, and you require and "random seed"
+            # rng argument can be used if net_transform.apply() is non-deterministic, and you require a "random seed"
             y_pred: jnp.ndarray = net_transform.apply(params, rng, x)
             loss_value: jnp.ndarray = loss_function(y_true, y_pred)
             return loss_value