&`i^ddlZddlmZeddejjfdZdS)N) PublicAPIbeta) stabilitytf_dataset_shardctj}tjjjj|j_|| tjj S)aA utility function that overrides default config for Tensorflow Dataset. This should be used on a TensorFlow ``Dataset`` created by calling ``iter_tf_batches()`` on a ``ray.data.Dataset`` returned by ``ray.train.get_dataset_shard()`` since the dataset has already been sharded across the workers. Args: tf_dataset_shard (tf.data.Dataset): A TensorFlow Dataset. Returns: A TensorFlow Dataset with: - autosharding turned off - prefetching turned on with autotune enabled ) tfdataOptions experimentalAutoShardPolicyOFFexperimental_distributeauto_shard_policy with_optionsprefetchAUTOTUNE)roptionss y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/train/tensorflow/train_loop_utils.pyprepare_dataset_shardrsW"gooG ,0 #5  ( ( 1 1 : :27;K L LL) tensorflowrray.util.annotationsrr Datasetrrrrsm****** VMBGOMMMMMMr