docstr: Vectorize module with a broadcast argument and a static one with incomplete batch argument.
docstr_tok: ["Vectorize","module","with","a","broadcast","argument","and","a","static","one","with","incomplete","batch","argument","."]
code: def test_vectorize_module_one_arg_one_static_missing_batched(self):
      """Vectorize module with a broadcast argument and a static one with incomplete batch argument."""
      f = objax.nn.Linear(3, 4)
      with self.assertRaises(AssertionError):
          _ = objax.Vectorize(
              lambda x, y: f(x) + y, f.vars(), batch_axis=(0,)
          )
  
file_path: objax/tests/vectorize.py
imports: import unittest
  import jax.numpy
  import numpy
  import objax
label_desc: Use image node_id 7 for calling the TestVectorize obj's underlying member method code with example usage: obj.test_vectorize_module_one_arg_one_static_missing_batched() without return types