fix: re-instante proper error message for gather rank>1

mimiquate · Nov 13, 2023 · d372c16 · d372c16
1 parent 08ef7ca
commit d372c16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/candlex/backend.ex b/lib/candlex/backend.ex
@@ -386,7 +386,7 @@ defmodule Candlex.Backend do
     |> to_nx(out)
   end
 
-  def gather(%T{} = _out, %T{} = _tensor, %T{} = _indices) do
+  def gather(%T{} = _out, %T{} = _tensor, %T{} = _indices, _opts) do
     raise("unsupported gather for tensor of rank greater than 1")
   end