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

fix and clarify tests

This commit is contained in:
Cadene 2024-04-11 13:16:47 +00:00
parent 92701088a3
commit 36de77ac18
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_datasets.py
View File

@ -150,7 +150,7 @@ def test_load_data_with_delta_timestamps_within_tolerance():
"index": torch.tensor([0, 1, 2, 3, 4]), "index": torch.tensor([0, 1, 2, 3, 4]),
} }
data_ids_per_episode = {0: torch.tensor([0, 1, 2, 3, 4])} data_ids_per_episode = {0: torch.tensor([0, 1, 2, 3, 4])}
delta_timestamps = {"index": [-0.2, 0, 0.24]} delta_timestamps = {"index": [-0.2, 0, 0.139]}
key = "index" key = "index"
current_ts = 0.3 current_ts = 0.3
episode = 0 episode = 0
@ -165,11 +165,11 @@ def test_load_data_with_delta_timestamps_outside_tolerance_inside_episode_range(
"index": torch.tensor([0, 1, 2, 3, 4]), "index": torch.tensor([0, 1, 2, 3, 4]),
} }
data_ids_per_episode = {0: torch.tensor([0, 1, 2, 3, 4])} data_ids_per_episode = {0: torch.tensor([0, 1, 2, 3, 4])}
delta_timestamps = {"index": [-0.2, 0, 0.14, 0.2]} delta_timestamps = {"index": [-0.2, 0, 0.141]}
key = "index" key = "index"
current_ts = 0.3 current_ts = 0.3
episode = 0 episode = 0
tol = 0.03 tol = 0.04
with pytest.raises(AssertionError): with pytest.raises(AssertionError):
load_data_with_delta_timestamps(data_dict, data_ids_per_episode, delta_timestamps, key, current_ts, episode, tol) load_data_with_delta_timestamps(data_dict, data_ids_per_episode, delta_timestamps, key, current_ts, episode, tol)