code: def uses_tfds(test_fn):
      def _new_test_fn(self, *args, **kwargs):
          if FLAGS.use_tfds:
              test_fn(self, *args, **kwargs)
          else:
              self.skipTest(
                  "Uses TensorFlow Datasets. Enable using --use_tfds"
              )
  
      return _new_test_fn
  
imports: import collections
  import importlib
  from absl import flags
  from absl import logging
  import jax
  import numpy
  import tensorflow.compat.v2
  import tensorflow_probability
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 1 for calling a global function with example usage: uses_tfds(test_fn) and returns: _new_test_fn