Increase test deadline for sliced_access

HumanCompatibleAI · Jan 17, 2024 · 7e618b0 · 7e618b0
1 parent 9239ca7
commit 7e618b0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/data/test_huggingface_utils.py b/tests/data/test_huggingface_utils.py
@@ -63,6 +63,8 @@ def test_save_load_roundtrip(
 
 
 @hypothesis.given(st.data(), h_strats.trajectories_list)
+# the first run sometimes takes longer, so we give it more time
+@hypothesis.settings(deadline=datetime.timedelta(milliseconds=300))
 def test_sliced_access(data: st.DataObject, trajectories: Sequence[types.Trajectory]):
     """Test that slicing a TrajectoryDatasetSequence behaves as expected."""
     # GIVEN