class_docstr: Wrapper class for torch.distributions.
class_docstr_tok: ["Wrapper","class","for","torch.distributions","."]
docstr: Returns the torch.distributions.Distribution object to use.
docstr_tok: ["Returns","the","torch.distributions.Distribution","object","to","use","."]
code: def _get_torch_distribution(
      self, *args, **kwargs
  ) -> "torch.distributions.Distribution":
      """Returns the torch.distributions.Distribution object to use."""
  
imports: import gymnasium
  import numpy
  from typing import Optional, List, Mapping, Iterable, Dict
  import tree
  import abc
  from ray.rllib.models.distributions import Distribution
  from ray.rllib.utils.annotations import override, DeveloperAPI
  from ray.rllib.utils.framework import try_import_torch
  from ray.rllib.utils.typing import TensorType, Union, Tuple
label_desc: Use image node_id 2 for calling the TorchDistribution obj's underlying member method code with example usage: obj._get_torch_distribution() without return types