Skip to content

Commit

Permalink
Make train arguments optional
Browse files Browse the repository at this point in the history
  • Loading branch information
Scienfitz committed Apr 26, 2024
1 parent 386ea9b commit f47729d
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions baybe/acquisition/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
import warnings
from abc import ABC
from inspect import signature
from typing import ClassVar, Union
from typing import ClassVar, Optional, Union

import pandas as pd
from attrs import define
Expand Down Expand Up @@ -36,8 +36,8 @@ def is_mc(cls) -> bool:
def to_botorch(
self,
surrogate: Surrogate,
train_x: pd.DataFrame,
train_y: pd.DataFrame,
train_x: Optional[pd.DataFrame],
train_y: Optional[pd.DataFrame],
):
"""Create the botorch-ready representation of the function."""
import botorch.acquisition as botorch_analytical_acqf
Expand Down

0 comments on commit f47729d

Please sign in to comment.