suppress errors in ax (#1142)

Summary: Pull Request resolved: #1142 Differential Revision: D39534094 fbshipit-source-id: 3d711fbc9e187fc6e751f905fb31e58b7a2a3c5e
facebook · Sep 15, 2022 · d3d3ca1 · d3d3ca1
1 parent a16e66b
commit d3d3ca1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ax/models/torch/alebo.py b/ax/models/torch/alebo.py
@@ -103,6 +103,7 @@ def forward(
         # Compute kernel distance
         z1 = torch.matmul(x1, U_t)
         z2 = torch.matmul(x2, U_t)
+        # pyre-fixme[7]: Expected `Tensor` but got `Union[LinearOperator, Tensor]`.
         return self.covar_dist(
             z1,
             z2,