add schema parameter to the repartition method (#192)

sararb · sararb · commit 2fc6889cc5e9 · 2022-12-29T18:25:43.000-05:00
diff --git a/merlin/io/dataset.py b/merlin/io/dataset.py
@@ -543,7 +543,8 @@ def repartition(self, npartitions=None, partition_size=None):
             .repartition(
                 npartitions=npartitions,
                 partition_size=partition_size,
-            )
+            ),
+            schema=self.schema,
         )
 
     @classmethod

Original file line number	Diff line number	Diff line change
`@@ -543,7 +543,8 @@ def repartition(self, npartitions=None, partition_size=None):`
`543`	`543`	`.repartition(`
`544`	`544`	`npartitions=npartitions,`
`545`	`545`	`partition_size=partition_size,`
`546`		`- )`
	`546`	`+ ),`
	`547`	`+ schema=self.schema,`
`547`	`548`	`)`
`548`	`549`
`549`	`550`	`@classmethod`