Increase test deadline for sliced_info_dict_access.

HumanCompatibleAI · Dec 18, 2023 · 7474c6c · 7474c6c
1 parent d2c22c1
commit 7474c6c
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,