diff --git a/test/test_transforms.py b/test/test_transforms.py index 20bf5b0c0f7..d9c219ee526 100644 --- a/test/test_transforms.py +++ b/test/test_transforms.py @@ -4,6 +4,7 @@ # LICENSE file in the root directory of this source tree. import argparse from copy import copy +from time import sleep import pytest import torch @@ -154,6 +155,8 @@ def test_vecnorm_parallel(nprc): for idx in range(nprc): queues[idx][1].put(msg) del queues + for p in prcs: + p.join() def _test_vecnorm_subproc_auto(idx, make_env, queue_out: mp.Queue, queue_in: mp.Queue): @@ -219,6 +222,9 @@ def test_vecnorm_parallel_auto(nprc): for idx in range(nprc): queues[idx][1].put(msg) + sleep(0.01) # Further fix for flacky test: vecnorm should have a locking + # mechanism + obs_sum = td.get("next_observation_sum").clone() obs_ssq = td.get("next_observation_ssq").clone() obs_count = td.get("next_observation_count").clone()