Merge branch 'main' into main

Author: zachferr

pipeline_dp/pipeline_operations.py

@@ -150,7 +150,7 @@ class LocalPipelineOperations(PipelineOperations):
     def map(self, col, fn, stage_name: str = None):
         return map(fn, col)
 
-    def map_tuple(self, col, fn, stage_name: str):
+    def map_tuple(self, col, fn, stage_name: str = None):
         return (fn(k, v) for k, v in col)
 
     def map_values(self, col, fn, stage_name: str):

tests/pipeline_operations_test.py

@@ -55,20 +55,10 @@ def test_local_map(self):
         self.assertEqual(list(self.ops.map(range(5), lambda x: x ** 2)),
                          [0, 1, 4, 9, 16])
 
-    def test_local_map_tuple(self):
-        some_map = self.ops.map([1, 2, 3], lambda x: x)
-        # some_map is its own consumable iterator
-        self.assertIs(some_map, iter(some_map))
-
-        self.assertEqual(list(self.ops.map([1, 2, 3], str)),
-                         ["1", "2", "3"])
-        self.assertEqual(list(self.ops.map(range(5), lambda x: x ** 2)),
-                         [0, 1, 4, 9, 16])
-
     def test_local_map_tuple(self):
         tuple_list = [(1, 2), (2, 3), (3, 4)]
 
-        self.assertEqual(list(self.ops.map_tuple(tuple_list, lambda k, v: k + v)),
+        self.assertEqual(list(self.ops.map_tuple(tuple_list, lambda k, v: k+v)),
                          [3, 5, 7])
 
         self.assertEqual(list(self.ops.map_tuple(tuple_list, lambda k, v: (