typo

PiperOrigin-RevId: 473291975
google · Sep 9, 2022 · 7c8729d · 7c8729d
1 parent 628ce0e
commit 7c8729d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neural_tangents/_src/empirical.py b/neural_tangents/_src/empirical.py
@@ -1307,8 +1307,8 @@ def empirical_ntk_vp_fn(
       second batch of inputs. `x2=None` means `x2=x1`.
 
     params:
-      A `PyTree` of parameters about which we would like to compute the
-        neural tangent kernel.
+      A `PyTree` of parameters about which we would like to compute the neural
+      tangent kernel.
 
     **apply_fn_kwargs:
       keyword arguments passed to `f`. `apply_fn_kwargs` will be split into