code: def setUp(self):
      super().setUp()
      (
          self.train_ds,
          self.eval_ds,
          self.predict_ds,
      ) = self._get_datasets()
  
file_path: flax/flax/experimental/nnx/examples/lm1b/input_pipeline_test.py
imports: import os
  import pathlib
  import tempfile
  import input_pipeline
  import tensorflow_datasets
  from absl.testing import absltest
  from configs import default
label_desc: Use image node_id 1 for calling the InputPipelineTest obj's underlying member method code with example usage: obj.setUp() without return types