(model)
return axis_indexers
Returns a dict that maps size names to `Axis` indexers.

  Args:
    model: An instance of `mjbindings.MjModelWrapper`.

  Returns:
    A `dict` mapping from a size name (e.g. `'nbody'`) to an `Axis` instance.
Returns a dict that maps size names to `Axis` indexers.
Returns a dict that maps size names to Axis indexers .
def make_axis_indexers(model):
  """Returns a dict that maps size names to `Axis` indexers.

  Args:
    model: An instance of `mjbindings.MjModelWrapper`.

  Returns:
    A `dict` mapping from a size name (e.g. `'nbody'`) to an `Axis` instance.
  """

  size_name_to_element_names = _get_size_name_to_element_names(model)
  size_name_to_element_sizes = _get_size_name_to_element_sizes(model)

  # Unrecognized size names are treated as unnamed axes.
  axis_indexers = collections.defaultdict(UnnamedAxis)

  for size_name in size_name_to_element_names:
    element_names = size_name_to_element_names[size_name]
    if size_name in _RAGGED_ADDRS:
      element_sizes = size_name_to_element_sizes[size_name]
      indexer = RaggedNamedAxis(element_names, element_sizes)
    else:
      indexer = RegularNamedAxis(element_names)
    axis_indexers[size_name] = indexer

  return axis_indexers
def make_axis_indexers ( model ) : size_name_to_element_names = _get_size_name_to_element_names ( model ) size_name_to_element_sizes = _get_size_name_to_element_sizes ( model ) # Unrecognized size names are treated as unnamed axes. axis_indexers = collections . defaultdict ( UnnamedAxis ) for size_name in size_name_to_element_names : element_names = size_name_to_element_names [ size_name ] if size_name in _RAGGED_ADDRS : element_sizes = size_name_to_element_sizes [ size_name ] indexer = RaggedNamedAxis ( element_names , element_sizes ) else : indexer = RegularNamedAxis ( element_names ) axis_indexers [ size_name ] = indexer return axis_indexers