Soss tests passing

Author: cscherrer

src/combinators/power.jl

@@ -24,12 +24,17 @@ function Pretty.tile(μ::PowerMeasure)
     return Pretty.pair_layout(arg1, arg2; sep = " ^ ")
 end
 
-function Base.rand(rng::AbstractRNG, ::Type{T}, d::PowerMeasure) where {T}
+function Base.rand(rng::AbstractRNG, ::Type{T}, d::PowerMeasure{M}) where {T, M<:AbstractMeasure}
     map(CartesianIndices(d.axes)) do _
         rand(rng, T, d.parent)
     end
 end
 
+function Base.rand(rng::AbstractRNG, ::Type{T}, d::PowerMeasure) where {T}
+    map(CartesianIndices(d.axes)) do _
+        rand(rng, d.parent)
+    end
+end
 
 @inline function powermeasure(x::T, sz::Tuple{Vararg{<:Any,N}}) where {T, N}
     a = axes(Fill{T, N}(x, sz))

src/combinators/product.jl

@@ -25,11 +25,22 @@ Base.size(μ::AbstractProductMeasure) = size(marginals(μ))
 basemeasure(d::AbstractProductMeasure) = productmeasure(map(basemeasure, marginals(d)))
 
 function Base.rand(rng::AbstractRNG, ::Type{T}, d::AbstractProductMeasure) where {T}
-    map(marginals(d)) do dⱼ
+    mar = marginals(d)
+    _rand_product(rng, T, mar, eltype(mar))
+end
+
+function _rand_product(rng::AbstractRNG, ::Type{T}, mar, ::Type{M}) where {T,M<:AbstractMeasure}
+    map(mar) do dⱼ
         rand(rng, T, dⱼ)
     end
 end
 
+function _rand_product(rng::AbstractRNG, ::Type{T}, mar, ::Type{M}) where {T,M}
+    map(mar) do dⱼ
+        rand(rng, dⱼ)
+    end
+end
+
 @inline function logdensity_def(d::AbstractProductMeasure, x)
     mapreduce(logdensity_def, +, marginals(d), x)
 end