class_docstr: Dataset partitioning helper 
class_docstr_tok: ["Dataset","partitioning","helper"]
code: def __getitem__(self, index):
      data_idx = self.index[index]
      return self.data[data_idx]
  
imports: import os
  import subprocess
  import sys
  from argparse import ArgumentParser
  from datetime import timedelta
  from math import ceil
  from random import Random
  import torch
  import torch.distributed
  import torch.nn
  import torch.nn.functional
  from torch import optim
  from torchvision import datasets, transforms
  from clearml import Task
label_desc: Use image node_id 3 for calling the Partition obj's underlying member method code with example usage: obj.__getitem__(index) and returns: self