Increase test deadline for sliced_info_dict_access.

HumanCompatibleAI · Jan 17, 2024 · 9239ca7 · 9239ca7
1 parent 4b7b263
commit 9239ca7
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
@@ -84,6 +84,8 @@ def test_sliced_access(data: st.DataObject, trajectories: Sequence[types.Traject
 
 
 @hypothesis.given(st.data(), h_strats.trajectory)
+# the first run sometimes takes longer, so we give it more time
+@hypothesis.settings(deadline=datetime.timedelta(milliseconds=300))
 def test_sliced_info_dict_access(
     data: st.DataObject,
     trajectory: types.Trajectory,