code: def _get_uniform_sharding_options(
      sharding_options: List[ShardingOption],
  ) -> List[ShardingOption]:
      uniform_sharding_options: List[ShardingOption] = []
      for sharding_option in sharding_options:
          if (
              sharding_option.partition_by
              == PartitionByType.UNIFORM.value
          ):
              uniform_sharding_options.append(sharding_option)
      return uniform_sharding_options
  
imports: import copy
  import heapq
  import logging
  from dataclasses import dataclass
  from enum import Enum
  from typing import cast, Dict, List, Optional
  from torchrec.distributed.planner.perf_models import NoopPerfModel
  from torchrec.distributed.planner.types import DeviceHardware, PartitionByType, Partitioner, Perf, PerfModel, PlannerError, PlannerErrorType, ShardingOption, Storage, Topology
  from torchrec.distributed.planner.utils import bytes_to_gb, reset_shard_rank
  from torchrec.distributed.types import ShardingType
label_desc: Use image node_id 2 for calling a global function with example usage: _get_uniform_sharding_options(sharding_options) and returns: uniform_sharding_options