Pyre Configurationless migration for] [batch:2/15]

ax/analysis/cross_validation_plot.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from copy import deepcopy
 from typing import Any, Dict, List, Optional
 

ax/analysis/helpers/color_helpers.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 
 from numbers import Real
 from typing import Tuple

ax/analysis/helpers/constants.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 import enum
 
 # Constants used for numerous plots

ax/analysis/helpers/cross_validation_helpers.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from typing import Any, Dict, List, Optional, Tuple
 
 import numpy as np

ax/analysis/helpers/layout_helpers.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from typing import Any, Dict, List, Tuple, Type
 
 import plotly.graph_objs as go

ax/analysis/helpers/plot_data_df_helpers.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from typing import List, Set
 
 import numpy as np

ax/analysis/helpers/plot_helpers.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 
 from logging import Logger
 from typing import Dict, List, Optional, Tuple, Union

ax/analysis/helpers/scatter_helpers.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 import numbers
 
 from typing import Any, Dict, List, Optional, Tuple

ax/analysis/helpers/tests/test_cross_validation_helpers.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 import tempfile
 
 import plotly.graph_objects as go

ax/analysis/helpers/tests/test_cv_consistency_checks.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 import copy
 
 import plotly.graph_objects as go

ax/analysis/tests/test_cross_validation_plot.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 import plotly.graph_objects as go
 from ax.analysis.cross_validation_plot import CrossValidationPlot
 

ax/analysis/tests/test_predicted_outcomes_dot_plot.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 
 import unittest
 

ax/benchmark/metrics/__init__.py

@@ -2,3 +2,5 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
+
+# pyre-strict

ax/benchmark/metrics/base.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 """
 Module containing the metric base classes for benchmarks. The key property of
 a benchmark metric is whether it has a ground truth or not, which is indicated

ax/benchmark/metrics/benchmark.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from __future__ import annotations
 
 from typing import Any, Optional

ax/benchmark/metrics/utils.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from typing import Optional
 
 import pandas as pd

ax/benchmark/runners/__init__.py

@@ -2,3 +2,5 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
+
+# pyre-strict

ax/benchmark/runners/base.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from abc import ABC, abstractmethod
 from math import sqrt
 from typing import Any, Dict, List, Optional, Tuple, Union

ax/benchmark/runners/surrogate.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 import warnings
 from typing import Any, Dict, Iterable, List, Optional, Set, Tuple, Union
 

ax/modelbridge/external_generation_node.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 import time
 from abc import ABC, abstractmethod
 from logging import Logger

ax/modelbridge/tests/test_external_generation_node.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from typing import List, Optional
 from unittest.mock import MagicMock
 

ax/modelbridge/tests/test_hierarchical_search_space.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 
 from contextlib import ExitStack
 from random import random