Skip to content

Commit

Permalink
Silence mypy error
Browse files Browse the repository at this point in the history
  • Loading branch information
Scienfitz committed Jan 31, 2025
1 parent 4f48d6b commit 190bbe9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion baybe/surrogates/naive.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ def _estimate_moments(
import torch

# TODO: use target value bounds for covariance scaling when explicitly provided
mean = self._model * torch.ones([len(candidates_comp_scaled)])
mean = self._model * torch.ones([len(candidates_comp_scaled)]) # type: ignore[operator]
var = torch.ones(len(candidates_comp_scaled))
return mean, var

Expand Down

0 comments on commit 190bbe9

Please # to comment.