From 82312733f3de2d6aa6422c4fe23601bc4f9eb0b6 Mon Sep 17 00:00:00 2001 From: generatedunixname89002005307016 Date: Wed, 6 Nov 2024 10:40:57 -0800 Subject: [PATCH] Add missing Pyre mode headers] [batch:15/1238] [shard:6/N] Differential Revision: D65528124 fbshipit-source-id: 3f02d1f3ecafeef663e66fda9d6eebf9eadf557e --- ax/analysis/healthcheck/can_generate_candidates.py | 2 ++ ax/analysis/healthcheck/should_generate_candidates.py | 2 ++ ax/analysis/healthcheck/tests/test_can_generate_candidates.py | 2 ++ .../healthcheck/tests/test_should_generate_candidates.py | 2 ++ ax/analysis/plotly/arm_effects/__init__.py | 2 ++ ax/analysis/plotly/arm_effects/insample_effects.py | 2 ++ ax/analysis/plotly/arm_effects/predicted_effects.py | 2 ++ ax/analysis/plotly/arm_effects/utils.py | 2 ++ ax/analysis/plotly/tests/test_insample_effects.py | 2 ++ ax/analysis/plotly/tests/test_predicted_effects.py | 2 ++ ax/analysis/plotly/tests/test_utils.py | 2 ++ 11 files changed, 22 insertions(+) diff --git a/ax/analysis/healthcheck/can_generate_candidates.py b/ax/analysis/healthcheck/can_generate_candidates.py index da2ee2bcd6b..5915aa77dc4 100644 --- a/ax/analysis/healthcheck/can_generate_candidates.py +++ b/ax/analysis/healthcheck/can_generate_candidates.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-unsafe + import json from datetime import datetime from typing import Optional diff --git a/ax/analysis/healthcheck/should_generate_candidates.py b/ax/analysis/healthcheck/should_generate_candidates.py index c6bb8562d68..43ade80af94 100644 --- a/ax/analysis/healthcheck/should_generate_candidates.py +++ b/ax/analysis/healthcheck/should_generate_candidates.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-unsafe + import json import pandas as pd diff --git a/ax/analysis/healthcheck/tests/test_can_generate_candidates.py b/ax/analysis/healthcheck/tests/test_can_generate_candidates.py index 8d3d3217de4..6ccd9754308 100644 --- a/ax/analysis/healthcheck/tests/test_can_generate_candidates.py +++ b/ax/analysis/healthcheck/tests/test_can_generate_candidates.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-unsafe + from datetime import datetime, timedelta import pandas as pd diff --git a/ax/analysis/healthcheck/tests/test_should_generate_candidates.py b/ax/analysis/healthcheck/tests/test_should_generate_candidates.py index 320a244f727..ed5298ffcf9 100644 --- a/ax/analysis/healthcheck/tests/test_should_generate_candidates.py +++ b/ax/analysis/healthcheck/tests/test_should_generate_candidates.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-unsafe + from random import randint from ax.analysis.analysis import AnalysisCardLevel diff --git a/ax/analysis/plotly/arm_effects/__init__.py b/ax/analysis/plotly/arm_effects/__init__.py index 8e84578e63d..355cfc7b681 100644 --- a/ax/analysis/plotly/arm_effects/__init__.py +++ b/ax/analysis/plotly/arm_effects/__init__.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-unsafe + from ax.analysis.plotly.arm_effects.insample_effects import InSampleEffectsPlot from ax.analysis.plotly.arm_effects.predicted_effects import PredictedEffectsPlot diff --git a/ax/analysis/plotly/arm_effects/insample_effects.py b/ax/analysis/plotly/arm_effects/insample_effects.py index a42b57cdeed..025439a8618 100644 --- a/ax/analysis/plotly/arm_effects/insample_effects.py +++ b/ax/analysis/plotly/arm_effects/insample_effects.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-unsafe + from itertools import chain from logging import Logger diff --git a/ax/analysis/plotly/arm_effects/predicted_effects.py b/ax/analysis/plotly/arm_effects/predicted_effects.py index 2bb7dfc210d..ac6eb8ac210 100644 --- a/ax/analysis/plotly/arm_effects/predicted_effects.py +++ b/ax/analysis/plotly/arm_effects/predicted_effects.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-unsafe + from itertools import chain from typing import Any diff --git a/ax/analysis/plotly/arm_effects/utils.py b/ax/analysis/plotly/arm_effects/utils.py index c8244ec0e5c..2d87cadce89 100644 --- a/ax/analysis/plotly/arm_effects/utils.py +++ b/ax/analysis/plotly/arm_effects/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-unsafe + from typing import Any import pandas as pd diff --git a/ax/analysis/plotly/tests/test_insample_effects.py b/ax/analysis/plotly/tests/test_insample_effects.py index ac4683de41b..11e17a5ec98 100644 --- a/ax/analysis/plotly/tests/test_insample_effects.py +++ b/ax/analysis/plotly/tests/test_insample_effects.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-unsafe + from unittest.mock import patch import torch diff --git a/ax/analysis/plotly/tests/test_predicted_effects.py b/ax/analysis/plotly/tests/test_predicted_effects.py index 8ff1c4ed850..1ea9314ba3a 100644 --- a/ax/analysis/plotly/tests/test_predicted_effects.py +++ b/ax/analysis/plotly/tests/test_predicted_effects.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-unsafe + from unittest.mock import patch import torch diff --git a/ax/analysis/plotly/tests/test_utils.py b/ax/analysis/plotly/tests/test_utils.py index c8626dd9af7..077de31b6f2 100644 --- a/ax/analysis/plotly/tests/test_utils.py +++ b/ax/analysis/plotly/tests/test_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-unsafe + from ax.analysis.plotly.utils import get_constraint_violated_probabilities from ax.core.metric import Metric from ax.core.outcome_constraint import ComparisonOp, OutcomeConstraint