CI: fix lint check

jax/_src/nn/functions.py

@@ -919,7 +919,7 @@ def _ensure_4d(t):
     if dims_to_add > 0:
       return jnp.expand_dims(t, axis=tuple(range(dims_to_add)))
     return t
-  
+
   query = _ensure_4d(query)
   key = _ensure_4d(key)
   value = _ensure_4d(value)