ci: add missing filter_nans kwarg to Pipeline test

MArpogaus · Sep 12, 2024 · d5729f5 · d5729f5
1 parent 740e51c
commit d5729f5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/test_pipleine.py b/tests/test_pipleine.py
@@ -196,7 +196,9 @@ def test_windowed_time_series_pipeline(
         batch_size=batch_size,
         drop_remainder=True,
     )
-    pipeline_kwargs = dict(cycle_length=1, shuffle_buffer_size=100, cache=True)
+    pipeline_kwargs = dict(
+        cycle_length=1, shuffle_buffer_size=100, cache=True, filter_nans=False
+    )
 
     with validate_args(
         history_size=history_size,
@@ -239,7 +241,9 @@ def test_windowed_time_series_pipeline_groupby(
         batch_size=batch_size,
         drop_remainder=False,
     )
-    pipeline_kwargs = dict(cycle_length=len(ids), shuffle_buffer_size=1000, cache=True)
+    pipeline_kwargs = dict(
+        cycle_length=len(ids), shuffle_buffer_size=1000, cache=True, filter_nans=False
+    )
 
     with validate_args(
         history_size=history_size,