From ac8e8e7e8eef14347a352eade797deb85b59997c Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Thu, 17 Feb 2022 17:26:06 +0100
Subject: [PATCH] Update file split script
---
split_into_files.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/split_into_files.py b/split_into_files.py
index b689e9b..a640a96 100644
--- a/split_into_files.py
+++ b/split_into_files.py
@@ -1,5 +1,10 @@
+import sys
from decentralizepy.datasets.Femnist import Femnist
-f = Femnist()
-f.file_per_user("leaf/data/femnist/data/train", "leaf/data/femnist/per_user_data/train")
+if __name__ == "__main__":
+ f = Femnist(None, None, None)
+ assert len(sys.argv) == 3
+ frm = sys.argv[1]
+ to = sys.argv[2]
+ f.file_per_user(frm, to)
--
GitLab