docstr: Vectorize module with a broadcast argument and a static one with incomplete batch argument.
  Catch exception during call for variadic functions.
docstr_tok: ["Vectorize","module","with","a","broadcast","argument","and","a","static","one","with","incomplete","batch","argument",".","Catch","exception","during","call","for","variadic","functions","."]
code: def test_vectorize_module_one_arg_one_static_missing_batched_call(
      self,
  ):
      """Vectorize module with a broadcast argument and a static one with incomplete batch argument.
      Catch exception during call for variadic functions."""
      f = objax.nn.Linear(3, 4)
      c = objax.random.normal([4])
      fv = objax.Vectorize(
          lambda *a: f(a[0]) + a[1], f.vars(), batch_axis=(0,)
      )
      x = objax.random.normal((96, 3))
      with self.assertRaises(AssertionError):
          _ = fv(x, c)
  
file_path: objax/tests/vectorize.py
imports: import unittest
  import jax.numpy
  import numpy
  import objax
label_desc: Use image node_id 8 for calling the TestVectorize obj's underlying member method code with example usage: obj.test_vectorize_module_one_arg_one_static_missing_batched_call() without return types