Fix DataFrame.filter_with spec

I believe since the callback can return a list of lazy series queries
elixir-explorer · Aug 30, 2024 · 54f19a3 · 54f19a3
1 parent 65b01e3
commit 54f19a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/explorer/data_frame.ex b/lib/explorer/data_frame.ex
@@ -2606,7 +2606,7 @@ defmodule Explorer.DataFrame do
   @doc type: :single
   @spec filter_with(
           df :: DataFrame.t(),
-          callback :: (Explorer.Backend.LazyFrame.t() -> Series.lazy_t())
+          callback :: (Explorer.Backend.LazyFrame.t() -> [Series.lazy_t()])
         ) :: DataFrame.t()
   def filter_with(df, fun) when is_function(fun, 1) do
     ldf = Explorer.Backend.LazyFrame.new(df)