set global deadline

PlaytikaOSS · Oct 16, 2023 · add3186 · add3186
1 parent 11557be
commit add3186
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_cmab.py b/tests/test_cmab.py
@@ -53,6 +53,7 @@
 # CmabBernoulli with strategy=ClassicBandit()
 
 
+@settings(deadline=400)  # Set the global deadline to 400 ms
 @given(st.integers(max_value=100))
 def test_create_cmab_bernoulli_cold_start(a_int):
     # n_features must be > 0
@@ -231,7 +232,6 @@ def run_predict(context):
     run_predict(context=context)
 
 
-@settings(deadline=400)  # Set the deadline to 400 ms
 @given(st.integers(min_value=1, max_value=100), st.integers(min_value=1, max_value=3))
 def test_cmab_predict_not_cold_start(n_samples, n_features):
     def run_predict(context):