No public description

PiperOrigin-RevId: 676336378 Change-Id: I1befc7ab0986d12b7db937fda6b795124e90aeb6
google · Sep 19, 2024 · 78d9b0a · 78d9b0a
1 parent 4406aaa
commit 78d9b0a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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):