fix selftests error

GrainLearning · Dec 9, 2023 · eaec558 · eaec558
1 parent 6edc2a9
commit eaec558
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/tests/integration/test_gmm.py b/tests/integration/test_gmm.py
@@ -27,7 +27,9 @@ def test_gmm():
                 "param_names": ['a', 'b'],
             },
             "calibration": {
-                "inference": {"ess_target": 0.3},
+                "inference": {
+                    "ess_target": 0.3,
+                    "scale_cov_with_max": True},
                 "sampling": {
                     "max_num_components": 1,
                     "covariance_type": "full",

diff --git a/tests/integration/test_smc.py b/tests/integration/test_smc.py
@@ -24,7 +24,10 @@ def test_smc():
                 "param_names": ['a', 'b'],
             },
             "calibration": {
-                "inference": {"ess_target": 0.3},
+                "inference": {
+                    "ess_target": 0.3,
+                    "scale_cov_with_max": True
+                },
                 "sampling": {"max_num_components": 1},
             },
         }

diff --git a/tests/integration/test_smc_mse.py b/tests/integration/test_smc_mse.py
@@ -33,7 +33,10 @@ def test_smc_mse():
                 "callback": run_sim,
             },
             "calibration": {
-                "inference": {"ess_target": 0.3},
+                "inference": {
+                    "ess_target": 0.3,
+                    "scale_cov_with_max": True
+                },
                 "sampling": {"max_num_components": 1},
             }
         }