code: def _handle_sharding(self, dataset):
      from tensorflow.python.distribute.input_ops import (
          auto_shard_dataset,
      )
  
      dataset = auto_shard_dataset(dataset, self.size, self.rank)
      return dataset
  
imports: import logging
  import json
  import os
  import numpy
  import ray
  from contextlib import closing
  import logging
  import socket
  from zoo.orca.data.utils import ray_partitions_get_data_label
label_desc: Use image node_id 2 for calling the HorovodDatasetHanlder obj's underlying member method code with example usage: obj._handle_sharding(dataset) and returns: dataset