class_docstr: Base class for wrapping Torch probability distributions.
  
  :param dist: The torch probability distribution.
  :param action_space: The gym action space.
class_docstr_tok: ["Base","class","for","wrapping","Torch","probability","distributions",".",":","param","dist",":","The","torch","probability","distribution",".",":","param","action_space",":","The","gym","action","space","."]
code: def __init__(self, dist: T, action_space: spaces.Space):
      self.dist = dist
      self.action_space = action_space
  
file_path: maze/maze/distributions/torch_dist.py
imports: from abc import ABC, abstractmethod
  from typing import Sequence, Generic, TypeVar
  import torch
  import torch.distributions
  from gym import spaces
  from maze.core.annotations import override
  from maze.distributions.distribution import ProbabilityDistribution
label_desc: Use image node_id 2 to create a new TorchProbabilityDistribution object from inherited base classes: ProbabilityDistribution, Generic, ABC with example: obj = TorchProbabilityDistribution(dist, action_space)