class_docstr: Wrapper around SSKernel parameterizations.
  
  The SSKernel is expected to support the interface
  forward()
  default_state()
  _setup_step()
  step()
class_docstr_tok: ["Wrapper","around","SSKernel","parameterizations",".","The","SSKernel","is","expected","to","support","the","interface","forward","(",")","default_state","(",")","_setup_step","(",")","step","(",")"]
code: def _setup_step(self, **kwargs):
      # This method is intended to be private so that setting up an S4 module with
      # ```
      # if hasattr(module, 'setup_step'): module.setup_step()
      # ```
      # will not trigger this method multiple times
      self.kernel._setup_step(**kwargs)
  
imports: import logging
  import math
  import os
  from functools import wraps
  from typing import Any, Callable, Optional
  import numpy
  import opt_einsum
  import torch
  import torch.nn
  import torch.nn.functional
  from einops import rearrange, repeat
  from espnet2.asr.state_spaces.components import Activation, DropoutNd, LinearActivation
label_desc: Use image node_id 4 for calling the SSKernel obj's underlying member method code with example usage: obj._setup_step() without return types