code: def shard_tests(
      test_targets: List[str],
      shard_count: int,
      shard_id: int,
  ) -> List[str]:
      """
      Shard tests into N shards and return the shard corresponding to shard_id
      """
      return bazel_sharding.main(
          test_targets, index=shard_id, count=shard_count
      )
  
file_path: ray/ci/ray_ci/utils.py
imports: import base64
  import logging
  import subprocess
  import sys
  import tempfile
  import boto3
  from typing import List
  from math import ceil
  import ci.ray_ci.bazel_sharding
label_desc: Use image node_id 2 for calling a global function with example usage: shard_tests(test_targets, shard_count, shard_id) and returns: bazel_sharding