made network for test a bit larger

cbg-ethz · Feb 29, 2024 · 32a227e · 32a227e
1 parent b850b69
commit 32a227e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/estimators/neural/test_mine.py b/tests/estimators/neural/test_mine.py
@@ -37,7 +37,7 @@ def test_mine_estimator_2d(n_points: int = 8_000, correlation: float = 0.7) -> N
     points_x, points_y = distribution.sample(n_points, rng=19)
 
     estimator = mine.MINEEstimator(
-        hidden_layers=(8, 5),
+        hidden_layers=(10, 5),
         verbose=False,
     )