Skip to content

Commit

Permalink
Fix formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
liam-sbhoo committed Sep 15, 2024
1 parent 5a6077c commit cb3519e
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
2 changes: 1 addition & 1 deletion tabpfn_client/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ def init(use_server=True):
# Only do the following if the initialization has not been done yet
return

if use_server:
if use_server:
service_client = ServiceClient()
user_auth_handler = UserAuthenticationClient(service_client)

Expand Down
3 changes: 1 addition & 2 deletions tabpfn_client/estimator.py
Original file line number Diff line number Diff line change
Expand Up @@ -232,7 +232,6 @@ def predict_proba(self, X):


class TabPFNRegressor(BaseEstimator, RegressorMixin):

_AVAILABLE_MODELS = [
"default",
"2noar4o2",
Expand Down Expand Up @@ -370,7 +369,7 @@ def predict_full(self, X):
train_set_uid=self.last_train_set_uid,
config=self.get_params(),
)

@classmethod
def list_available_models(cls) -> list[str]:
return cls._AVAILABLE_MODELS
Expand Down

0 comments on commit cb3519e

Please sign in to comment.