code: def _select_cols(a_dict, keys):
      """Filters out entries in a dictionary that have a key which is not part of 'keys' argument. `a_dict` is not
      modified and a new dictionary is returned."""
      if keys == list(a_dict.keys()):
          return a_dict
      else:
          return {field_name: a_dict[field_name] for field_name in keys}
  
imports: from __future__ import division
  import hashlib
  import threading
  from collections.abc import Iterable
  import numpy
  from pyarrow import parquet
  from pyarrow.parquet import ParquetFile
  from petastorm import utils
  from petastorm.cache import NullCache
  from petastorm.workers_pool import EmptyResultError
  from petastorm.workers_pool.worker_base import WorkerBase
label_desc: Use image node_id 3 for calling a global function with example usage: _select_cols(a_dict, keys) and returns: a_dict, unknown