From cb3519e554d881d963e9e138f968f08886137e34 Mon Sep 17 00:00:00 2001
From: "Liam, SB Hoo" <shibinhoo@gmail.com>
Date: Sun, 15 Sep 2024 12:03:27 +0200
Subject: [PATCH] Fix formatting

---
 tabpfn_client/config.py    | 2 +-
 tabpfn_client/estimator.py | 3 +--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/tabpfn_client/config.py b/tabpfn_client/config.py
index 336b686..92e0c16 100644
--- a/tabpfn_client/config.py
+++ b/tabpfn_client/config.py
@@ -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)
 
diff --git a/tabpfn_client/estimator.py b/tabpfn_client/estimator.py
index a1b6599..e6db036 100644
--- a/tabpfn_client/estimator.py
+++ b/tabpfn_client/estimator.py
@@ -232,7 +232,6 @@ def predict_proba(self, X):
 
 
 class TabPFNRegressor(BaseEstimator, RegressorMixin):
-
     _AVAILABLE_MODELS = [
         "default",
         "2noar4o2",
@@ -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