Actual fix for LazyArrays extension (#247)

torfjelde · web-flow · commit 2aef97536719 · 2023-06-13T18:25:00.000+01:00
* actual fix for lazyarrays extension

* patch version bump
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "DistributionsAD"
 uuid = "ced4e74d-a319-5a8a-b0ac-84af2272839c"
-version = "0.6.48"
+version = "0.6.49"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
diff --git a/ext/DistributionsADLazyArraysExt.jl b/ext/DistributionsADLazyArraysExt.jl
@@ -3,12 +3,12 @@ module DistributionsADLazyArraysExt
 if isdefined(Base, :get_extension)
     using DistributionsAD
     using LazyArrays
-    using DistributionsAD: Distributions, ValueSupport
+    using DistributionsAD: Distributions, ValueSupport, UnivariateDistribution, VectorOfUnivariate, MatrixOfUnivariate
     using LazyArrays: BroadcastArray, BroadcastVector, LazyArray
 else
     using ..DistributionsAD
     using ..LazyArrays
-    using ..DistributionsAD: Distributions, ValueSupport
+    using ..DistributionsAD: Distributions, ValueSupport, UnivariateDistribution, VectorOfUnivariate, MatrixOfUnivariate
     using ..LazyArrays: BroadcastArray, BroadcastVector, LazyArray
 end
 
@@ -49,4 +49,4 @@ end
 
 DistributionsAD.lazyarray(f, x...) = LazyArray(Base.broadcasted(f, x...))
 
-end # module
+end # module
