TypingError in plot_media_channel_posteriors

[ngockhueanhvan]

Hi everyone,

I encountered a TypingErorr when trying to run the below line of code in the demo workbook locally from Jupyter Notebook (it's fine running in Google Colab though).

plot.plot_media_channel_posteriors(media_mix_model=mmm)

The full error can be found below. Did anyone have the same issue and how should we fix it?
Thank you all!

---------------------------------------------------------------------------
TypingError                               Traceback (most recent call last)
Cell In[20], line 1
----> 1 plot.plot_media_channel_posteriors(media_mix_model=mmm)

File ~/anaconda3/envs/pytorch_p310/lib/python3.10/site-packages/lightweight_mmm/plot.py:818, in plot_media_channel_posteriors(media_mix_model, channel_names, quantiles, fig_size)
    816     geo_axis.set_xlabel(axis_label)
    817 else:
--> 818   channel_axis = arviz.plot_kde(
    819       media_channel_posteriors[:, channel_i],
    820       quantiles=quantiles,
    821       ax=channel_axis)
    822   axis_label = f"media channel {channel_names[channel_i]}"
    823   channel_axis.set_xlabel(axis_label)

File ~/anaconda3/envs/pytorch_p310/lib/python3.10/site-packages/arviz/plots/kdeplot.py:263, in plot_kde(values, values2, cumulative, rug, label, bw, adaptive, quantiles, rotated, contour, hdi_probs, fill_last, figsize, textsize, plot_kwargs, fill_kwargs, rug_kwargs, contour_kwargs, contourf_kwargs, pcolormesh_kwargs, is_circular, ax, legend, backend, backend_kwargs, show, return_glyph, **kwargs)
    260 if bw == "default":
    261     bw = "taylor" if is_circular else "experimental"
--> 263 grid, density = kde(values, is_circular, bw=bw, adaptive=adaptive, cumulative=cumulative)
    264 lower, upper = grid[0], grid[-1]
    266 density_q = density if cumulative else density.cumsum() / density.sum()

File ~/anaconda3/envs/pytorch_p310/lib/python3.10/site-packages/arviz/stats/density_utils.py:498, in kde(x, circular, **kwargs)
    495 else:
    496     kde_fun = _kde_linear
--> 498 return kde_fun(x, **kwargs)

File ~/anaconda3/envs/pytorch_p310/lib/python3.10/site-packages/arviz/stats/density_utils.py:581, in _kde_linear(x, bw, adaptive, extend, bound_correction, extend_fct, bw_fct, bw_return, custom_lims, cumulative, grid_len, **kwargs)
    577 # Determine grid
    578 grid_min, grid_max, grid_len = _get_grid(
    579     x_min, x_max, x_std, extend_fct, grid_len, custom_lims, extend, bound_correction
    580 )
--> 581 grid_counts, _, grid_edges = histogram(x, grid_len, (grid_min, grid_max))
    583 # Bandwidth estimation
    584 bw = bw_fct * _get_bw(x, bw, grid_counts, x_std, x_range)

File ~/anaconda3/envs/pytorch_p310/lib/python3.10/site-packages/arviz/utils.py:193, in maybe_numba_fn.__call__(self, *args, **kwargs)
    191 """Call the jitted function or normal, depending on flag."""
    192 if Numba.numba_flag:
--> 193     return self.numba_fn(*args, **kwargs)
    194 else:
    195     return self.function(*args, **kwargs)

File ~/anaconda3/envs/pytorch_p310/lib/python3.10/site-packages/numba/core/dispatcher.py:468, in _DispatcherBase._compile_for_args(self, *args, **kws)
    464         msg = (f"{str(e).rstrip()} \n\nThis error may have been caused "
    465                f"by the following argument(s):\n{args_str}\n")
    466         e.patch_message(msg)
--> 468     error_rewrite(e, 'typing')
    469 except errors.UnsupportedError as e:
    470     # Something unsupported is present in the user code, add help info
    471     error_rewrite(e, 'unsupported_error')

File ~/anaconda3/envs/pytorch_p310/lib/python3.10/site-packages/numba/core/dispatcher.py:409, in _DispatcherBase._compile_for_args.<locals>.error_rewrite(e, issue_type)
    407     raise e
    408 else:
--> 409     raise e.with_traceback(None)

TypingError: Failed in nopython mode pipeline (step: nopython frontend)
No implementation of function Function(<function histogram at 0x7f706427a4d0>) found for signature:
 
 >>> histogram(readonly buffer(float32, 1d, C), bins=int64, range=UniTuple(readonly buffer(float32, 0d, C) x 2))
 
There are 2 candidate implementations:
  - Of which 2 did not match due to:
  Overload in function 'np_histogram': File: numba/np/arraymath.py: Line 3895.
    With argument(s): '(readonly buffer(float32, 1d, C), bins=int64, range=UniTuple(readonly buffer(float32, 0d, C) x 2))':
   Rejected as the implementation raised a specific error:
     TypingError: Failed in nopython mode pipeline (step: nopython frontend)
   No implementation of function Function(<function linspace at 0x7f70643da440>) found for signature:
    
    >>> linspace(readonly buffer(float32, 0d, C), readonly buffer(float32, 0d, C), int64)
    
   There are 2 candidate implementations:
         - Of which 2 did not match due to:
         Overload of function 'linspace': File: numba/np/arrayobj.py: Line 4638.
           With argument(s): '(readonly buffer(float32, 0d, C), readonly buffer(float32, 0d, C), int64)':
          No match.
   
   During: resolving callee type: Function(<function linspace at 0x7f70643da440>)
   During: typing of call at /home/ec2-user/anaconda3/envs/pytorch_p310/lib/python3.10/site-packages/numba/np/arraymath.py (3936)
   
   
   File "../anaconda3/envs/pytorch_p310/lib/python3.10/site-packages/numba/np/arraymath.py", line 3936:
               def histogram_impl(a, bins=10, range=None):
                   <source elided>
   
                   bins_array = np.linspace(bin_min, bin_max, bins + 1)
                   ^

  raised from /home/ec2-user/anaconda3/envs/pytorch_p310/lib/python3.10/site-packages/numba/core/typeinfer.py:1086

During: resolving callee type: Function(<function histogram at 0x7f706427a4d0>)
During: typing of call at /home/ec2-user/anaconda3/envs/pytorch_p310/lib/python3.10/site-packages/arviz/stats/density_utils.py (979)


File "../anaconda3/envs/pytorch_p310/lib/python3.10/site-packages/arviz/stats/density_utils.py", line 979:
def histogram(data, bins, range_hist=None):
    <source elided>
    """
    hist, bin_edges = np.histogram(data, bins=bins, range=range_hist)
    ^