code: def get_next_batch(self, batch_size, split=None):
      if split is None:
          return self.dset.get_next_batch(batch_size)
      elif split == "train":
          return self.train_set.get_next_batch(batch_size)
      elif split == "val":
          return self.val_set.get_next_batch(batch_size)
      else:
          raise NotImplementedError
  
file_path: baselines/baselines/gail/dataset/mujoco_dset.py
imports: from baselines import logger
  import numpy
label_desc: Use image node_id 3 for calling the Mujoco_Dset obj's underlying member method code with example usage: obj.get_next_batch(batch_size, split) and returns: self, self, self