code: def subset_dict(
      original_dict: Dict[str, Any], keys_subset: Sequence[str]
  ) -> Dict[str, Any]:
      """Create a subset of a dictionary using a subset of keys.
  
      :param original_dict: The original dictionary.
      :param keys_subset: The subset of keys to extract.
      :return: A dictionary containing only the specified keys.
      """
      return {
          key: original_dict[key]
          for key in keys_subset
          if key in original_dict
      }
  
file_path: wandb/wandb/integration/cohere/resolver.py
imports: import logging
  from datetime import datetime
  from typing import Any, Dict, List, Optional, Sequence, Tuple
  import wandb
  from wandb.sdk.integration_utils.auto_logging import Response
  from wandb.sdk.lib.runid import generate_id
label_desc: Use image node_id 1 for calling a global function with example usage: subset_dict(original_dict, keys_subset) and returns: unknown