docstr: :rtype: torch.distributions.Distribution
docstr_tok: [":","rtype",":","torch.distributions.Distribution"]
code: def condition(self, context):
      """:rtype: torch.distributions.Distribution"""
      raise NotImplementedError
  
imports: from abc import ABC, abstractmethod
  import torch
  import torch.nn
  from .torch import TransformedDistribution
  from .torch_transform import ComposeTransformModule
label_desc: Use image node_id 1 for calling the ConditionalDistribution obj's underlying member method code with example usage: obj.condition(context) without return types