variable pi not assigned

sigopt · May 15, 2024 · 80b0af3 · 80b0af3
1 parent 85619c1
commit 80b0af3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/libsigopt/compute/search.py b/libsigopt/compute/search.py
@@ -15,8 +15,8 @@
 @dataclass(frozen=True, slots=True)
 class SearchCoreComponents:
   search_points_to_evaluate: numpy.ndarray
-  pi: numpy.ndarray
-  pi_grad: numpy.ndarray
+  pi: numpy.ndarray | None
+  pi_grad: numpy.ndarray | None
 
 
 class FakePredictor(Predictor):
@@ -66,6 +66,7 @@ def compute_core_components(self, points_to_evaluate, option):
     assert len(eval_shape) == 2 and eval_shape[1] == self.dim
     search_points_to_evaluate = convert_one_hot_to_search_hypercube_points(self.domain, points_to_evaluate)
     pi_grad = None
+    pi = None
     if option in ("func",):
       pi = self.failure_model.compute_probability_of_success(points_to_evaluate)
     return SearchCoreComponents(search_points_to_evaluate, pi, pi_grad)