Added extra branch to main autoencoder for rule_based prediction

tests/test_train_utils.py

@@ -82,7 +82,7 @@ def test_get_callbacks(
     )
 
 
-@settings(max_examples=1, deadline=None, suppress_health_check=[HealthCheck.too_slow])
+@settings(max_examples=10, deadline=None, suppress_health_check=[HealthCheck.too_slow])
 @given(
     loss=st.one_of(st.just("ELBO"), st.just("MMD"), st.just("ELBO+MMD")),
     next_sequence_prediction=st.one_of(st.just(0.0), st.just(1.0)),