Readding the Augmented Likelihood macro

theogf · Jan 13, 2021 · 6c50719 · 6c50719
1 parent 33e3ce6
commit 6c50719
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/AugmentedGaussianProcesses.jl b/src/AugmentedGaussianProcesses.jl
@@ -19,6 +19,7 @@ module AugmentedGaussianProcesses
     export predict_f, predict_y, proba_y
     export fstar, ELBO
     export covariance, diag_covariance, prior_mean
+    export @augmodel
 
     #General modules
     using Reexport
@@ -89,6 +90,8 @@ module AugmentedGaussianProcesses
     include(joinpath("inference", "inference.jl"))
     include(joinpath("likelihood", "likelihood.jl"))
 
+    include(joinpath("likelihood", "generic_likelihood.jl"))
+
     include(joinpath("functions", "KLdivergences.jl"))
     include(joinpath("data", "utils.jl"))
     include(joinpath("functions", "plotting.jl"))