code: def _get_inaturalist_dataset(
      dataset_root, super_categories, download
  ):
      if dataset_root is None:
          dataset_root = default_dataset_location("inatuarlist2018")
  
      train_set = INATURALIST2018(
          str(dataset_root),
          split="train",
          supcats=super_categories,
          download=download,
      )
      test_set = INATURALIST2018(
          str(dataset_root),
          split="val",
          supcats=super_categories,
          download=download,
      )
  
      return train_set, test_set
  
file_path: avalanche/avalanche/benchmarks/classic/cinaturalist.py
imports: from pathlib import Path
  from typing import Union, Any, Optional
  from avalanche.benchmarks.classic.classic_benchmarks_utils import check_vision_benchmark
  from avalanche.benchmarks.datasets import INATURALIST2018, default_dataset_location
  from avalanche.benchmarks import nc_benchmark
  from torchvision import transforms
label_desc: Use image node_id 2 for calling a global function with example usage: _get_inaturalist_dataset(dataset_root, super_categories, download) and returns: train_set,  test_set