Update tests/test_save_load.py

Co-authored-by: Copilot <[email protected]>
lnccbrown · Dec 29, 2024 · 30b2ef0 · 30b2ef0
1 parent cb93184
commit 30b2ef0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_save_load.py b/tests/test_save_load.py
@@ -56,7 +56,7 @@ def test_save_load_vi_mcmc(basic_hssm_model):
 
     # 2
     # Save whole model after running vi as well
-    print("Saving model and vi traces")
+    basic_hssm_model.vi(method="advi", niter=1000)
     basic_hssm_model.vi(method="advi", niter=1000)
     basic_hssm_model.save_model(model_name=tmp_model_name, base_path=tmp_folder_name)