code: def _run_batch_config(
      ds_config, train_batch=None, micro_batch=None, gas=None
  ):
      ds_config.train_batch_size = train_batch
      ds_config.train_micro_batch_size_per_gpu = micro_batch
      ds_config.gradient_accumulation_steps = gas
      success = True
      try:
          ds_config._configure_train_batch_size()
      except AssertionError:
          success = False
      return success
  
imports: import os
  import pytest
  import json
  import hjson
  import argparse
  from deepspeed.runtime.zero.config import DeepSpeedZeroConfig
  from deepspeed.accelerator import get_accelerator
  from unit.common import DistributedTest, get_test_path
  from unit.simple_model import SimpleModel, create_config_from_dict, random_dataloader
  import deepspeed.comm
  import deepspeed
  from deepspeed.runtime.config import DeepSpeedConfig, get_bfloat16_enabled
label_desc: Use image node_id 2 for calling a global function with example usage: _run_batch_config(ds_config, train_batch, micro_batch, gas) and returns: success