From 6913890bfe7ba00e0e660a54f684bc85ed609ae9 Mon Sep 17 00:00:00 2001 From: Reid Johnson Date: Fri, 9 Feb 2024 20:24:28 -0800 Subject: [PATCH] Linting --- quantile_forest/_quantile_forest.py | 4 +++- quantile_forest/tests/test_quantile_forest.py | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/quantile_forest/_quantile_forest.py b/quantile_forest/_quantile_forest.py index d4e3041..517117e 100755 --- a/quantile_forest/_quantile_forest.py +++ b/quantile_forest/_quantile_forest.py @@ -793,7 +793,9 @@ def proximity_counts( if return_sorted: # Sort each dict of proximities in descending order by count. - proximities = [sorted(p.items(), key=lambda x: x[1], reverse=True) for p in proximities] + proximities = [ + sorted(p.items(), key=lambda x: x[1], reverse=True) for p in proximities + ] else: proximities = [p.items() for p in proximities] diff --git a/quantile_forest/tests/test_quantile_forest.py b/quantile_forest/tests/test_quantile_forest.py index d35970e..67126ca 100755 --- a/quantile_forest/tests/test_quantile_forest.py +++ b/quantile_forest/tests/test_quantile_forest.py @@ -1264,7 +1264,9 @@ def _dicts_to_input_pairs(input_dicts): ] # Check that empty array is returned for weights that sum to 0. - actual = [calc_weighted_quantile(i, w, quantiles) for i, w in _dicts_to_weighted_inputs(inputs)] + actual = [ + calc_weighted_quantile(i, w, quantiles) for i, w in _dicts_to_weighted_inputs(inputs) + ] expected = [[], [], []] assert_array_equal(actual, expected)