Skip to content

Commit bebc232

Browse files
committed
Fix ruff D209 and E501 in differentiable_input tests
1 parent e9b23f2 commit bebc232

1 file changed

Lines changed: 11 additions & 6 deletions

File tree

tests/test_regressor_interface.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1073,7 +1073,8 @@ def test__fit_with_differentiable_input__categorical_features_rejected() -> None
10731073

10741074
def 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:
11071109
def 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

11471151
def 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

Comments
 (0)