original_docstring: 
          Returns the index(es) of the given IDs.
          
docstring: Returns the index(es) of the given IDs.
docstring_tokens: Returns the index ( es ) of the given IDs .
code: def index(self, id_vals):
          """
          Returns the index(es) of the given IDs.
          """
          if isinstance(id_vals, (list, np.ndarray, set)):
              return [self._id_index[int(id_val)] for id_val in id_vals]
          else:
              return self._id_index[int(id_vals)]
code_tokens: def index ( self , id_vals ) : if isinstance ( id_vals , ( list , np . ndarray , set ) ) : return [ self . _id_index [ int ( id_val ) ] for id_val in id_vals ] else : return self . _id_index [ int ( id_vals ) ]
short_docstring: Returns the index(es) of the given IDs.
short_docstring_tokens: Returns the index ( es ) of the given IDs .
comment: """
          Returns the index(es) of the given IDs.
          """