code: def __post_init__(self):
      # When providing polymorphic shapes, tensor_spec should be provided as well.
      assert bool(self.polymorphic_shapes) == bool(self.tensor_spec)
  
imports: from __future__ import annotations
  from collections.abc import Sequence
  import dataclasses
  import functools
  from typing import Any, Callable, Optional, Union
  import re
  import numpy
  import jraph
  from jax.experimental.jax2tf.tests.flax_models import actor_critic
  from jax.experimental.jax2tf.tests.flax_models import bilstm_classifier
  from jax.experimental.jax2tf.tests.flax_models import cnn
  from jax.experimental.jax2tf.tests.flax_models import gnn
  from jax.experimental.jax2tf.tests.flax_models import resnet
  from jax.experimental.jax2tf.tests.flax_models import seq2seq_lstm
  from jax.experimental.jax2tf.tests.flax_models import transformer_lm1b
  from jax.experimental.jax2tf.tests.flax_models import transformer_nlp_seq
  from jax.experimental.jax2tf.tests.flax_models import transformer_wmt
  from jax.experimental.jax2tf.tests.flax_models import vae
  import jax
  from jax import random
  import tensorflow
label_desc: Use image node_id 1 for calling the ModelHarness obj's underlying member method code with example usage: obj.__post_init__() without return types