Skip to content

Commit ffd16f4

Browse files
committed
add two-arg seed_for_storage variant
1 parent b8d9fa1 commit ffd16f4

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

src/sampling.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ function sample_joint(
193193
end
194194

195195
# seed
196-
@everywhere seed_for_storage!($storage)
196+
@everywhere seed_for_storage!((Array,$storage))
197197

198198
# initialize chains
199199
if (filename != nothing) && isfile(filename)

src/util.jl

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -353,13 +353,16 @@ end
353353

354354
@doc doc"""
355355
seed_for_storage!(storage[, seed])
356+
seed_for_storage!((storage1, storage2, ...)[, seed])
356357
357358
Set the global random seed for the RNG which controls `storage`-type.
358359
"""
359360
seed_for_storage!(::Type{<:Array}, seed=nothing) =
360361
Random.seed!((seed == nothing ? () : (seed,))...)
361362
seed_for_storage!(storage::Any, seed=nothing) =
362363
error("Don't know how to set seed for storage=$storage")
364+
seed_for_storage!(storages::Tuple, seed=nothing) =
365+
seed_for_storage!.(storages, seed)
363366

364367

365368
### parallel utility function

0 commit comments

Comments
 (0)