code: def expand(self, batch_shape, _instance=None):
      new = self._get_checked_instance(
          _PyroWrapper_CategoricalLogits, _instance
      )
      return super().expand(batch_shape, _instance=new)
  
imports: import functools
  import numbers
  from typing import Tuple, Union
  import pyro.distributions
  import pyro.distributions.testing.fakes
  import torch
  from pyro.distributions.torch_distribution import ExpandedDistribution, MaskedDistribution
  import funsor.ops
  from funsor.cnf import Contraction
  from funsor.constant import Constant
  from funsor.distribution import FUNSOR_DIST_NAMES, Bernoulli, LogNormal, backenddist_to_funsor, eager_beta, eager_beta_bernoulli, eager_binomial, eager_categorical_funsor, eager_categorical_tensor, eager_delta_funsor_funsor, eager_delta_funsor_variable, eager_delta_tensor, eager_delta_variable_variable, eager_dirichlet_categorical, eager_dirichlet_multinomial, eager_dirichlet_posterior, eager_gamma_gamma, eager_gamma_poisson, eager_multinomial, eager_mvn, eager_normal, eager_plate_multinomial, expandeddist_to_funsor, indepdist_to_funsor, make_dist, maskeddist_to_funsor, transformeddist_to_funsor
  from funsor.domains import Real, Reals
  from funsor.interpretations import eager
  from funsor.tensor import Tensor
  from funsor.terms import Binary, Funsor, Reduce, Unary, Variable, to_data, to_funsor
  from funsor.util import methodof
label_desc: Use image node_id 2 for calling the _PyroWrapper_CategoricalLogits obj's underlying member method code with example usage: obj.expand(batch_shape, _instance) and returns: super