class_docstr: Test cases for LM library.
class_docstr_tok: ["Test","cases","for","LM","library","."]
code: def setUp(self):
      super().setUp()
      tf.config.experimental.set_visible_devices([], "GPU")
  
file_path: flax/flax/experimental/nnx/examples/lm1b/train_test.py
imports: import pathlib
  import tempfile
  import jax
  import tensorflow
  import tensorflow_datasets
  import train
  from absl import logging
  from absl.testing import absltest
  from configs import default
label_desc: Use image node_id 1 for calling the TrainTest obj's underlying member method code with example usage: obj.setUp() without return types