Incorporate feedback.

Co-authored-by: José Valim <[email protected]>
elixir-nx · Dec 24, 2023 · 2e120b6 · 2e120b6
1 parent 1613da4
commit 2e120b6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/scholar/neighbors/random_projection_forest.ex b/lib/scholar/neighbors/random_projection_forest.ex
@@ -60,7 +60,8 @@ defmodule Scholar.Neighbors.RandomProjectionForest do
 
       iex> key = Nx.Random.key(12)
       iex> tensor = Nx.iota({5, 2})
-      iex> Scholar.Neighbors.RandomProjectionForest.fit(tensor, num_trees: 3, min_leaf_size: 2, key: key).indices
+      iex> forest = Scholar.Neighbors.RandomProjectionForest.fit(tensor, num_trees: 3, min_leaf_size: 2, key: key)
+      iex> forest.indices
       #Nx.Tensor<
         u32[3][5]
         [