Fix MaskedCategorical warning: "Distribution subclass MaskedCategorical inherits `_parameter_properties from its parent (Categorical)"
Fix MaskedCategorical warning:
"Distribution subclass MaskedCategorical inherits `_parameter_properties from its parent (Categorical)"

PiperOrigin-RevId: 415526548
Change-Id: I665eeef79e6da497c91fd12dfbb29b209740eaec

tensorflow/agents,tensorflow/agents