numpyro icon indicating copy to clipboard operation
numpyro copied to clipboard

Masking a Mixture Model makes `component_` methods unavailable

Open nstarman opened this issue 1 year ago • 5 comments

The MaskedDistribution object wraps models, but does not necessarily expose their public API. I'm raising this Issue to perhaps add support for the components_ methods.

Possible Options:

  1. Some clever method of auto-wrapping methods accessed by __getattr__.
  2. Have a MaskedMixtureDistribution class that is returned by MixtureDistribution.mask().
  3. instead of .mask() masking the Distribution, it returns the same mixture, but calling .mask() on all the component distributions.

nstarman avatar Oct 10 '24 14:10 nstarman