Skip to content

Commit e51b945

Browse files
committed
Remove flaky test
1 parent b5bc9ff commit e51b945

File tree

1 file changed

+1
-27
lines changed

1 file changed

+1
-27
lines changed

pymc3/tests/test_smc.py

Lines changed: 1 addition & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,6 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
import time
16-
1715
import aesara
1816
import aesara.tensor as at
1917
import numpy as np
@@ -144,31 +142,7 @@ def test_convergence_checks(self):
144142
):
145143
pm.sample_smc(draws=99)
146144

147-
def test_parallel_sampling(self):
148-
# Cache graph
149-
with self.slow_model:
150-
_ = pm.sample_smc(draws=10, chains=1, cores=1, return_inferencedata=False)
151-
152-
chains = 4
153-
draws = 100
154-
155-
t0 = time.time()
156-
with self.slow_model:
157-
idata = pm.sample_smc(draws=draws, chains=chains, cores=4)
158-
t_mp = time.time() - t0
159-
assert idata.posterior.dims["chain"] == chains
160-
assert idata.posterior.dims["draw"] == draws
161-
162-
t0 = time.time()
163-
with self.slow_model:
164-
idata = pm.sample_smc(draws=draws, chains=chains, cores=1)
165-
t_seq = time.time() - t0
166-
assert idata.posterior.dims["chain"] == chains
167-
assert idata.posterior.dims["draw"] == draws
168-
169-
assert t_mp < t_seq
170-
171-
def test_depracated_parallel_arg(self):
145+
def test_deprecated_parallel_arg(self):
172146
with self.fast_model:
173147
with pytest.warns(
174148
DeprecationWarning,

0 commit comments

Comments
 (0)