ray.rllib.core.models.catalog.Catalog.get_action_dist_cls#

Catalog.get_action_dist_cls(framework: str)[source]#

Get the action distribution class.

The default behavior is to get the action distribution from the Catalog._action_dist_class_fn.

You should override this to have RLlib build your custom action distribution instead of the default one. For example, if you don’t want to use RLlib’s default RLModules with their default models, but only want to change the distribution that Catalog returns.

Parameters:

framework – The framework to use. Either “torch” or “tf2”.

Returns:

The action distribution.