Commit 8c2a744a authored by lucas_miranda's avatar lucas_miranda
Browse files

Fixed issue with gmvaep when returning both prior and posterior

parent 56bc7c7c
Pipeline #98190 canceled with stages
in 12 minutes and 14 seconds
......@@ -188,7 +188,7 @@ class SEQ_2_SEQ_GMVAE(HyperModel):
return gmvaep
......@@ -137,7 +137,7 @@ def test_autoencoder_fitting(
batch_size=st.integers(min_value=128, max_value=512),
encoding_size=st.integers(min_value=1, max_value=16),
hpt_type=st.one_of(st.just("bayopt"), st.just("hypermodel")),
hpt_type=st.one_of(st.just("bayopt"), st.just("hyperband")),
k=st.integers(min_value=1, max_value=10),
loss=st.one_of(st.just("ELBO"), st.just("MMD")),
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment