gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ci/runtests.sh - pass all folders to pytest (#342)

Browse Source 
* ci/runtests.sh - pass all folders to pytest

* mpi_optimizer_test precision 1e-4

* fixes to tests

* search for tests in the entire jax folder, also remove unnecessary humor
This commit is contained in:
pzhokhov
2019-04-22 14:41:46 -07:00
committed by Peter Zhokhov
parent ddcab1606d
commit 8e0282ee94
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
baselines/common/tests/test_with_mpi.py
View File

@@ -4,6 +4,7 @@ import subprocess
import cloudpickle import cloudpickle
import base64 import base64
import pytest import pytest
from functools import wraps
try: try:
from mpi4py import MPI from mpi4py import MPI
@@ -12,6 +13,7 @@ except ImportError:
def with_mpi(nproc=2, timeout=30, skip_if_no_mpi=True): def with_mpi(nproc=2, timeout=30, skip_if_no_mpi=True):
def outer_thunk(fn): def outer_thunk(fn):
@wraps(fn)
def thunk(*args, **kwargs): def thunk(*args, **kwargs):
serialized_fn = base64.b64encode(cloudpickle.dumps(lambda: fn(*args, **kwargs))) serialized_fn = base64.b64encode(cloudpickle.dumps(lambda: fn(*args, **kwargs)))
subprocess.check_call([ subprocess.check_call([