Complete migration from arraydist/filldist to product_distribution

.gitignore

@@ -1,2 +1,3 @@
 Manifest.toml
 *.swp
+docs/build/

ext/DistributionsADLazyArraysExt.jl

@@ -3,12 +3,12 @@ module DistributionsADLazyArraysExt
 if isdefined(Base, :get_extension)
     using DistributionsAD
     using LazyArrays
-    using DistributionsAD: Distributions, ValueSupport, UnivariateDistribution, VectorOfUnivariate, MatrixOfUnivariate
+    using DistributionsAD: Distributions, ValueSupport, UnivariateDistribution, VectorOfUnivariate
     using LazyArrays: BroadcastArray, BroadcastVector, LazyArray
 else
     using ..DistributionsAD
     using ..LazyArrays
-    using ..DistributionsAD: Distributions, ValueSupport, UnivariateDistribution, VectorOfUnivariate, MatrixOfUnivariate
+    using ..DistributionsAD: Distributions, ValueSupport, UnivariateDistribution, VectorOfUnivariate
     using ..LazyArrays: BroadcastArray, BroadcastVector, LazyArray
 end
 
@@ -34,19 +34,6 @@ function Distributions.logpdf(
     return vec(sum(copy(Distributions.logpdf.(dists, x)), dims = 1))
 end
 
-const LazyMatrixOfUnivariate{
-    S<:ValueSupport,
-    T<:UnivariateDistribution{S},
-    Tdists<:BroadcastArray{T,2},
-} = MatrixOfUnivariate{S,T,Tdists}
-
-function Distributions._logpdf(
-    dist::LazyMatrixOfUnivariate,
-    x::AbstractMatrix{<:Real},
-)
-    return sum(copy(Distributions.logpdf.(dist.dists, x)))
-end
-
 DistributionsAD.lazyarray(f, x...) = LazyArray(Base.broadcasted(f, x...))
 
 end # module

src/DistributionsAD.jl

@@ -19,13 +19,10 @@ export TuringScalMvNormal,
        TuringMvLogNormal,
        TuringPoissonBinomial,
        TuringWishart,
-       TuringInverseWishart,
-       arraydist,
-       filldist
+       TuringInverseWishart
 
 include("common.jl")
-include("arraydist.jl")
-include("filldist.jl")
+include("product_distribution_compat.jl")
 include("univariate.jl")
 include("multivariate.jl")
 include("matrixvariate.jl")

src/product_distribution_compat.jl

@@ -0,0 +1,4 @@
+# Compatibility aliases for product distributions
+# These are maintained for compatibility with extensions
+
+const VectorOfUnivariate = Distributions.Product

src/zygote.jl

@@ -6,14 +6,6 @@ ZygoteRules.@adjoint function Distributions._logpdf(d::Product, x::AbstractVecto
         sum(map(logpdf, d.v, x))
     end
 end
-ZygoteRules.@adjoint function Distributions._logpdf(
-    d::FillVectorOfUnivariate,
-    x::AbstractVector{<:Real},
-)
-    return ZygoteRules.pullback(d, x) do d, x
-        _flat_logpdf(d.v.value, x)
-    end
-end
 
 # Loglikelihood of multi- and matrixvariate distributions: multiple samples
 # workaround for Zygote issues discussed in