@@ -1073,7 +1073,8 @@ def test__fit_with_differentiable_input__categorical_features_rejected() -> None
10731073
10741074def test__fit_with_differentiable_input__constant_target_rejected () -> None :
10751075 """A constant-target y has no signal to predict differentiably and would
1076- collapse the bardist borders; reject with a clear error."""
1076+ collapse the bardist borders; reject with a clear error.
1077+ """
10771078 reg = TabPFNRegressor (
10781079 n_estimators = 1 ,
10791080 ignore_pretraining_limits = True ,
@@ -1091,7 +1092,8 @@ def test__fit_with_differentiable_input__single_sample_y_does_not_nan() -> None:
10911092 N=1. Our path uses correction=0 (population std) so std is well defined
10921093 even for a single sample (it just collapses to 0, which then trips the
10931094 constant-target guard — what we want). Verify the failure mode is the
1094- explicit ValueError, not a downstream NaN."""
1095+ explicit ValueError, not a downstream NaN.
1096+ """
10951097 reg = TabPFNRegressor (
10961098 n_estimators = 1 ,
10971099 ignore_pretraining_limits = True ,
@@ -1107,7 +1109,8 @@ def test__fit_with_differentiable_input__single_sample_y_does_not_nan() -> None:
11071109def test__fit_with_differentiable_input__std_matches_population_definition () -> None :
11081110 """The differentiable path's y_train_std_ should match np.std (population
11091111 std, ddof=0), not torch's default sample std (correction=1), so it lines
1110- up with the standard fit() path."""
1112+ up with the standard fit() path.
1113+ """
11111114 reg = TabPFNRegressor (
11121115 n_estimators = 1 ,
11131116 ignore_pretraining_limits = True ,
@@ -1125,9 +1128,10 @@ def test__fit_with_differentiable_input__std_matches_population_definition() ->
11251128 )
11261129
11271130
1128- def test__fit_with_differentiable_input__feature_schema_columns_are_independent () -> None :
1131+ def test__fit_with_differentiable_input__feature_schema_cols_independent () -> None :
11291132 """Each column's Feature must be a distinct instance — list multiplication
1130- `[Feature(...)] * n` would alias all columns to one mutable dataclass."""
1133+ `[Feature(...)] * n` would alias all columns to one mutable dataclass.
1134+ """
11311135 reg = TabPFNRegressor (
11321136 n_estimators = 1 ,
11331137 ignore_pretraining_limits = True ,
@@ -1146,7 +1150,8 @@ def test__fit_with_differentiable_input__feature_schema_columns_are_independent(
11461150
11471151def test__fit_with_differentiable_input__second_call_refreshes_target_stats () -> None :
11481152 """A second call with different y must update y_train_mean_/std_ and the
1149- raw_space_bardist_; only the model load and ensemble configs are cached."""
1153+ raw_space_bardist_; only the model load and ensemble configs are cached.
1154+ """
11501155 torch .manual_seed (0 )
11511156 reg = TabPFNRegressor (
11521157 n_estimators = 1 ,
0 commit comments