From 2786931915b958be35908ab131c6a36a7e26c275 Mon Sep 17 00:00:00 2001
From: Reid Johnson <reidj@zillowgroup.com>
Date: Fri, 6 Sep 2024 03:26:33 -0700
Subject: [PATCH] Update docstrings

---
 quantile_forest/_quantile_forest.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/quantile_forest/_quantile_forest.py b/quantile_forest/_quantile_forest.py
index e2a80e7..5f05e53 100755
--- a/quantile_forest/_quantile_forest.py
+++ b/quantile_forest/_quantile_forest.py
@@ -934,8 +934,8 @@ def proximity_counts(
 
         Returns
         -------
-        proximities : list of tuples
-            List of tuples mapping sample indices to proximity counts.
+        proximities : list of tuples, length = n_samples
+            List of tuples mapping training sample indices to proximity counts.
 
         Notes
         -----