Fix mypy error

mlcommons · Nov 26, 2024 · 04dcac3 · 04dcac3
1 parent 196278b
commit 04dcac3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/mlcroissant/mlcroissant/_src/datasets.py b/python/mlcroissant/mlcroissant/_src/datasets.py
@@ -176,15 +176,17 @@ def __iter__(self):
                 record_set=self.record_set, operations=operations
             )
 
-    def beam_reader(self, pipeline: beam.Pipeline, filters: Mapping[str, Any]):
+    def beam_reader(
+        self, pipeline: beam.Pipeline, filters: Mapping[str, Any] | None = None
+    ):
         """See ReadFromCroissant docstring."""
         operations = self._filter_interesting_operations(self.filters)
         execute_downloads(operations)
         return execute_operations_in_beam(
             pipeline=pipeline,
             record_set=self.record_set,
             operations=operations,
-            filters=self.filters,
+            filters=filters or self.filters,
         )
 
     def _filter_interesting_operations(self, filters: Filters | None) -> Operations: