def _remove_keys(keys, keys_to_remove):
  """Remove given keys from list of keys.

  Args:
    keys: List of keys to subtract from.
    keys_to_remove: List of keys to remove.

  Returns:
    A list of keys after subtraction.
  """
  return list(set(keys) - set(keys_to_remove))