|
17 | 17 | from tests.subprocesstest import run_in_subprocess |
18 | 18 | from tests.utils import DummyTracer |
19 | 19 | from tests.utils import DummyWriter |
| 20 | +from tests.utils import flaky |
20 | 21 | from tests.utils import override_global_config |
21 | 22 |
|
22 | 23 |
|
@@ -140,18 +141,22 @@ def _test_span_sampling(self, rate): |
140 | 141 | sampled += 1 |
141 | 142 | assert (rate * num_completions - 30) < sampled < (rate * num_completions + 30) |
142 | 143 |
|
| 144 | + @flaky(until=1752686557) |
143 | 145 | @run_in_subprocess(env_overrides=dict(DD_BEDROCK_SPAN_PROMPT_COMPLETION_SAMPLE_RATE="0.0")) |
144 | 146 | def test_span_sampling_0(self): |
145 | 147 | self._test_span_sampling(rate=float(os.getenv("DD_BEDROCK_SPAN_PROMPT_COMPLETION_SAMPLE_RATE"))) |
146 | 148 |
|
| 149 | + @flaky(until=1752686557) |
147 | 150 | @run_in_subprocess(env_overrides=dict(DD_BEDROCK_SPAN_PROMPT_COMPLETION_SAMPLE_RATE="0.25")) |
148 | 151 | def test_span_sampling_25(self): |
149 | 152 | self._test_span_sampling(rate=float(os.getenv("DD_BEDROCK_SPAN_PROMPT_COMPLETION_SAMPLE_RATE"))) |
150 | 153 |
|
| 154 | + @flaky(until=1752686557) |
151 | 155 | @run_in_subprocess(env_overrides=dict(DD_BEDROCK_SPAN_PROMPT_COMPLETION_SAMPLE_RATE="0.75")) |
152 | 156 | def test_span_sampling_75(self): |
153 | 157 | self._test_span_sampling(rate=float(os.getenv("DD_BEDROCK_SPAN_PROMPT_COMPLETION_SAMPLE_RATE"))) |
154 | 158 |
|
| 159 | + @flaky(until=1752686557) |
155 | 160 | @run_in_subprocess(env_overrides=dict(DD_BEDROCK_SPAN_PROMPT_COMPLETION_SAMPLE_RATE="1.0")) |
156 | 161 | def test_span_sampling_100(self): |
157 | 162 | self._test_span_sampling(rate=float(os.getenv("DD_BEDROCK_SPAN_PROMPT_COMPLETION_SAMPLE_RATE"))) |
|
0 commit comments