pre-commit run --all-files

This commit is contained in:
Remi Cadene 2024-05-30 09:11:51 +00:00
parent 2a046dd73b
commit 52529805e8
1 changed files with 1 additions and 1 deletions

View File

@ -114,7 +114,7 @@ def load_from_raw(raw_dir: Path, out_dir: Path, fps: int):
# sanity check
has_nan = df.isna().any().any()
if has_nan:
raise ValueError("Dataset contains Nan values.")
raise ValueError("Dataset contains Nan values.")
# sanity check episode indices go from 0 to n-1
ep_ids = [ep_idx for ep_idx, _ in df.groupby("episode_index")]