fix doctests

elixir-nx · Nov 27, 2024 · f7b89bc · f7b89bc
1 parent 2a28cd4
commit f7b89bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/scholar/naive_bayes/bernoulli.ex b/lib/scholar/naive_bayes/bernoulli.ex
@@ -302,7 +302,7 @@ defmodule Scholar.NaiveBayes.Bernoulli do
       iex> model = Scholar.NaiveBayes.Bernoulli.fit(x, y, num_classes: 3)
       iex> Scholar.NaiveBayes.Bernoulli.predict(model, Nx.tensor([[6, 2, 4], [8, 5, 9]]), Nx.tensor([0, 1, 2]))
       #Nx.Tensor<
-        s64[2]
+        s32[2]
         [2, 2]
       >
   """