make MetricChain type stable. Define v0.1.2

Author: mrazomej

Project.toml

@@ -1,7 +1,7 @@
 name = "AutoEncoderToolkit"
 uuid = "1575904b-3569-468b-ae7a-8569c4d6dae9"
 authors = ["Manuel Razo-Mejia"]
-version = "0.1.1"
+version = "0.1.2"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"

src/rhvae.jl

@@ -3,6 +3,8 @@ import Flux
 
 # Import ChainRulesCore to ignore functions when computing gradients
 using ChainRulesCore: ignore_derivatives
+# Import ConcreteStructs module
+using ConcreteStructs: @concrete
 
 # Import AutoDiff backends
 import TaylorDiff
@@ -91,7 +93,7 @@ lower = Flux.Dense(10, 15)
 metric_chain = MetricChain(mlp, diag, lower)
 ```
 """
-struct MetricChain
+@concrete struct MetricChain
     mlp::Flux.Chain
     diag::Flux.Dense
     lower::Flux.Dense