From d3d3ca12d1c3b482ecf67da9200dda25881168da Mon Sep 17 00:00:00 2001
From: Pyre Bot Jr <>
Date: Thu, 15 Sep 2022 08:52:33 -0700
Subject: [PATCH] suppress errors in `ax` (#1142)

Summary: Pull Request resolved: https://github.com/facebook/Ax/pull/1142

Differential Revision: D39534094

fbshipit-source-id: 3d711fbc9e187fc6e751f905fb31e58b7a2a3c5e
---
 ax/models/torch/alebo.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ax/models/torch/alebo.py b/ax/models/torch/alebo.py
index d420347b24f..52a4d3ae1f5 100644
--- 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,