Apply suggestions from code review

Co-authored-by: David Widmann <[email protected]>
TuringLang · Apr 15, 2024 · 7f86afc · 7f86afc
1 parent e4b7835
commit 7f86afc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/arraydist.jl b/src/arraydist.jl
@@ -3,7 +3,7 @@
 
 Create a distribution from an array of distributions.
 """
-arraydist(dists::AbstractArray) = product_distribution(dists)
+arraydist(dists::AbstractArray{<:Distribution}) = product_distribution(dists)
 
 # Univariate
 

diff --git a/src/filldist.jl b/src/filldist.jl
@@ -4,7 +4,7 @@
 
 Create a product distribution using `FillArrays.Fill` as the array type.
 """
-filldist(d::Distribution, ns...) = product_distribution(Fill(d, ns...))
+filldist(d::Distribution, n1::Int, ns::Int...) = product_distribution(Fill(d, n1, ns...))
 
 # Univariate