code: def disabled_test_convert_wrn(self):
      # Make a model
      model = WideResNet(NCHANNELS, NCLASSES, depth=4, width=1)
      # Prediction op without JIT
      predict_op = objax.nn.Sequential(
          [
              objax.ForceArgs(model, training=False),
              objax.functional.softmax,
          ]
      )
      predict_tf = objax.util.Objax2Tf(predict_op)
      # Compare results
      self.verify_converted_predict_op(
          predict_op,
          predict_tf,
          shape=(BATCH_SIZE, NCHANNELS, IMAGE_SIZE, IMAGE_SIZE),
      )
      # Predict op with JIT
      predict_op_jit = objax.Jit(predict_op)
      predict_tf_jit = objax.util.Objax2Tf(predict_op_jit)
      # Compare results
      self.verify_converted_predict_op(
          predict_op_jit,
          predict_tf_jit,
          shape=(BATCH_SIZE, NCHANNELS, IMAGE_SIZE, IMAGE_SIZE),
      )
  
file_path: objax/tests/objax2tf.py
imports: import shutil
  import tempfile
  import unittest
  import numpy
  import objax
  from objax.zoo.wide_resnet import WideResNet
  import tensorflow
label_desc: Use image node_id 2 for calling the TestObjax2Tf obj's underlying member method code with example usage: obj.disabled_test_convert_wrn() without return types