wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add comment on test

This commit is contained in:
Alexander Soare 2024-04-02 16:57:25 +01:00
parent a6edb85da4
commit 148df1c1d5
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_datasets.py
View File

@ -54,7 +54,8 @@ def test_compute_stats():
sampler=SamplerWithoutReplacement(), sampler=SamplerWithoutReplacement(),
).sample().float() ).sample().float()
# Note: we set the batch size to be smaller than the whole dataset to make sure we are testing batched # Note: we set the batch size to be smaller than the whole dataset to make sure we are testing batched
# computation of the statistics. # computation of the statistics. While doing this, we also make sure it works when we don't divide the
# dataset into even batches.
computed_stats = buffer._compute_stats(batch_size=int(len(all_data) * 0.75)) computed_stats = buffer._compute_stats(batch_size=int(len(all_data) * 0.75))
for k, pattern in buffer.stats_patterns.items(): for k, pattern in buffer.stats_patterns.items():
expected_mean = einops.reduce(all_data[k], pattern, "mean") expected_mean = einops.reduce(all_data[k], pattern, "mean")