diff --git a/grakel_replace/graph_aware_gp_optimization_example.py b/grakel_replace/graph_aware_gp_optimization_example.py index 145cbc2f..7f5c0fdb 100644 --- a/grakel_replace/graph_aware_gp_optimization_example.py +++ b/grakel_replace/graph_aware_gp_optimization_example.py @@ -13,8 +13,8 @@ from gpytorch import ExactMarginalLogLikelihood from gpytorch.kernels import AdditiveKernel, MaternKernel from grakel_replace.context_managers import set_graph_lookup -from grakel_replace.optimization import optimize_acqf_graph from grakel_replace.kernels import BoTorchWLKernel +from grakel_replace.optimization import optimize_acqf_graph from grakel_replace.utils import min_max_scale, seed_all if TYPE_CHECKING: