This commit is contained in:
Thomas Wolf 2024-06-04 15:43:10 +02:00
parent 63a5d0be39
commit 3c2dd1b881
1 changed files with 1 additions and 1 deletions

View File

@ -244,7 +244,7 @@ def test_load_previous_and_future_frames_outside_tolerance_inside_episode_range(
delta_timestamps = {"index": [-0.2, 0, 0.141]} delta_timestamps = {"index": [-0.2, 0, 0.141]}
tol = 0.04 tol = 0.04
item = hf_dataset[2] item = hf_dataset[2]
with pytest.raises(AssertionError): with pytest.raises(ValueError):
load_previous_and_future_frames(item, hf_dataset, episode_data_index, delta_timestamps, tol) load_previous_and_future_frames(item, hf_dataset, episode_data_index, delta_timestamps, tol)