code: def test_read_parquet_images_tf_dataset(self):
      temp_dir = tempfile.mkdtemp()
  
      try:
          ParquetDataset.write(
              "file://" + temp_dir,
              images_generator(),
              images_schema,
              block_size=4,
          )
          path = "file://" + temp_dir
          output_types = {
              "id": tf.string,
              "image": tf.string,
              "label": tf.float32,
          }
          dataset = read_parquet(
              "tf_dataset", path=path, output_types=output_types
          )
          for dt in dataset.take(1):
              print(dt.keys())
  
          num_shards, rank = 3, 1
          dataset_shard = read_parquet(
              "tf_dataset",
              path=path,
              config={"num_shards": num_shards, "rank": rank},
              output_types=output_types,
          )
          assert (
              len(list(dataset_shard))
              <= len(list(dataset)) // num_shards
          ), "len of dataset_shard should be 1/`num_shards` of the whole dataset."
  
          dataloader = read_parquet("dataloader", path=path)
          dataloader_shard = read_parquet(
              "dataloader",
              path=path,
              config={"num_shards": num_shards, "rank": rank},
          )
          cur_dl = iter(dataloader_shard)
          cur_count = 0
          while True:
              try:
                  print(next(cur_dl)["label"])
                  cur_count += 1
              except StopIteration:
                  break
          assert cur_count == len(list(dataset_shard))
      finally:
          shutil.rmtree(temp_dir)
  
file_path: analytics-zoo/pyzoo/test/zoo/orca/data/test_read_parquet_images.py
imports: import tempfile
  import shutil
  import pytest
  from unittest import TestCase
  import os
  from zoo.orca.data.image.parquet_dataset import ParquetDataset, read_parquet
  from zoo.orca.data.image.utils import DType, FeatureType, SchemaField
  import tensorflow
  from zoo.ray import RayContext
label_desc: Use image node_id 1 for calling the TestReadParquet obj's underlying member method code with example usage: obj.test_read_parquet_images_tf_dataset() without return types