From 32e02907f39f585cee28ec42cf8e87ea5aaf3957 Mon Sep 17 00:00:00 2001 From: AdrianSosic Date: Fri, 26 Apr 2024 10:28:41 +0200 Subject: [PATCH] Explicitly load torch to make doc pipeline pass --- baybe/acquisition/adapter.py | 1 + 1 file changed, 1 insertion(+) diff --git a/baybe/acquisition/adapter.py b/baybe/acquisition/adapter.py index e14908309..f5ca7f3f7 100644 --- a/baybe/acquisition/adapter.py +++ b/baybe/acquisition/adapter.py @@ -4,6 +4,7 @@ from typing import Any, Callable, Optional import gpytorch.distributions +import torch # noqa: F401 from botorch.acquisition import AcquisitionFunction as BotorchAcquisitionFunction from botorch.models.gpytorch import Model from botorch.posteriors import Posterior