code: def test_estimate_padding_budget_invalid(self, invalid_batch_size):
      with self.assertRaises(ValueError):
          input_pipeline.estimate_padding_budget_for_batch_size(
              self.datasets["train"],
              invalid_batch_size,
              num_estimation_graphs=1,
          )
  
file_path: flax/examples/ogbg_molpcba/input_pipeline_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import input_pipeline
  import jraph
  import tensorflow
label_desc: Use image node_id 3 for calling the InputPipelineTest obj's underlying member method code with example usage: obj.test_estimate_padding_budget_invalid(invalid_batch_size) without return types