Bug fix for labels in corner plot (#136)

This fixed #136
undark-lab · Jul 28, 2023 · bd1a71b · bd1a71b
1 parent 5ec9cd7
commit bd1a71b
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/swyft/plot/plot2.py b/swyft/plot/plot2.py
@@ -353,7 +353,6 @@ def corner(
     label_args={},
     contours_1d: bool = True,
     fig=None,
-    labeler=None,
     smooth=0.0,
 ) -> None:
     """Make a beautiful corner plot.
@@ -384,10 +383,7 @@ def corner(
 
     diagnostics = {}
 
-    if labeler is not None:
-        labels = [labeler.get(k, k) for k in parnames]
-    else:
-        labels = parnames
+    labels = labels if labels is not None else parnames
 
     for i in range(K):
         for j in range(K):