code: def naive_bucket_assignment(
      costs: np.ndarray, num_buckets: int
  ) -> np.ndarray:
      ordered_indexes = np.argsort(costs)
      assignment = ordered_indexes % num_buckets
      return assignment
  
imports: import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: naive_bucket_assignment(costs, num_buckets) and returns: assignment