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