code: def get_dataset_config(seq_length):
      """Returns dataset config."""
      if seq_length == 1024:
          return ModelNet40DatasetConfig(seq_length=1024)
      elif seq_length == 2048:
          return ModelNet40DatasetConfig(
              num_train_pointclouds=9840,
              seq_length=2048,
              dataset_name="modelnet40_h5",
          )
      elif seq_length == 4096:
          return ModelNet40DatasetConfig(
              seq_length=4096, version="1.1.0"
          )
      elif seq_length == 8192:
          return ModelNet40DatasetConfig(
              seq_length=8192, version="1.2.0"
          )
      return ModelNet40DatasetConfig()
  
file_path: scenic/scenic/projects/pointcloud/pointcloud_dataset.py
imports: import dataclasses
  import functools
  from typing import Optional
  from absl import logging
  from flax import jax_utils
  import jax.numpy
  from scenic.dataset_lib import dataset_utils
  from scenic.dataset_lib import datasets
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 1 for calling a global function with example usage: get_dataset_config(seq_length) and returns: ModelNet40DatasetConfig, ModelNet40DatasetConfig, ModelNet40DatasetConfig, ModelNet40DatasetConfig, ModelNet40DatasetConfig