From 2aef975367197879fe4b41cef737d42025523463 Mon Sep 17 00:00:00 2001
From: Tor Erlend Fjelde <tor.erlend95@gmail.com>
Date: Tue, 13 Jun 2023 18:25:00 +0100
Subject: [PATCH] Actual fix for LazyArrays extension (#247)

* actual fix for lazyarrays extension

* patch version bump
---
 Project.toml                        | 2 +-
 ext/DistributionsADLazyArraysExt.jl | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/Project.toml b/Project.toml
index a6f20d6..6eae08d 100644
--- 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
index 26c8361..f47fa2c 100644
--- 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
\ No newline at end of file
+end # module