From fa86ac7b230c529c947b4653d09712420dc98195 Mon Sep 17 00:00:00 2001 From: Predrag Gruevski Date: Thu, 14 Jan 2021 00:42:21 -0500 Subject: [PATCH] Fix type mismatches in traceplot.py. --- arviz/plots/traceplot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/arviz/plots/traceplot.py b/arviz/plots/traceplot.py index 8aee7b5bfa..8262abafee 100644 --- a/arviz/plots/traceplot.py +++ b/arviz/plots/traceplot.py @@ -20,7 +20,7 @@ def plot_trace( rug: bool = False, lines: Optional[List[Tuple[str, CoordSpec, Any]]] = None, circ_var_names: Optional[List[str]] = None, - circ_var_units: bool = "radians", + circ_var_units: str = "radians", compact: bool = True, compact_prop: Optional[Union[str, Mapping[str, Any]]] = None, combined: bool = False, @@ -160,7 +160,7 @@ def plot_trace( try: divergence_data = convert_to_dataset(data, group="sample_stats").diverging except (ValueError, AttributeError): # No sample_stats, or no `.diverging` - divergences = False + divergences = None if coords is None: coords = {}