Skip to content

Commit 5335f5e

Browse files
committed
Fix ruff D209 and E501 in differentiable_input tests
1 parent 401d96d commit 5335f5e

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
@@ -1054,7 +1054,8 @@ def test__fit_with_differentiable_input__categorical_features_rejected() -> None
10541054

10551055
def 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:
10881090
def 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

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

Comments
 (0)