def METHOD_NAME(self, batched_sampling: bool):
    def no_lag_cfg(cfg: Config):
        cfg.num_epochs = cfg.num_batches_per_epoch = 1
    self._run_test_env(
        env="mujoco_ant",
        num_workers=1,
        train_steps=4096,
        batch_size=2048,
        batched_sampling=batched_sampling,
        serial_mode=False,
        async_rl=False,
        expected_max_policy_lag=0,
        customize_cfg_func=no_lag_cfg,
    )