Skip to content

Commit

Permalink
Fix nightly jobs by updating hypothesis strategies to account for skl…
Browse files Browse the repository at this point in the history
…earn change (#5925)

Nightly jobs are showing a failure from hypothesis having a strategy that calls sklearn make_regression with n_samples being zero, which is no longer supported. This PR fixes that.

Authors:
  - Dante Gama Dessavre (https://github.com/dantegd)

Approvers:
  - Bradley Dice (https://github.com/bdice)

URL: #5925
  • Loading branch information
dantegd authored Jun 13, 2024
1 parent 2aac623 commit 697a37c
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions python/cuml/testing/strategies.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright (c) 2022-2023, NVIDIA CORPORATION.
# Copyright (c) 2022-2024, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand Down Expand Up @@ -396,8 +396,8 @@ def combined_datasets_strategy(*datasets, name=None, doc=None):
def strategy(
draw,
dtypes=floating_dtypes(),
n_samples=integers(min_value=0, max_value=200),
n_features=integers(min_value=0, max_value=200),
n_samples=integers(min_value=1, max_value=200),
n_features=integers(min_value=1, max_value=200),
):
"""Datasets strategy composed of multiple datasets strategies."""
datasets_strategies = (
Expand Down

0 comments on commit 697a37c

Please sign in to comment.