From 8352b386f60eedcdb2df72ce8d88769a51a5228c Mon Sep 17 00:00:00 2001
From: Mike Henry <11765982+mikemhenry@users.noreply.github.com>
Date: Wed, 13 Nov 2024 13:21:08 -0700
Subject: [PATCH] Fix 2 more tests that fail sometimes in CI (#751)

---
 openmmtools/tests/test_sampling.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/openmmtools/tests/test_sampling.py b/openmmtools/tests/test_sampling.py
index c4905b25..e4c0710a 100644
--- a/openmmtools/tests/test_sampling.py
+++ b/openmmtools/tests/test_sampling.py
@@ -316,6 +316,7 @@ def test_with_unsampled_states(self):
     # on windows we get a ZeroDivisionError: float division by zero
     # when measuring the timing data
     @pytest.mark.skipif(sys.platform.startswith("win"), reason="Test fails on windows")
+    @pytest.mark.flaky(reruns=3)
     def test_without_unsampled_states(self):
         """Test multistate sampler on a harmonic oscillator without unsampled endstates"""
         self.run(include_unsampled_states=False)