⚡ TabPFN: Foundation Model for Tabular Data ⚡
1307 matches across 14 categories. Click a row to expand file-level details.
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | tests/test_inference.py | 93 | def reset_save_peak_mem_factor(self, factor: int | None = None) -> None: |
| LOW | tests/test_inference.py | 136 | def reset_save_peak_mem_factor(self, factor: int | None = None) -> None: |
| LOW | tests/test_inference.py | 215 | def reset_save_peak_mem_factor(self, factor: int | None = None) -> None: |
| LOW | tests/test_inference.py | 219 | def test__cache_preprocessing__result_equal_in_serial_and_in_parallel() -> None: |
| LOW | tests/test_inference.py | 272 | def test__cache_preprocessing__with_outlier_removal() -> None: |
| LOW | tests/test_inference.py | 326 | def test__on_demand__result_equal_in_serial_and_in_parallel() -> None: |
| LOW | tests/test_inference.py | 389 | def test__iter_outputs__task_type_forwarded( |
| LOW | tests/test_inference.py | 433 | def _create_test_ensemble_configs( |
| LOW | tests/test_inference.py | 482 | def test__explicit_kv_cache__produces_outputs() -> None: |
| LOW | tests/test_inference.py | 531 | def test__explicit_kv_cache__keep_cache_on_device() -> None: |
| LOW | tests/test_consistency.py | 40 | def _get_tiny_classification_data() -> tuple[np.ndarray, np.ndarray, np.ndarray]: |
| LOW | tests/test_consistency.py | 52 | def _get_tiny_regression_data() -> tuple[np.ndarray, np.ndarray, np.ndarray]: |
| LOW | tests/test_consistency.py | 64 | def _get_iris_multiclass_data() -> tuple[np.ndarray, np.ndarray, np.ndarray]: |
| LOW | tests/test_consistency.py | 218 | def _get_current_platform_string() -> str: |
| LOW | tests/test_consistency.py | 244 | def test__fit_predict__predictions_match_reference( |
| LOW | tests/test_consistency.py | 284 | def _load_reference_predictions_or_fail(test_name: str) -> np.ndarray: |
| LOW | tests/test_consistency.py | 294 | def save_reference_predictions() -> None: |
| LOW | tests/test_utils.py | 21 | def test__infer_devices__auto__cuda_and_mps_not_available__selects_cpu( |
| LOW | tests/test_utils.py | 30 | def test__infer_devices__auto__single_cuda_gpu_available__selects_it( |
| LOW | tests/test_utils.py | 41 | def test__infer_devices__auto__multiple_cuda_gpus_available__selects_all( |
| LOW | tests/test_utils.py | 57 | def test__infer_devices__auto__cuda_and_mps_available_but_excluded__selects_cpu( |
| LOW | tests/test_utils.py | 68 | def test__infer_devices__auto__mps_available_but_torch_too_old__selects_cpu( |
| LOW | tests/test_utils.py | 77 | def test__infer_devices__device_specified__selects_it( |
| LOW | tests/test_utils.py | 88 | def test__infer_devices__multiple_devices_specified___selects_them( |
| LOW | tests/test_utils.py | 101 | def test__infer_devices__device_specified_twice__raises() -> None: |
| LOW | tests/test_utils.py | 109 | def test__infer_devices__mps_specified_but_torch_too_old__raises( |
| LOW | tests/test_utils.py | 199 | def test_balance_probas_by_class_counts(): |
| LOW | tests/test_utils.py | 215 | def test__translate_probs_across_borders__matches_unchunked(batch: int) -> None: |
| LOW | tests/test_utils.py | 232 | def test__translate_probs_across_borders__forces_chunking( |
| LOW | tests/test_estimators.py | 30 | def test__to__before_fit__does_not_crash( |
| LOW | tests/test_estimators.py | 46 | def test__to__between_fit_and_predict__does_not_crash( |
| LOW | tests/test_estimators.py | 62 | def test__to__between_fits__outputs_equal( |
| LOW | tests/test_estimators.py | 112 | def test__to__after_fit__no_tensors_left_on_old_device( |
| LOW | tests/test_estimators.py | 135 | def test__to__after_fit_and_predict__no_tensors_left_on_old_device( |
| LOW | tests/test_notebook_installs.py | 96 | def test_notebook_resolves_latest_pinned_packages( |
| LOW | tests/test_save_load_fitted_model.py | 29 | def _make_classification_data_with_categoricals() -> tuple[np.ndarray, np.ndarray]: |
| LOW | tests/test_save_load_fitted_model.py | 56 | def test__save_and_load_twice__predictions_equal_to_before_save( |
| LOW | tests/test_save_load_fitted_model.py | 101 | def test_saving_unfitted_model_raises_error(tmp_path: Path) -> None: |
| LOW | tests/test_save_load_fitted_model.py | 108 | def test__load_regressor_state_in_classifier__raises_error(tmp_path: Path) -> None: |
| LOW | tests/test_save_load_fitted_model.py | 121 | def test__load_classifier_state_in_regressor__raises_error(tmp_path: Path) -> None: |
| LOW | tests/test_save_load_fitted_model.py | 134 | def _init_and_save_unique_checkpoint( |
| LOW | tests/test_save_load_fitted_model.py | 149 | def test_saving_and_loading_model_with_weights(tmp_path: Path) -> None: |
| LOW | tests/test_save_load_fitted_model.py | 188 | def test_saving_and_loading_multiple_models_with_weights( |
| LOW | tests/test_save_load_fitted_model.py | 235 | def test_saving_and_loading_with_tuning_config( |
| LOW | tests/test_inference_config.py | 22 | def test__save_and_load__loaded_value_equal_to_saved() -> None: |
| LOW | tests/test_inference_config.py | 35 | def test__override_with_user_input__dict_of_overrides__sets_values_correctly() -> None: |
| LOW | tests/test_inference_config.py | 58 | def test__override_with_user_input__config_override__replaces_entire_config() -> None: |
| LOW | tests/test_inference_config.py | 126 | def test__classifier_get_inference_config__before_fit__returns_config() -> None: |
| LOW | tests/test_inference_config.py | 135 | def test__classifier_get_inference_config__returns_deepcopy() -> None: |
| LOW | tests/test_inference_config.py | 144 | def test__classifier_get_inference_config__with_override__applies_override() -> None: |
| LOW | tests/test_inference_config.py | 156 | def test__regressor_get_inference_config__before_fit__returns_config() -> None: |
| LOW | tests/test_inference_config.py | 165 | def test__regressor_get_inference_config__returns_deepcopy() -> None: |
| LOW | tests/test_inference_config.py | 174 | def test__regressor_get_inference_config__with_override__applies_override() -> None: |
| LOW | tests/test_parallel_execute.py | 14 | def test__parallel_execute__single_device__executes_in_current_thread() -> None: |
| LOW | tests/test_parallel_execute.py | 26 | def test__parallel_execute__single_device__results_in_same_order_as_functions() -> None: |
| LOW | tests/test_parallel_execute.py | 41 | def test__parallel_execute__multiple_devices__executes_in_worker_threads() -> None: |
| LOW | tests/test_parallel_execute.py | 55 | def test__parallel_execute__multiple_devices__results_in_same_order_as_functions() -> ( |
| LOW | tests/test_finetuning_regressor.py | 50 | def create_mock_architecture_forward_regression() -> Callable[..., torch.Tensor]: |
| LOW | tests/test_finetuning_regressor.py | 103 | def synthetic_regression_data() -> tuple[np.ndarray, np.ndarray]: |
| LOW | tests/test_finetuning_regressor.py | 135 | def variable_synthetic_regression_dataset_collection() -> list[ |
| 692 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | tests/test_finetuning_logging.py | 15 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_finetuning_logging.py | 17 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 56 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 58 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 108 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 110 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 131 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 133 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 591 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 593 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 26 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 28 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 235 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 237 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 358 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 360 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 405 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_browser_auth.py | 407 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_finetuning_classifier.py | 322 | # ============================================================================= |
| MEDIUM | tests/test_finetuning_classifier.py | 324 | # ============================================================================= |
| MEDIUM | tests/test_finetuning_classifier.py | 397 | # ============================================================================= |
| MEDIUM | tests/test_finetuning_classifier.py | 399 | # ============================================================================= |
| MEDIUM | tests/test_finetuning_classifier.py | 747 | # ============================================================================= |
| MEDIUM | tests/test_finetuning_classifier.py | 749 | # ============================================================================= |
| MEDIUM | tests/test_finetuning_classifier.py | 95 | # ============================================================================= |
| MEDIUM | tests/test_finetuning_classifier.py | 97 | # ============================================================================= |
| MEDIUM | …ch_preprocessing/test_gpu_preprocessing_consistency.py | 39 | # --------------------------------------------------------------------------- |
| MEDIUM | …ch_preprocessing/test_gpu_preprocessing_consistency.py | 41 | # --------------------------------------------------------------------------- |
| MEDIUM | …ch_preprocessing/test_gpu_preprocessing_consistency.py | 199 | # --------------------------------------------------------------------------- |
| MEDIUM | …ch_preprocessing/test_gpu_preprocessing_consistency.py | 201 | # --------------------------------------------------------------------------- |
| MEDIUM | …ch_preprocessing/test_gpu_preprocessing_consistency.py | 226 | # --------------------------------------------------------------------------- |
| MEDIUM | …ch_preprocessing/test_gpu_preprocessing_consistency.py | 228 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_architectures/test_mlx_backend.py | 175 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_architectures/test_mlx_backend.py | 177 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_architectures/test_mlx_backend.py | 248 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_architectures/test_mlx_backend.py | 250 | # --------------------------------------------------------------------------- |
| MEDIUM | tests/test_architectures/test_attention_backends.py | 61 | # --------------------------------------------------------------------- |
| MEDIUM | tests/test_architectures/test_attention_backends.py | 63 | # --------------------------------------------------------------------- |
| MEDIUM | tests/test_architectures/test_attention_backends.py | 128 | # --------------------------------------------------------------------- |
| MEDIUM | tests/test_architectures/test_attention_backends.py | 130 | # --------------------------------------------------------------------- |
| MEDIUM | examples/finetune_classifier.py | 39 | # ============================================================================= |
| MEDIUM | examples/finetune_classifier.py | 45 | # ============================================================================= |
| MEDIUM | examples/sagemaker.py | 45 | # --------------------------------------------------------------------------- |
| MEDIUM | examples/sagemaker.py | 47 | # --------------------------------------------------------------------------- |
| MEDIUM | examples/sagemaker.py | 57 | # --------------------------------------------------------------------------- |
| MEDIUM | examples/sagemaker.py | 59 | # --------------------------------------------------------------------------- |
| MEDIUM | examples/sagemaker.py | 74 | # --------------------------------------------------------------------------- |
| MEDIUM | examples/sagemaker.py | 76 | # --------------------------------------------------------------------------- |
| MEDIUM | examples/sagemaker.py | 353 | # --------------------------------------------------------------------------- |
| MEDIUM | examples/sagemaker.py | 355 | # --------------------------------------------------------------------------- |
| MEDIUM | examples/finetune_regressor.py | 36 | # ============================================================================= |
| MEDIUM | examples/finetune_regressor.py | 42 | # ============================================================================= |
| MEDIUM | .github/workflows/ci.yml | 138 | # ------------------------------------------------------------------- |
| MEDIUM | .github/workflows/ci.yml | 140 | # ------------------------------------------------------------------- |
| MEDIUM | .github/workflows/ci.yml | 184 | # ------------------------------------------------------------------- |
| MEDIUM | .github/workflows/ci.yml | 186 | # ------------------------------------------------------------------- |
| MEDIUM | .github/workflows/ci.yml | 253 | # ------------------------------------------------------------------- |
| MEDIUM | .github/workflows/ci.yml | 257 | # ------------------------------------------------------------------- |
| MEDIUM | src/tabpfn/browser_auth.py | 142 | # --------------------------------------------------------------------------- |
| MEDIUM | src/tabpfn/browser_auth.py | 144 | # --------------------------------------------------------------------------- |
| 38 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | tests/test_debug_versions.py | 3 | |
| LOW | tests/test_inference.py | 5 | |
| LOW | tests/test_consistency.py | 16 | |
| LOW | tests/test_utils.py | 3 | |
| LOW | tests/test_estimators.py | 5 | |
| LOW | tests/conftest.py | 10 | |
| LOW | tests/test_notebook_installs.py | 16 | |
| LOW | tests/quick_test.py | 9 | |
| LOW | tests/test_save_load_fitted_model.py | 5 | |
| LOW | tests/test_inference_config.py | 5 | |
| LOW | tests/test_parallel_execute.py | 5 | |
| LOW | tests/test_finetuning_regressor.py | 13 | |
| LOW | tests/test_model_move_backwards_compatibility.py | 3 | |
| LOW | tests/test_finetuning_logging.py | 5 | |
| LOW | tests/test_ft_utils.py | 3 | |
| LOW | tests/test_thinking_tokens.py | 3 | |
| LOW | tests/utils.py | 3 | |
| LOW | tests/test_config.py | 3 | |
| LOW | tests/test_model_loading.py | 3 | |
| LOW | tests/test_browser_auth.py | 5 | |
| LOW | tests/test_regressor_interface.py | 3 | |
| LOW | tests/test_telemetry_disabled.py | 5 | |
| LOW | tests/test_download_fallbacks.py | 3 | |
| LOW | tests/test_inference_tuning.py | 3 | |
| LOW | tests/test_transformer.py | 3 | |
| LOW | tests/test_finetuning_classifier.py | 12 | |
| LOW | tests/test_checkpoint.py | 3 | |
| LOW | tests/test_settings.py | 3 | |
| LOW | tests/test_classifier_interface.py | 3 | |
| LOW | …st_preprocessing/test_adaptive_quantile_transformer.py | 3 | |
| LOW | …preprocessing/test_encode_categorical_features_step.py | 5 | |
| LOW | …ocessing/test_nan_handling_polynomial_features_step.py | 5 | |
| LOW | tests/test_preprocessing/test_modality_detection.py | 5 | |
| LOW | tests/test_preprocessing/test_power_transformer.py | 3 | |
| LOW | tests/test_preprocessing/test_data_cleaning.py | 5 | |
| LOW | …reprocessing/test_reshape_feature_distribution_step.py | 5 | |
| LOW | tests/test_preprocessing/test_add_svd_features_step.py | 5 | |
| LOW | tests/test_preprocessing/test_kdi_transformer.py | 3 | |
| LOW | tests/test_preprocessing/test_preprocessing_steps.py | 3 | |
| LOW | tests/test_preprocessing/test_label_encoder.py | 5 | |
| LOW | …st_preprocessing/test_add_fingerprint_features_step.py | 5 | |
| LOW | …s/test_preprocessing/test_differentiable_znorm_step.py | 3 | |
| LOW | tests/test_preprocessing/test_datamodel.py | 5 | |
| LOW | tests/test_preprocessing/test_ensemble.py | 3 | |
| LOW | …st_preprocessing/test_remove_constant_features_step.py | 5 | |
| LOW | tests/test_preprocessing/test_pipeline_consistency.py | 16 | |
| LOW | tests/test_encoders/test_encoders.py | 5 | |
| LOW | tests/test_encoders/test_projections.py | 5 | |
| LOW | tests/test_encoders/test_ops.py | 5 | |
| LOW | tests/test_model/test_seperate_train_inference.py | 3 | |
| LOW | tests/test_model/test_bar_distribution.py | 3 | |
| LOW | tests/test_model/test_attention.py | 3 | |
| LOW | …torch_preprocessing/test_torch_quantile_transformer.py | 5 | |
| LOW | …orch_preprocessing/test_torch_shuffle_features_step.py | 5 | |
| LOW | …ts/test_torch_preprocessing/test_pipeline_interface.py | 5 | |
| LOW | …t_torch_preprocessing/test_torch_soft_clip_outliers.py | 5 | |
| LOW | …est_torch_preprocessing/test_torch_squashing_scaler.py | 5 | |
| LOW | …ch_preprocessing/test_gpu_preprocessing_consistency.py | 11 | |
| LOW | tests/test_torch_preprocessing/test_torch_svd.py | 5 | |
| LOW | …test_torch_preprocessing/test_torch_standard_scaler.py | 5 | |
| 277 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| HIGH | tests/test_preprocessing/test_add_svd_features_step.py | 0 | test that the step returns added columns when used in a pipeline. |
| HIGH | tests/test_preprocessing/test_add_svd_features_step.py | 0 | test that the step returns added columns when used in a pipeline. |
| HIGH | …st_preprocessing/test_add_fingerprint_features_step.py | 0 | test that the step returns added columns when used in a pipeline. |
| HIGH | …torch_preprocessing/test_torch_quantile_transformer.py | 0 | test with 3d tensor (t, b, h) shape commonly used in tabpfn. |
| HIGH | …t_torch_preprocessing/test_torch_soft_clip_outliers.py | 0 | test with 3d tensor (t, b, h) shape commonly used in tabpfn. |
| HIGH | …test_torch_preprocessing/test_torch_standard_scaler.py | 0 | test with 3d tensor (t, b, h) shape commonly used in tabpfn. |
| HIGH | tests/test_architectures/test_tabpfn_v2_6.py | 0 | construct v2.5 and base such that they have the same outputs. |
| HIGH | tests/test_architectures/test_tabpfn_v3.py | 0 | construct v2.5 and base such that they have the same outputs. |
| HIGH | tests/test_architectures/test_tabpfn_v2_5.py | 0 | construct v2.5 and base such that they have the same outputs. |
| HIGH | src/tabpfn/architectures/tabpfn_v2_6.py | 0 | if > 1, the features will be grouped into groups of this size and the attention is across groups. |
| HIGH | src/tabpfn/architectures/tabpfn_v2_5.py | 0 | if > 1, the features will be grouped into groups of this size and the attention is across groups. |
| HIGH | src/tabpfn/architectures/base/config.py | 0 | if > 1, the features will be grouped into groups of this size and the attention is across groups. |
| HIGH | src/tabpfn/architectures/tabpfn_v2_6.py | 0 | perform a forward pass. see modelinterface.forward() for the full docstring. |
| HIGH | src/tabpfn/architectures/tabpfn_v2_5.py | 0 | perform a forward pass. see modelinterface.forward() for the full docstring. |
| HIGH | src/tabpfn/architectures/base/transformer.py | 0 | perform a forward pass. see modelinterface.forward() for the full docstring. |
| HIGH | src/tabpfn/finetuning/finetuned_regressor.py | 0 | fine-tune the tabpfn model on the provided training data. args: x: the training input samples of shape (n_samples, n_fea |
| HIGH | src/tabpfn/finetuning/finetuned_base.py | 0 | fine-tune the tabpfn model on the provided training data. args: x: the training input samples of shape (n_samples, n_fea |
| HIGH | src/tabpfn/finetuning/finetuned_classifier.py | 0 | fine-tune the tabpfn model on the provided training data. args: x: the training input samples of shape (n_samples, n_fea |
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | tests/test_model_loading.py | 512 | # Create a fake checkpoint to simulate a successful download |
| MEDIUM | tests/test_regressor_interface.py | 457 | # Create a simple preprocessing pipeline |
| MEDIUM | tests/test_regressor_interface.py | 682 | # Create a CPU model |
| MEDIUM | tests/test_regressor_interface.py | 723 | # Create a CPU model |
| MEDIUM | tests/test_regressor_interface.py | 771 | # Create a TabPFNRegressor with fixed random state for reproducibility |
| MEDIUM | tests/test_regressor_interface.py | 788 | # Create a new dataset with added constant features |
| MEDIUM | tests/test_finetuning_classifier.py | 1026 | # Initialize two classifiers with the necessary modes |
| MEDIUM | tests/test_finetuning_classifier.py | 1194 | # Create a larger dataset where columns have distinct mean values |
| MEDIUM | tests/test_classifier_interface.py | 932 | # Create a TabPFNClassifier with fixed random state for reproducibility |
| MEDIUM | tests/test_classifier_interface.py | 940 | # Create a new dataset with added constant features |
| MEDIUM | tests/test_classifier_interface.py | 505 | # Create a subset of X to match the length of y_imbalanced |
| MEDIUM | tests/test_classifier_interface.py | 697 | # Create an IMBALANCED dataset |
| MEDIUM | tests/test_classifier_interface.py | 739 | # Create a simple preprocessing pipeline |
| MEDIUM | tests/test_classifier_interface.py | 976 | # Create a DataFrame with text and NA values |
| MEDIUM | tests/test_preprocessing/test_add_svd_features_step.py | 239 | # Create an invalid enum value by bypassing the enum |
| MEDIUM | examples/tabpfn_with_tuning.py | 38 | # Initialize a classifier with tuning and fit |
| MEDIUM | examples/tabpfn_with_tuning.py | 46 | # Initialize a classifier with tuning and fit |
| MEDIUM | examples/tabpfn_for_binary_classification.py | 20 | # Initialize a classifier |
| MEDIUM | examples/tabpfn_for_multiclass_classification.py | 23 | # Initialize a classifier |
| MEDIUM | src/tabpfn/model_loading.py | 784 | # Create a validated copy of logged params; avoid passing in arbitrary params |
| MEDIUM | src/tabpfn/regressor.py | 1053 | # Create a dictionary of outputs with proper typing via TypedDict |
| MEDIUM | src/tabpfn/misc/_sklearn_compat.py | 31 | # The following code does not depend on the sklearn version |
| MEDIUM | src/tabpfn/architectures/tabpfn_v2_6.py | 986 | # The following class imputation is performed for backwards compatibility. |
| MEDIUM | src/tabpfn/architectures/tabpfn_v2_5.py | 1094 | # The following class imputation is performed for backwards compatibility. |
| MEDIUM | src/tabpfn/architectures/encoders/steps/_ops.py | 282 | # Create a mask to zero out the padding positions. |
| MEDIUM | …tabpfn/preprocessing/torch/torch_soft_clip_outliers.py | 61 | # Create a clean copy with outliers masked as NaN |
| MEDIUM | …reprocessing/steps/encode_categorical_features_step.py | 153 | # Create a column transformer |
| MEDIUM | …reprocessing/steps/encode_categorical_features_step.py | 202 | # Create a column transformer |
| Severity | File | Line | Snippet |
|---|---|---|---|
| HIGH | examples/sagemaker.py | 87 | Serialize TabPFN input data and parameters into a JSON request body. This utility converts NumPy arrays and configu |
| HIGH | examples/sagemaker.py | 173 | Invoke the TabPFN SageMaker endpoint with tabular data. This helper prepares the payload for a TabPFN 2.5 model, se |
| HIGH | src/tabpfn/utils.py | 237 | Infer whether fp16 inference should be enabled. Args: devices: The devices to validate against. ena |
| HIGH | src/tabpfn/architectures/tabpfn_v2_6.py | 828 | Parse the config dict into a TabPFNV2Config, return unused keys. Args: config: Config dict to parse. This f |
| HIGH | src/tabpfn/architectures/interface.py | 117 | Parses the given config dict into ArchitectureConfig or a subclass. This config will then be passed to get_arch |
| HIGH | src/tabpfn/architectures/tabpfn_v2_5.py | 837 | Parse the config dict into a TabPFNV2Config, return unused keys. Args: config: Config dict to parse. This f |
| HIGH | src/tabpfn/architectures/base/__init__.py | 35 | Parse the config dict into a ModelConfig, a subclass of ArchitectureConfig. This method implements the interface de |
| HIGH | src/tabpfn/finetuning/data_util.py | 340 | Retrieves, splits, and preprocesses the dataset config at the index. Performs train/test splitting and applies |
| HIGH | src/tabpfn/finetuning/train_util.py | 240 | Create and initialize AdamW optimizer, optionally loading from checkpoint. Args: model_parameters: Iterator |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | …reprocessing/test_reshape_feature_distribution_step.py | 331 | |
| LOW | tests/test_preprocessing/test_ensemble.py | 777 | |
| LOW | tests/test_encoders/test_encoders.py | 166 | |
| LOW | src/tabpfn/classifier.py | 505 | |
| LOW | src/tabpfn/classifier.py | 1407 | |
| LOW | src/tabpfn/model_loading.py | 199 | |
| LOW | src/tabpfn/model_loading.py | 226 | |
| LOW | src/tabpfn/model_loading.py | 319 | |
| LOW | src/tabpfn/model_loading.py | 608 | |
| LOW | src/tabpfn/utils.py | 280 | |
| LOW | src/tabpfn/regressor.py | 493 | |
| LOW | src/tabpfn/regressor.py | 1085 | |
| LOW | src/tabpfn/misc/debug_versions.py | 188 | |
| LOW | src/tabpfn/misc/debug_versions.py | 269 | |
| LOW | src/tabpfn/misc/debug_versions.py | 347 | |
| LOW | src/tabpfn/architectures/interface.py | 43 | |
| LOW | src/tabpfn/architectures/tabpfn_v3.py | 1643 | |
| LOW | src/tabpfn/architectures/tabpfn_v3.py | 2017 | |
| LOW | src/tabpfn/architectures/base/transformer.py | 755 | |
| LOW | src/tabpfn/architectures/base/transformer.py | 602 | |
| LOW | src/tabpfn/finetuning/finetuned_base.py | 543 | |
| LOW | src/tabpfn/preprocessing/clean.py | 68 | |
| LOW | …eprocessing/steps/reshape_feature_distribution_step.py | 107 | |
| LOW | …eprocessing/steps/reshape_feature_distribution_step.py | 185 | |
| LOW | …reprocessing/steps/encode_categorical_features_step.py | 56 | |
| LOW | …n/preprocessing/steps/add_fingerprint_features_step.py | 68 |
| Severity | File | Line | Snippet |
|---|---|---|---|
| HIGH | src/tabpfn/browser_auth.py | 629 | ' 4. Set the environment variable: export TABPFN_TOKEN="<your-api-key>"\n' |
| HIGH | src/tabpfn/browser_auth.py | 631 | ' import os; os.environ["TABPFN_TOKEN"] = "<your-api-key>"' |
| HIGH | src/tabpfn/errors.py | 49 | ' export TABPFN_TOKEN="<your-api-key>"\n' |
| HIGH | src/tabpfn/errors.py | 51 | ' import os; os.environ["TABPFN_TOKEN"] = "<your-api-key>"' |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | tests/test_preprocessing/test_pipeline_consistency.py | 434 | except Exception as e: |
| MEDIUM | examples/sagemaker.py | 348 | print(f"Error parsing JSON response: {e}") |
| LOW | src/tabpfn/classifier.py | 752 | except Exception as e: # noqa: BLE001 |
| LOW | src/tabpfn/browser_auth.py | 137 | except Exception: |
| LOW | src/tabpfn/browser_auth.py | 207 | except Exception: |
| LOW | src/tabpfn/browser_auth.py | 166 | except Exception as exc: |
| LOW | src/tabpfn/browser_auth.py | 331 | except Exception: # noqa: BLE001 |
| LOW | src/tabpfn/browser_auth.py | 507 | except Exception: |
| LOW | src/tabpfn/model_loading.py | 302 | except Exception as e: # noqa: BLE001 |
| LOW | src/tabpfn/model_loading.py | 362 | except Exception: # noqa: BLE001 |
| LOW | src/tabpfn/model_loading.py | 367 | except Exception as e: # noqa: BLE001 |
| LOW | src/tabpfn/model_loading.py | 514 | except Exception as e: # noqa: BLE001 |
| LOW | src/tabpfn/model_loading.py | 525 | except Exception as e: # noqa: BLE001 |
| LOW | src/tabpfn/model_loading.py | 553 | except Exception as e: # noqa: BLE001 |
| LOW | src/tabpfn/preprocessing/steps/kdi_transformer.py | 156 | except Exception: # noqa: BLE001 |
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | pyproject.toml | 152 | # "DTZ", # One day I should know how to utilize timezones and dates... |
| MEDIUM | tests/test_finetuning_classifier.py | 1002 | n_total = 50 # Increased slightly for more robust testing |
| MEDIUM | …t_torch_preprocessing/test_torch_soft_clip_outliers.py | 58 | """Test that two-pass approach produces more robust bounds.""" |
| MEDIUM | src/tabpfn/inference_tuning.py | 262 | """Selects the robust optimal threshold for the given metric. |
| LOW | src/tabpfn/parallel_execute.py | 85 | # If we only have one device then just use the current thread to avoid overhead. |
| MEDIUM | src/tabpfn/preprocessing/configs.py | 71 | "robust", # a standard sklearn robust scaler |
| MEDIUM | src/tabpfn/preprocessing/steps/preprocessing_helpers.py | 116 | # TODO: Could utilize the categorical dtype values directly instead of "auto" |
| MEDIUM | …fn/preprocessing/steps/squashing_scaler_transformer.py | 88 | r"""Perform robust centering and scaling followed by soft clipping. |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | src/tabpfn/model_loading.py | 309 | # Check if this is an authentication/gating error |
| LOW | src/tabpfn/model_loading.py | 798 | # Check if the annotation is a string and the fit_mode is in it. |
| LOW | src/tabpfn/utils.py | 212 | # Check if the function is available in torch |
| LOW | …/tabpfn/architectures/base/attention/full_attention.py | 607 | # Check if we should use PyTorch 2.0's GQA support |
| LOW | src/tabpfn/preprocessing/steps/preprocessing_helpers.py | 56 | # Check if there is a single transformer, of subtype OneToOneFeatureMixin |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | examples/finetune_classifier.py | 41 | # For details and more options see FinetunedTabPFNClassifier |
| LOW | examples/sagemaker.py | 61 | # |
| LOW | src/tabpfn/misc/_sklearn_compat.py | 1001 | |
| LOW | …coders/steps/remove_duplicate_features_encoder_step.py | 61 | return state |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | tests/test_parallel_execute.py | 15 | def test_function(device: torch.device) -> int: # noqa: ARG001 |
| LOW | tests/test_parallel_execute.py | 42 | def test_function(device: torch.device) -> int: # noqa: ARG001 |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | tests/test_finetuning_classifier.py | 1235 | # Step 3: Validate that columns are in the same order across all batches. |