Skip to content

Commit 881c65a

Browse files
aseyboldtjunpenglao
andcommitted
Better formatting in test_parallel_sampling
Co-authored-by: Junpeng Lao <[email protected]>
1 parent 8cc6f8f commit 881c65a

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

pymc3/tests/test_parallel_sampling.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -50,13 +50,8 @@ def test_bad_unpickle():
5050
assert 'could not be unpickled' in str(exc_info.getrepr(style='short'))
5151

5252

53-
@theano.as_op(
54-
[
55-
tt.dvector if theano.config.floatX == "float64" else tt.fvector,
56-
tt.iscalar,
57-
],
58-
[tt.dvector if theano.config.floatX == "float64" else tt.fvector],
59-
)
53+
tt_vector = tt.TensorType(theano.config.floatX, [False])
54+
@theano.as_op([tt_vector, tt.iscalar], [tt_vector])
6055
def _crash_remote_process(a, master_pid):
6156
if os.getpid() != master_pid:
6257
os.exit(0)

0 commit comments

Comments
 (0)