code: def add_prefix_for_sharding(
      sharded_state_dict: ShardedStateDict, prefix: str
  ):
      def add_prefix(t):
          if isinstance(t, ShardedTensor):
              t.key = f"{prefix}.{t.key}"
          return t
  
      dict_list_map_inplace(add_prefix, sharded_state_dict)
  
imports: from typing import Tuple
  from .dict_utils import dict_list_map_inplace, extract_matching_values
  from .mapping import LocalNonpersitentObject, ShardedStateDict, ShardedTensor, ShardedTensorFactory, StateDict
label_desc: Use image node_id 4 for calling a global function with example usage: add_prefix_for_sharding(sharded_state_dict, prefix) and returns: t