More pytests

hcho3 · May 2, 2023 · 68e5f4d · 68e5f4d
1 parent e7dac2a
commit 68e5f4d
Show file tree

Hide file tree

Showing 3 changed files with 440 additions and 2 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -35,7 +35,8 @@ repos:
       language: python
       types: [python]
       additional_dependencies: [
-        pylint, hatchling, pytest, scikit-learn, hypothesis, pandas, treelite
+        pylint, hatchling, pytest, scikit-learn, hypothesis, pandas, treelite,
+        lightgbm, xgboost
       ]
 - repo: https://github.com/pocc/pre-commit-hooks
   rev: v1.3.5

diff --git a/tests/python/hypothesis_util.py b/tests/python/hypothesis_util.py
@@ -247,7 +247,8 @@ def standard_regression_datasets(
     n_features_ = draw(n_features)
     if n_informative is None:
         n_informative = just(max(min(n_features_, 1), int(0.1 * n_features_)))
-    X, y, _ = make_regression(
+    # pylint: disable=unbalanced-tuple-unpacking
+    X, y = make_regression(
         n_samples=draw(n_samples),
         n_features=n_features_,
         n_informative=draw(n_informative),