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