doc: correct the comments

pytorch · Nov 20, 2024 · d75734f · d75734f
1 parent 570a2cb
commit d75734f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/optim/test_optimize.py b/test/optim/test_optimize.py
@@ -204,13 +204,13 @@ def test_optimize_acqf_joint(
                 cnt += 1
                 self.assertEqual(mock_gen_batch_initial_conditions.call_count, cnt)
 
-                # test case where provided initial conditions larger than raw_samples
+                # test case where provided initial conditions equal to raw_samples
                 candidates, acq_vals = optimize_acqf(
                     acq_function=mock_acq_function,
                     bounds=bounds,
                     q=q,
                     num_restarts=num_restarts,
-                    raw_samples=3,
+                    raw_samples=raw_samples,
                     options=options,
                     return_best_only=False,
                     batch_initial_conditions=torch.zeros(
@@ -222,7 +222,7 @@ def test_optimize_acqf_joint(
                 self.assertTrue(torch.equal(acq_vals, mock_acq_values))
                 self.assertEqual(mock_gen_batch_initial_conditions.call_count, cnt)
 
-                # test generation with provided initial conditions less than raw_samples
+                # test generation with provided initial conditions less than num_restarts
                 candidates, acq_vals = optimize_acqf(
                     acq_function=mock_acq_function,
                     bounds=bounds,