Source code for utils.dict_utils

#
# For licensing see accompanying LICENSE file.
# Copyright (C) 2023 Apple Inc. All Rights Reserved.
#
from typing import Collection, Dict, Optional


[docs]def filter_keys( d: Dict, whitelist: Optional[Collection[str]] = None, ) -> Dict: """Returns a copy of the input dict @d, with a subset of keys that are in @whitelist. Args: d: Intput dictionary that will be copied with a subset of keys. whitelist: List of keys to keep in the output (if exist in input dict). """ return {key: d[key] for key in whitelist if key in d}