nit

facebookresearch · Oct 10, 2023 · 51f4d2f · 51f4d2f
1 parent 4b71687
commit 51f4d2f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fairseq2n/python/src/fairseq2n/bindings/data/data_pipeline.cc b/fairseq2n/python/src/fairseq2n/bindings/data/data_pipeline.cc
@@ -420,9 +420,8 @@ def_data_pipeline(py::module_ &data_module)
                     opt_overrides = *std::move(maybe_opt_overrides);
 
                 map_fn f = collater(opts, std::move(opt_overrides));
-                element_mapper mapper{f, std::nullopt};
 
-                self = std::move(self).map(std::move(mapper), num_parallel_calls);
+                self = std::move(self).map(std::move(f), num_parallel_calls);
 
                 return self;
             },

diff --git a/src/fairseq2/data/data_pipeline.py b/src/fairseq2/data/data_pipeline.py
@@ -163,6 +163,7 @@ def collate(
         ) -> Self:
             """Concatenate a list of inputs into a single inputs.
 
+            This is equivalent to calling `.map(Collater())`.
             See :py:class:`fairseq2.data.Collater` for details.
             """