From 3c2dd1b881645186888ec12cc2a00f13f4973c48 Mon Sep 17 00:00:00 2001
From: Thomas Wolf <thomas@huggingface.co>
Date: Tue, 4 Jun 2024 15:43:10 +0200
Subject: [PATCH] fix test
---
tests/test_datasets.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/test_datasets.py b/tests/test_datasets.py
index da0ae755..358c7627 100644
--- a/tests/test_datasets.py
+++ b/tests/test_datasets.py
@@ -244,7 +244,7 @@ def test_load_previous_and_future_frames_outside_tolerance_inside_episode_range(
delta_timestamps = {"index": [-0.2, 0, 0.141]}
tol = 0.04
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)