Listify conditional values so they can be sorted

bambinos · Oct 3, 2023 · 8d7e161 · 8d7e161
1 parent b074ba1
commit 8d7e161
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bambi/interpret/plotting.py b/bambi/interpret/plotting.py
@@ -316,7 +316,7 @@ def plot_comparisons(
         raise ValueError("Must specify at least one of 'conditional' or 'average_by'.")
 
     if isinstance(conditional, dict):
-        conditional = {key: sorted(value) for key, value in conditional.items()}
+        conditional = {key: sorted(listify(value)) for key, value in conditional.items()}
     elif conditional is not None:
         conditional = listify(conditional)
         if len(conditional) > 3 and average_by is None:
@@ -474,7 +474,7 @@ def plot_slopes(
         raise ValueError("Must specify at least one of 'conditional' or 'average_by'.")
 
     if isinstance(conditional, dict):
-        conditional = {key: sorted(value) for key, value in conditional.items()}
+        conditional = {key: sorted(listify(value)) for key, value in conditional.items()}
     elif conditional is not None:
         conditional = listify(conditional)
         if len(conditional) > 3 and average_by is None: