code: def _get_embeddings_file_names(
      shard_idx: int,
  ) -> Tuple[str, str, str]:
      return (
          f"img_emb_{shard_idx}.npy",
          f"text_emb_{shard_idx}.npy",
          f"metadata_{shard_idx}.parquet",
      )
  
file_path: datasets/tensorflow_datasets/vision_language/laion400m/laion400m.py
imports: import functools
  from typing import Dict, Tuple
  from etils import epath
  import numpy
  from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow
  import tensorflow_datasets.public_api
label_desc: Use image node_id 2 for calling a global function with example usage: _get_embeddings_file_names(shard_idx) and returns: str, shard_idx, str,  str, shard_idx, str,  str, shard_idx, str