Increase test deadline for sliced_info_dict_access.

HumanCompatibleAI · Dec 18, 2023 · 563bccf · 563bccf
1 parent c9f0956
commit 563bccf
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,