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 = tf.zeros([0, 5, 5, kwargs["embedding_dim"]])
      vqvae_module(inputs, is_training=False)
  
file_path: sonnet/sonnet/src/nets/vqvae_test.py
imports: from absl.testing import parameterized
  import numpy
  from sonnet.src import test_utils
  from sonnet.src.nets import vqvae
  import tensorflow
  import tree
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