code: def get_chunk_size(num_elements: int, min_chunk_size: int) -> int:
      """Adjust chunk_size to minimize imbalance between chunk sizes"""
      if min_chunk_size >= num_elements:
          return min_chunk_size
      leftover_elements = num_elements % min_chunk_size
      num_chunks = num_elements // min_chunk_size
      return min_chunk_size + (leftover_elements - 1) // num_chunks + 1
  
imports: import math
  import os
  import warnings
  from abc import ABC, abstractmethod
  from concurrent.futures import ThreadPoolExecutor
  from typing import Tuple
  import numpy
  import torch
  from hivemind.compression.base import CompressionBase, CompressionInfo
  from hivemind.proto import runtime_pb2
label_desc: Use image node_id 2 for calling a global function with example usage: get_chunk_size(num_elements, min_chunk_size) and returns: unknown, min_chunk_size