Correctly expose sub-distribution parameters in distributions created by

inflated_factory.

PiperOrigin-RevId: 471879189

Author: Googler

tensorflow_probability/python/distributions/inflated.py

@@ -224,12 +224,16 @@ def inflated_factory(default_name, distribution_class, inflated_loc,
   def my_init(self,
               inflated_loc_logits=None, inflated_loc_probs=None,
               name=default_name, **kwargs):
+    parameters = dict(locals())
     if 'distribution' in kwargs:
       dist = kwargs['distribution']
     else:
       dist = distribution_class(**{**kwargs, **more_kwargs})
     Inflated.__init__(self, dist, inflated_loc_logits, inflated_loc_probs,
                       inflated_loc, name=name)
+    # pylint: disable=protected-access
+    self._parameters = {**parameters, **more_kwargs}
+    # pylint: enable=protected-access
 
   def my_parameter_properties(unused_cls, dtype, num_classes=None):
     return dict(