Skip to content

Commit 19e8da0

Browse files
authored
docs: Fix plot_employee_salaries.py example (#1395)
Fix #1363 and #1393
1 parent 8cab80d commit 19e8da0

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

examples/use_cases/plot_employee_salaries.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -188,9 +188,7 @@ def periodic_spline_transformer(period, n_splines=None, degree=3):
188188
import warnings
189189

190190
with warnings.catch_warnings():
191-
# catch the warnings raised by the OneHotEncoder for seeing unknown categories
192-
# at transform time
193-
warnings.simplefilter(action="ignore", category=UserWarning)
191+
warnings.simplefilter(action="ignore", category=FutureWarning)
194192
report.cache_predictions(n_jobs=4)
195193

196194
# %%
@@ -299,14 +297,16 @@ def periodic_spline_transformer(period, n_splines=None, degree=3):
299297
scoring_names=scoring_names,
300298
aggregate=["mean", "std"],
301299
),
302-
]
300+
],
301+
axis=1,
303302
)
304303
results
305304

306305
# %%
307306
# .. note::
308307
# We could have also used the :class:`skore.ComparisonReport` to compare estimator
309308
# reports.
309+
# This is done in :ref:`example_feature_importance`.
310310

311311
# %%
312312
#

0 commit comments

Comments
 (0)