code: def get_attrs_with_suffix(
      attrs: Optional[List[str]],
      store: BaseStorage,
      suffix: str,
  ) -> List[str]:
      if attrs is not None:
          return attrs
      return [key for key in store.keys() if key.endswith(suffix)]
  
file_path: pytorch_geometric/torch_geometric/transforms/mask.py
imports: from typing import List, Optional, Sequence, Union
  from torch_geometric.data import Data, HeteroData
  from torch_geometric.data.datapipes import functional_transform
  from torch_geometric.data.storage import BaseStorage
  from torch_geometric.transforms import BaseTransform
  from torch_geometric.utils import index_to_mask, mask_to_index
label_desc: Use image node_id 1 for calling a global function with example usage: get_attrs_with_suffix(attrs, store, suffix) and returns: unknown, attrs