From bae0c216bcf49f055d74ce036dd6ca9f55ab4f54 Mon Sep 17 00:00:00 2001 From: Remi Date: Thu, 18 Apr 2024 11:04:19 +0200 Subject: [PATCH] Update examples/1_load_hugging_face_dataset.py Co-authored-by: Alexander Soare --- examples/1_load_hugging_face_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/1_load_hugging_face_dataset.py b/examples/1_load_hugging_face_dataset.py index 7f146feb..437d96b1 100644 --- a/examples/1_load_hugging_face_dataset.py +++ b/examples/1_load_hugging_face_dataset.py @@ -44,7 +44,7 @@ print(f"{hf_dataset=}") print(f"{hf_dataset.features=}") # display useful statistics about frames and episodes, which are sequences of frames from the same video -print(f"number of samples/frames: {len(hf_dataset)=}") +print(f"number of frames: {len(hf_dataset)=}") print(f"number of episodes: {len(hf_dataset.unique('episode_id'))=}") print(f"average number of frames per episode: {len(hf_dataset) / len(hf_dataset.unique('episode_id')):.3f}")