docstr: Check that vqvae can be built on input with a None batch dimension.
docstr_tok: ["Check","that","vqvae","can","be","built","on","input","with","a","None","batch","dimension","."]
code: def testNoneBatch(self, constructor, kwargs):
      """Check that vqvae can be built on input with a None batch dimension."""
      vqvae_module = constructor(**kwargs)
      inputs = jnp.zeros([0, 5, 5, kwargs["embedding_dim"]])
      vqvae_module(inputs, is_training=False)
  
file_path: dm-haiku/haiku/_src/nets/vqvae_test.py
imports: import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import stateful
  from haiku._src import test_utils
  from haiku._src import transform
  from haiku._src.nets import vqvae
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 3 for calling the VqvaeTest obj's underlying member method code with example usage: obj.testNoneBatch(constructor, kwargs) without return types