@@ -1054,7 +1054,8 @@ def test__fit_with_differentiable_input__categorical_features_rejected() -> None
10541054
10551055def test__fit_with_differentiable_input__constant_target_rejected () -> None :
10561056 """A constant-target y has no signal to predict differentiably and would
1057- collapse the bardist borders; reject with a clear error."""
1057+ collapse the bardist borders; reject with a clear error.
1058+ """
10581059 reg = TabPFNRegressor (
10591060 n_estimators = 1 ,
10601061 ignore_pretraining_limits = True ,
@@ -1072,7 +1073,8 @@ def test__fit_with_differentiable_input__single_sample_y_does_not_nan() -> None:
10721073 N=1. Our path uses correction=0 (population std) so std is well defined
10731074 even for a single sample (it just collapses to 0, which then trips the
10741075 constant-target guard — what we want). Verify the failure mode is the
1075- explicit ValueError, not a downstream NaN."""
1076+ explicit ValueError, not a downstream NaN.
1077+ """
10761078 reg = TabPFNRegressor (
10771079 n_estimators = 1 ,
10781080 ignore_pretraining_limits = True ,
@@ -1088,7 +1090,8 @@ def test__fit_with_differentiable_input__single_sample_y_does_not_nan() -> None:
10881090def test__fit_with_differentiable_input__std_matches_population_definition () -> None :
10891091 """The differentiable path's y_train_std_ should match np.std (population
10901092 std, ddof=0), not torch's default sample std (correction=1), so it lines
1091- up with the standard fit() path."""
1093+ up with the standard fit() path.
1094+ """
10921095 reg = TabPFNRegressor (
10931096 n_estimators = 1 ,
10941097 ignore_pretraining_limits = True ,
@@ -1106,9 +1109,10 @@ def test__fit_with_differentiable_input__std_matches_population_definition() ->
11061109 )
11071110
11081111
1109- def test__fit_with_differentiable_input__feature_schema_columns_are_independent () -> None :
1112+ def test__fit_with_differentiable_input__feature_schema_cols_independent () -> None :
11101113 """Each column's Feature must be a distinct instance — list multiplication
1111- `[Feature(...)] * n` would alias all columns to one mutable dataclass."""
1114+ `[Feature(...)] * n` would alias all columns to one mutable dataclass.
1115+ """
11121116 reg = TabPFNRegressor (
11131117 n_estimators = 1 ,
11141118 ignore_pretraining_limits = True ,
@@ -1127,7 +1131,8 @@ def test__fit_with_differentiable_input__feature_schema_columns_are_independent(
11271131
11281132def test__fit_with_differentiable_input__second_call_refreshes_target_stats () -> None :
11291133 """A second call with different y must update y_train_mean_/std_ and the
1130- raw_space_bardist_; only the model load and ensemble configs are cached."""
1134+ raw_space_bardist_; only the model load and ensemble configs are cached.
1135+ """
11311136 torch .manual_seed (0 )
11321137 reg = TabPFNRegressor (
11331138 n_estimators = 1 ,
0 commit comments