fixup

cbg-ethz · Mar 5, 2024 · 348f17f · 348f17f
1 parent 116b268
commit 348f17f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/bmi/estimators/neural/_mine_estimator.py b/src/bmi/estimators/neural/_mine_estimator.py
@@ -330,7 +330,7 @@ def estimate_with_info(self, x: ArrayLike, y: ArrayLike) -> EstimateResult:
         )
 
         if self._params.batch_size > len(xs_train):
-            if self.verbose:
+            if self._verbose:
                 print("ERROR: Batch size larger than train dataset.")
 
             return EstimateResult(

diff --git a/workflows/benchmark/_utils.py b/workflows/benchmark/_utils.py
@@ -105,7 +105,7 @@ def make_pretty(styler):
         )
         if converged_only:
             styler.set_caption(
-                "Estimates smaller than 10% of the maximal estimate are excluded."
+                "Estimates smaller than 10% of the maximal estimate are excluded. "
                 "This can help neural estimators which sometimes fail to converge."
             )
         return styler
@@ -154,7 +154,7 @@ def make_pretty(styler):
             cmap="gray",
         )
         styler.set_caption(
-            "Estimates higher than 10% of the maximal estimate are considered"
+            "Estimates higher than 10% of the maximal estimate are considered "
             "converged. This table shows the percentage of converged estimates."
         )
         return styler