From 78d9b0a81c8fbaeea0695af6d15ccd1cee8c6204 Mon Sep 17 00:00:00 2001 From: gPS Team Date: Thu, 19 Sep 2024 02:53:16 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 676336378 Change-Id: I1befc7ab0986d12b7db937fda6b795124e90aeb6 --- lightweight_mmm/plot.py | 1 + 1 file changed, 1 insertion(+) diff --git a/lightweight_mmm/plot.py b/lightweight_mmm/plot.py index cd1c74f..8d9a101 100644 --- a/lightweight_mmm/plot.py +++ b/lightweight_mmm/plot.py @@ -805,6 +805,7 @@ def plot_media_channel_posteriors( channel_names = np.arange(np.shape(media_channel_posteriors)[1]) fig, axes = plt.subplots( nrows=n_media_channels, ncols=n_geos, figsize=fig_size) + media_channel_posteriors = np.asarray(media_channel_posteriors) for channel_i, channel_axis in enumerate(axes): if isinstance(channel_axis, np.ndarray): for geo_i, geo_axis in enumerate(channel_axis):