bugfix

flatironinstitute · Feb 9, 2024 · e16dcad · e16dcad
1 parent f702c77
commit e16dcad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/nemos/utils.py b/src/nemos/utils.py
@@ -414,7 +414,7 @@ def check_invalid_entry(pytree: Any, pytree_name: str) -> Tuple[Any, Union[None,
     Returns
     -------
     is_valid:
-        A pytree with leaves flat boolean arrays of shape (n_sample_points, ) with
+        A boolean aray of shape (n_sample_points, ) with
         the invalid indices (samples with Nans or Infs).
     err:
         Either None, if all entries are valid, or an error message.