code: def remove_param_name_key(model_params: List) -> None:
      """Helper function to remove param_names key from model_params"""
      for param_info in model_params:
          if not isinstance(param_info, Dict):
              logger.error(
                  "Expected format is a Dict with three keys: params, weight_decay, param_names"
              )
  
          if not {"params", "weight_decay", "param_names"}.issubset(
              param_info.keys()
          ):
              logger.error(
                  "Parameter dict should have three keys: params, weight_decay, param_names"
              )
  
          param_info.pop("param_names")
  
file_path: ml-cvnets/optim/__init__.py
imports: import argparse
  from typing import Dict, List
  import torch.nn
  from optim.base_optim import BaseOptim
  from utils import logger
  from utils.common_utils import unwrap_model_fn
  from utils.registry import Registry
label_desc: Use image node_id 2 for calling a global function with example usage: remove_param_name_key(model_params) without return types