Fixed bug in hyperparameter tuning

tests/test_train_utils.py

@@ -100,7 +100,9 @@ def test_tune_search(
         bayopt_trials=1,
         hypermodel=hypermodel,
         k=k,
+        kl_warmup_epochs=0,
         loss=loss,
+        mmd_warmup_epochs=0,
         overlap_loss=overlap_loss,
         pheno_class=pheno_class,
         predictor=predictor,