code: def test_transform(self):
      def myloss(x):
          return (x**2).mean()
  
      g = objax.Grad(
          myloss, variables=objax.VarCollection(), input_argnums=(0,)
      )
      gv = objax.GradValues(
          myloss, variables=objax.VarCollection(), input_argnums=(0,)
      )
      gvp = objax.privacy.dpsgd.PrivateGradValues(
          myloss,
          objax.VarCollection(),
          noise_multiplier=1.0,
          l2_norm_clip=0.5,
          microbatch=1,
      )
      self.assertEqual(
          repr(g), "objax.Grad(f=myloss, input_argnums=(0,))"
      )
      self.assertEqual(
          repr(gv), "objax.GradValues(f=myloss, input_argnums=(0,))"
      )
      self.assertEqual(
          repr(gvp),
          "objax.privacy.dpsgd.gradient.PrivateGradValues(f=myloss, noise_multiplier=1.0,"
          " l2_norm_clip=0.5, microbatch=1, batch_axis=(0,))",
      )
      self.assertEqual(
          repr(objax.Jit(gv)),
          "objax.Jit(f=objax.GradValues(f=myloss, input_argnums=(0,)), static_argnums=None)",
      )
      self.assertEqual(
          repr(objax.Jit(myloss, vc=objax.VarCollection())),
          "objax.Jit(f=objax.Function(f=myloss), static_argnums=None)",
      )
      self.assertEqual(
          repr(objax.Parallel(gv)),
          "objax.Parallel(f=objax.GradValues(f=myloss, input_argnums=(0,)),"
          " reduce=concatenate(*, axis=0, dtype=None), axis_name='device', static_argnums=None)",
      )
      self.assertEqual(
          repr(objax.Vectorize(myloss, vc=objax.VarCollection())),
          "objax.Vectorize(f=objax.Function(f=myloss), batch_axis=(0,))",
      )
      self.assertEqual(
          repr(objax.ForceArgs(gv, training=True, word="hello")),
          "objax.ForceArgs(module=GradValues, training=True, word='hello')",
      )
  
file_path: objax/tests/repr.py
imports: import functools
  import unittest
  import jax.numpy
  import objax
label_desc: Use image node_id 5 for calling the TestRepr obj's underlying member method code with example usage: obj.test_transform() and returns: unknown