Skip to content

Commit

Permalink
Merge pull request #925 from stamate/bugfix-sklearn_version_valid_met…
Browse files Browse the repository at this point in the history
…rics

fixed sklearn sanity checks for valid_metrics
  • Loading branch information
DominiqueMakowski authored Nov 2, 2023
2 parents 323929f + 76a1efa commit c1e38e1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion neurokit2/complexity/utils_entropy.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,7 +128,7 @@ def _get_count(
# -------------------
# Sanity checks
sklearn_version = version.parse(sklearn.__version__)
if sklearn_version == version.parse("1.3.0"):
if sklearn_version in [version.parse("1.3.0"), version.parse("1.3.0rc1")]:
valid_metrics = sklearn.neighbors.KDTree.valid_metrics() + ["range"]
else:
valid_metrics = sklearn.neighbors.KDTree.valid_metrics + ["range"]
Expand Down

0 comments on commit c1e38e1

Please sign in to comment.