From 54f19a3fad328b12325984bd9982c3660cd9e37c Mon Sep 17 00:00:00 2001 From: Mitchell Hanberg Date: Fri, 30 Aug 2024 16:49:30 -0400 Subject: [PATCH] Fix DataFrame.filter_with spec I believe since the callback can return a list of lazy series queries --- lib/explorer/data_frame.ex | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/explorer/data_frame.ex b/lib/explorer/data_frame.ex index ea722d322..03e2effde 100644 --- 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)