From b7c36374fd2919d48b9dffdf81489f09a7b74baa Mon Sep 17 00:00:00 2001
From: Daniel Cohen <danielcohennyc@meta.com>
Date: Fri, 4 Oct 2024 08:06:38 -0700
Subject: [PATCH] Don't pass batch size so scheduler can use total concurrent
 arms for PTS

Summary: as per title

Differential Revision: D63835518
---
 ax/service/scheduler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ax/service/scheduler.py b/ax/service/scheduler.py
index b0d9f4be588..18625f71bde 100644
--- a/ax/service/scheduler.py
+++ b/ax/service/scheduler.py
@@ -1830,7 +1830,7 @@ def generate_candidates(
         """
         new_trials = self._get_next_trials(
             num_trials=num_trials,
-            n=self.options.batch_size or 1,
+            n=self.options.batch_size,
         )
         if len(new_trials) > 0:
             new_generator_runs = [gr for t in new_trials for gr in t.generator_runs]