Linting

zillow · Feb 10, 2024 · 6913890 · 6913890
1 parent 9aa9aa5
commit 6913890
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git 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
@@ -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)