fix

MurrellGroup · Dec 13, 2023 · 5293b0c · 5293b0c
1 parent 58ac5b5
commit 5293b0c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/Project.toml b/Project.toml
@@ -5,7 +5,9 @@ version = "1.0.0-DEV"
 
 [deps]
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
+LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
+Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 
 [compat]
 julia = "1"

diff --git a/src/ConditionalDensityLayers.jl b/src/ConditionalDensityLayers.jl
@@ -4,5 +4,6 @@ module ConditionalDensityLayers
 include("types.jl")
 include("LogDensityLayers.jl")
 include("GaussianMixtureLayers.jl")
+include("VonMisesMixtureLayers.jl")
 
 end
diff --git a/src/VonMisesMixtureLayers.jl b/src/VonMisesMixtureLayers.jl
@@ -8,6 +8,7 @@ using Distributions
 import StatsBase
 
 export VMMLayer
+export VonMisesNucleusSample
 
 struct VMMLayer 
     central_network::Chain