style(plot_counter): simplify

GalacticDynamics · Jan 28, 2024 · 6db3799 · 6db3799
1 parent b7d3d5e
commit 6db3799
Showing 1 changed file with 10 additions and 13 deletions.
diff --git a/src/galax/extensions/matplotlib/_mpl_mixin.py b/src/galax/extensions/matplotlib/_mpl_mixin.py
@@ -145,22 +145,19 @@ def plot_contours(
         )
         raise ValueError(msg)
 
+    kwargs = {
+        "ax": ax,
+        "grids": _grids,
+        "slices": _slices,
+        "labels": labels,
+        "kwargs": kwargs,
+    }
+
     if ndim == 1:
-        _plot_countours_1d(
-            pot, t, ax=ax, grids=_grids, slices=_slices, labels=labels, kwargs=kwargs
-        )
+        _plot_countours_1d(pot, t, **kwargs)
 
     else:
-        _plot_countours_2d(
-            pot,
-            t,
-            ax=ax,
-            grids=_grids,
-            slices=_slices,
-            labels=labels,
-            filled=filled,
-            kwargs=kwargs,
-        )
+        _plot_countours_2d(pot, t, **kwargs, filled=filled)
 
     return fig