Silence mypy error

emdgroup · Sep 6, 2024 · 591381b · 591381b
1 parent 4120fbb
commit 591381b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/baybe/surrogates/base.py b/baybe/surrogates/base.py
@@ -235,6 +235,10 @@ def _posterior_comp(self, candidates_comp: Tensor, /) -> Posterior:
             The same :class:`botorch.posteriors.Posterior` object as returned via
             :meth:`baybe.surrogates.base.Surrogate.posterior`.
         """
+        # FIXME[typing]: It seems there is currently no better way to inform the type
+        #   checker that the attribute is available at the time of the function call
+        assert self._input_scaler is not None
+
         p = self._posterior(self._input_scaler.transform(candidates_comp))
         if self._output_scaler is not _IDENTITY_TRANSFORM:
             p = self._output_scaler.untransform_posterior(p)