split_into_files.py

-from decentralizepy.datasets.Femnist import Femnist
+import sys
 
-f = Femnist()
+from decentralizepy.datasets.Reddit import Reddit
+from decentralizepy.mappings import Linear
 
-f.file_per_user("leaf/data/femnist/data/train", "leaf/data/femnist/per_user_data/train")
+if __name__ == "__main__":
+    mapping = Linear(6, 16)
+    f = Reddit(0, 0, mapping)
+    assert len(sys.argv) == 3
+    frm = sys.argv[1]
+    to = sys.argv[2]
+    f.file_per_user(frm, to)