From d65c9cfcf7def25dcc4da2b32795bad25cac4042 Mon Sep 17 00:00:00 2001 From: DevmatePyreMissingAnnotations Bot Date: Thu, 2 Jul 2026 10:04:03 -0700 Subject: [PATCH] fbcode/ax/orchestration/tests/test_orchestrator.py Reviewed By: mgarrard Differential Revision: D110485285 --- ax/orchestration/tests/test_orchestrator.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ax/orchestration/tests/test_orchestrator.py b/ax/orchestration/tests/test_orchestrator.py index 0a29ff034c0..29647603d59 100644 --- a/ax/orchestration/tests/test_orchestrator.py +++ b/ax/orchestration/tests/test_orchestrator.py @@ -5,6 +5,8 @@ # pyre-strict +from __future__ import annotations + import logging import os import re @@ -514,9 +516,8 @@ def test_run_all_trials_callback(self) -> None: ), db_settings=self.db_settings_if_always_needed, ) - trials_info = {"n_completed": 0} + trials_info: dict[str, int] = {"n_completed": 0} - # pyre-fixme[53]: Captured variable `trials_info` is not annotated. def write_n_trials(orchestrator: Orchestrator) -> None: trials_info["n_completed"] = len(orchestrator.experiment.trials) @@ -556,9 +557,8 @@ def test_run_n_trials(self) -> None: self.base_run_n_trials(None) def test_run_n_trials_callback(self) -> None: - test_obj = [0, 0] + test_obj: list[int | str | None] = [0, 0] - # pyre-fixme[53]: Captured variable `test_obj` is not annotated. def _callback(orchestrator: Orchestrator) -> None: # pyrefly: ignore [unsupported-operation] test_obj[0] = orchestrator._latest_optimization_start_timestamp