Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions src/tabpfn_client/constants.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,3 +13,7 @@ class ModelVersion(str, Enum):


CACHE_DIR = Path(__file__).parent.resolve() / ".tabpfn"

URL_TABPFN_CLIENT_GITHUB_ISSUES = "https://github.com/priorlabs/tabpfn-client/issues"
URL_PRIOR_LABS_TERMS_AND_CONDITIONS = "https://priorlabs.ai/general-terms-and-conditions"
URL_TABPFN_EXTENSIONS_GITHUB_MANY_CLASS_CODE = "https://github.com/PriorLabs/tabpfn-extensions/blob/main/src/tabpfn_extensions/many_class/many_class_classifier.py" # noqa: E501
24 changes: 12 additions & 12 deletions src/tabpfn_client/estimator.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,24 +3,25 @@

from __future__ import annotations

from typing import Any, Callable, Optional, Literal, Dict, Union
import logging
from typing_extensions import Self
import numpy as np
import pandas as pd
import sys
import time
from concurrent.futures import ThreadPoolExecutor
from tabpfn_client.client import ModelType
from tabpfn_client.config import init
from typing import Any, Callable, Dict, Literal, Optional, Union
from typing_extensions import Self

import numpy as np
import pandas as pd
from sklearn.base import BaseEstimator, ClassifierMixin, RegressorMixin
from sklearn.utils import column_or_1d
from sklearn.utils.multiclass import check_classification_targets
from sklearn.utils.validation import check_is_fitted

from tabpfn_client.config import Config
from tabpfn_client.client import PredictionResult
from tabpfn_client.constants import ModelVersion
from tabpfn_client.client import ModelType, PredictionResult
from tabpfn_client.config import Config, init
from tabpfn_client.constants import (
URL_TABPFN_EXTENSIONS_GITHUB_MANY_CLASS_CODE,
ModelVersion,
)
from tabpfn_client.service_wrapper import InferenceClient

try:
Expand Down Expand Up @@ -322,8 +323,7 @@ def _validate_targets_and_classes(self, y) -> np.ndarray:
f"Number of classes {len(self.classes_)} exceeds the maximal number of "
f"{MAX_NUMBER_OF_CLASSES} classes supported by TabPFN. Consider using "
"a strategy to reduce the number of classes. For code see "
"https://github.com/PriorLabs/tabpfn-extensions/blob/main/src/"
"tabpfn_extensions/many_class/many_class_classifier.py"
f"{URL_TABPFN_EXTENSIONS_GITHUB_MANY_CLASS_CODE}"
)


Expand Down
9 changes: 7 additions & 2 deletions src/tabpfn_client/prompt_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,10 @@

from password_strength import PasswordPolicy

from tabpfn_client.constants import (
URL_PRIOR_LABS_TERMS_AND_CONDITIONS,
URL_TABPFN_CLIENT_GITHUB_ISSUES,
)
from tabpfn_client.service_wrapper import UserAuthenticationClient
from tabpfn_client.ui import (
console,
Expand Down Expand Up @@ -107,7 +111,7 @@ def prompt_welcome(cls):
)
console.print(
cls.indent(
"[cyan]Report issues: https://github.com/priorlabs/tabpfn-client/issues[/cyan]"
f"[cyan]Report issues: {URL_TABPFN_CLIENT_GITHUB_ISSUES}[/cyan]"
)
)
console.print(cls.indent("[cyan]Press Ctrl+C anytime to exit[/cyan]"))
Expand Down Expand Up @@ -372,7 +376,8 @@ def _prompt_and_set_token_impl(cls) -> bool:
def prompt_terms_and_cond(cls) -> bool:
"""Simplified terms prompt for registration flow."""
console.print(
"By using TabPFN, you agree to the terms and conditions at [link=https://www.priorlabs.ai/terms]https://www.priorlabs.ai/terms[/link]"
"By using TabPFN, you agree to the terms and conditions at "
f"[link={URL_PRIOR_LABS_TERMS_AND_CONDITIONS}]{URL_PRIOR_LABS_TERMS_AND_CONDITIONS}[/link]"
)

while True:
Expand Down